Update app.py

app.py

@@ -36,13 +36,32 @@ def predict(message, history):
     return response.strip()
 
 # 创建Gradio界面
-demo = gr.ChatInterface(
-    predict,
-    title="测试Zhi-writing-dsr1-14b",
-    description="Zhihu-ai/Zhi-writing-dsr1-14b",
-    examples=["鲁迅口吻写五百字，描述桔猫的可爱！", "桔了个仔是谁", "介绍自己"],
-    theme=gr.themes.Soft()
-)
+# demo = gr.ChatInterface(
+#     predict,
+#     title="测试Zhi-writing-dsr1-14b",
+#     description="Zhihu-ai/Zhi-writing-dsr1-14b",
+#     examples=["鲁迅口吻写五百字，描述桔猫的可爱！", "桔了个仔是谁", "介绍自己"],
+#     theme=gr.themes.Soft()
+# )
+
+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)