Update app.py

app.py

@@ -67,6 +67,19 @@ def run_sync(user_input):
     print(f"Running sync with input: {user_input}")
     return asyncio.run(handle_input(user_input))
 
+# 定義反饋處理函數
+def save_feedback(user_input, response, feedback_type, improvement):
+    feedback = {
+        "user_input": user_input,
+        "response": response,
+        "feedback_type": feedback_type,
+        "improvement": improvement
+    }
+    print(f"Saving feedback: {feedback}")
+    # 假設你有一個保存反饋的機制，可以是保存到文件或發送到服務器
+    # 這裡簡單打印出來，實際應用中應該保存反饋
+    return "感謝您的反饋！"
+
 # 定義 Gradio 界面
 user_input = gr.Textbox(label='歡迎問我加密貨幣交易所的各種疑難雜症', placeholder='在此輸入問題...')
 examples = [
@@ -91,6 +104,10 @@ with gr.Blocks() as iface:
         user_input = gr.Textbox(label='輸入您的問題', placeholder="在此輸入問題...")
         submit_button = gr.Button("送出")
     gr.Examples(examples=examples, inputs=user_input)
+    with gr.Row():
+        like_button = gr.Button("👍")
+        dislike_button = gr.Button("👎")
+        improvement_input = gr.Textbox(label='請輸入改進建議', placeholder='請輸入如何改進模型回應的建議')
     with gr.Row():
         feedback_output = gr.Textbox(label='反饋結果', interactive=False)
 
@@ -99,6 +116,23 @@ with gr.Blocks() as iface:
         history.append((user_input, response))
         return history, history
 
+    def handle_feedback(response, feedback_type, improvement):
+        global last_user_input
+        feedback_message = save_feedback(last_user_input, response, feedback_type, improvement)
+        return feedback_message
+
     submit_button.click(fn=chat, inputs=[user_input, chatbot], outputs=[chatbot, chatbot])
 
+    like_button.click(
+        fn=lambda response, improvement: handle_feedback(response, "like", improvement),
+        inputs=[chatbot, improvement_input],
+        outputs=feedback_output
+    )
+
+    dislike_button.click(
+        fn=lambda response, improvement: handle_feedback(response, "dislike", improvement),
+        inputs=[chatbot, improvement_input],
+        outputs=feedback_output
+    )
+
 iface.launch()