Commit
·
eea06fe
1
Parent(s):
b3b99a7
Update app.py
Browse files
app.py
CHANGED
|
@@ -132,75 +132,75 @@ def gr_hide():
|
|
| 132 |
with gr.Blocks() as demo:
|
| 133 |
|
| 134 |
with gr.Tab("和ChatGLM谈天说地"):
|
| 135 |
-
|
| 136 |
-
|
| 137 |
|
| 138 |
-
|
| 139 |
-
|
| 140 |
|
| 141 |
-
|
| 142 |
|
| 143 |
### <center>🥰 - 滔滔AI,让有爱的AI滔滔不绝</center>
|
| 144 |
### <center>🦄 - TalkTalkAI, let lovely AI brighten the future</center>
|
| 145 |
|
| 146 |
-
|
| 147 |
-
|
| 148 |
-
|
| 149 |
-
|
| 150 |
-
|
| 151 |
-
|
| 152 |
-
|
| 153 |
-
|
| 154 |
-
|
| 155 |
-
|
| 156 |
-
|
| 157 |
-
|
| 158 |
-
|
| 159 |
-
|
| 160 |
-
|
| 161 |
-
|
| 162 |
-
|
| 163 |
-
|
| 164 |
-
|
| 165 |
-
|
| 166 |
-
|
| 167 |
-
|
| 168 |
-
|
| 169 |
-
|
| 170 |
-
|
| 171 |
-
|
| 172 |
-
|
| 173 |
-
|
| 174 |
-
|
| 175 |
-
|
| 176 |
-
|
| 177 |
-
|
| 178 |
-
|
| 179 |
-
|
| 180 |
-
|
| 181 |
-
|
| 182 |
-
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
|
| 186 |
-
|
| 187 |
-
|
| 188 |
-
|
| 189 |
-
|
| 190 |
-
|
| 191 |
-
|
| 192 |
-
|
| 193 |
-
|
| 194 |
-
|
| 195 |
-
|
| 196 |
-
|
| 197 |
|
| 198 |
-
|
| 199 |
<div class="footer">
|
| 200 |
-
<p>🖼️💕🎡 - 滔滔AI
|
| 201 |
</p>
|
| 202 |
</div>
|
| 203 |
-
|
| 204 |
|
| 205 |
with gr.Tab("发现更多有趣功能"):
|
| 206 |
|
|
@@ -223,12 +223,13 @@ with gr.Blocks() as demo:
|
|
| 223 |
)
|
| 224 |
|
| 225 |
|
| 226 |
-
|
| 227 |
<div class="footer">
|
| 228 |
-
<p>🖼️💕🎡 - 滔滔AI
|
| 229 |
</p>
|
|
|
|
| 230 |
</div>
|
| 231 |
-
|
| 232 |
|
| 233 |
if __name__ == '__main__':
|
| 234 |
demo.queue(concurrency_count=5, max_size=20).launch(debug=True)
|
|
|
|
| 132 |
with gr.Blocks() as demo:
|
| 133 |
|
| 134 |
with gr.Tab("和ChatGLM谈天说地"):
|
| 135 |
+
if not os.path.isfile('config.json'):
|
| 136 |
+
save_config()
|
| 137 |
|
| 138 |
+
with open('config.json', 'r', encoding='utf-8') as f:
|
| 139 |
+
configs = json.loads(f.read())
|
| 140 |
|
| 141 |
+
gr.Markdown('''# <center>🥳 滔滔AI 🎶</center>
|
| 142 |
|
| 143 |
### <center>🥰 - 滔滔AI,让有爱的AI滔滔不绝</center>
|
| 144 |
### <center>🦄 - TalkTalkAI, let lovely AI brighten the future</center>
|
| 145 |
|
| 146 |
+
''')
|
| 147 |
+
|
| 148 |
+
with gr.Row():
|
| 149 |
+
max_length = gr.Slider(minimum=4.0, maximum=4096.0, step=4.0, label='Max Length', value=configs['max_length'])
|
| 150 |
+
top_p = gr.Slider(minimum=0.01, maximum=1.0, step=0.01, label='Top P', value=configs['top_p'])
|
| 151 |
+
temperature = gr.Slider(minimum=0.01, maximum=2.0, step=0.01, label='Temperature', value=configs['temperature'])
|
| 152 |
+
memory_limit = gr.Slider(minimum=0.0, maximum=20.0, step=1.0, label='Memory Limit', value=configs['memory_limit'])
|
| 153 |
+
save_conf = gr.Button('保存设置', visible=False)
|
| 154 |
+
|
| 155 |
+
gr.Markdown('''<h2>Hint: click on a chat bubble to edit chat history</h2>''')
|
| 156 |
+
|
| 157 |
+
state = gr.State([])
|
| 158 |
+
chatbot = gr.Chatbot(elem_id='chatbot', show_label=False)
|
| 159 |
+
with gr.Row(visible=False) as edit_log:
|
| 160 |
+
with gr.Column():
|
| 161 |
+
log = gr.Textbox()
|
| 162 |
+
with gr.Row():
|
| 163 |
+
submit_log = gr.Button('保存')
|
| 164 |
+
cancel_log = gr.Button('取消')
|
| 165 |
+
log_idx = gr.State([])
|
| 166 |
+
|
| 167 |
+
message = gr.Textbox(placeholder='Input your message', label='Q:')
|
| 168 |
+
|
| 169 |
+
with gr.Row():
|
| 170 |
+
submit = gr.Button('Submit')
|
| 171 |
+
edit = gr.Button('Edit last question')
|
| 172 |
+
regen = gr.Button('Re-generate')
|
| 173 |
+
|
| 174 |
+
delete = gr.Button('Reset chat')
|
| 175 |
+
|
| 176 |
+
with gr.Row(visible=False):
|
| 177 |
+
save = gr.Button('保存对话(在 `log` 文件夹下)')
|
| 178 |
+
load = gr.UploadButton('读取对话', file_types=['file'], file_count='single')
|
| 179 |
+
|
| 180 |
+
input_list = [message, chatbot, state, max_length, top_p, temperature, memory_limit]
|
| 181 |
+
output_list = [chatbot, state, message]
|
| 182 |
+
edit_list = [edit_log, log, log_idx]
|
| 183 |
+
|
| 184 |
+
save_conf.click(save_config, inputs=input_list[3:])
|
| 185 |
+
load.upload(load_history, inputs=[load, chatbot, state], outputs=output_list + edit_list)
|
| 186 |
+
save.click(save_history, inputs=[state])
|
| 187 |
+
message.submit(chat_wrapper, inputs=input_list, outputs=output_list + edit_list)
|
| 188 |
+
submit.click(chat_wrapper, inputs=input_list, outputs=output_list + edit_list)
|
| 189 |
+
edit.click(edit_wrapper, inputs=input_list[1:3], outputs=output_list + edit_list)
|
| 190 |
+
regen.click(regenerate_wrapper, inputs=input_list[1:], outputs=output_list + edit_list)
|
| 191 |
+
delete.click(reset_history, outputs=output_list + edit_list)
|
| 192 |
+
chatbot.select(gr_show_and_load, inputs=[state], outputs=edit_list)
|
| 193 |
+
edit_kwargs = {'inputs': [chatbot, state, log, log_idx], 'outputs': [chatbot, state] + edit_list}
|
| 194 |
+
log.submit(update_history, **edit_kwargs)
|
| 195 |
+
submit_log.click(update_history, **edit_kwargs)
|
| 196 |
+
cancel_log.click(gr_hide, outputs=edit_list)
|
| 197 |
|
| 198 |
+
gr.HTML('''
|
| 199 |
<div class="footer">
|
| 200 |
+
<p>🖼️💕🎡 - 滔滔AI,为爱滔滔;有意清秋入衡霍,为君无尽写江天
|
| 201 |
</p>
|
| 202 |
</div>
|
| 203 |
+
''')
|
| 204 |
|
| 205 |
with gr.Tab("发现更多有趣功能"):
|
| 206 |
|
|
|
|
| 223 |
)
|
| 224 |
|
| 225 |
|
| 226 |
+
gr.HTML('''
|
| 227 |
<div class="footer">
|
| 228 |
+
<p>🖼️💕🎡 - 滔滔AI,为爱滔滔;有意清秋入衡霍,为君无尽写江天
|
| 229 |
</p>
|
| 230 |
+
|
| 231 |
</div>
|
| 232 |
+
''')
|
| 233 |
|
| 234 |
if __name__ == '__main__':
|
| 235 |
demo.queue(concurrency_count=5, max_size=20).launch(debug=True)
|