Update app.py

app.py

@@ -37,15 +37,26 @@ def predict(message, history):
     ):
         yield response.strip()
 
-# 创建Gradio界面
-demo = gr.ChatInterface(
-    predict,
-    title="Zhi-writing-dsr1-14",
-    description="这是一个基于Zhi-writing-dsr1-14的文章生成器。",
-    examples=["以鲁迅口吻写一篇500字关于桔了个仔的散文", "用知乎常见的表达方式讲讲什么是AI？", "告诉我一个我大概率不知道的人生哲理"],
-    theme=gr.themes.Soft(),
-    streaming=True
-)
+
+
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# Zhi-writing-dsr1-14")
+    gr.Markdown("这是一个基于Zhi-writing-dsr1-14的文章生成器")
+    
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(label="输入消息")
+    clear = gr.Button("清除对话")
+    
+    def respond(message, chat_history):
+        bot_message = ""
+        for response in predict(message, chat_history):
+            bot_message = response
+            chat_history.append((message, bot_message))
+            yield chat_history
+        return "", chat_history
+    
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+    clear.click(lambda: None, None, chatbot, queue=False)
 
 if __name__ == "__main__":
     demo.launch(share=True)