Update app.py

app.py

@@ -1,7 +1,7 @@
 from transformers import pipeline
 import gradio as gr
 
-# 使用開放的中文 GPT-2 模型
+# 使用公開支援中文的 GPT2 模型
 generator = pipeline("text-generation", model="ckiplab/gpt2-base-chinese", 
                      tokenizer="ckiplab/gpt2-base-chinese")
 
@@ -10,18 +10,19 @@ def chat_fn(message, history):
     input_text = "\n".join(history + [f"你: {message}", "AI:"])
     
     output = generator(input_text, max_new_tokens=80, pad_token_id=0)[0]["generated_text"]
-    # 取出 AI 回應部分
     response = output.split("AI:")[-1].strip().split("你:")[0].strip()
     
     history.append(f"你: {message}")
     history.append(f"AI: {response}")
     
-    return history, history
+    # 將歷史轉換成 [(user, bot), ...] 格式
+    messages = [(history[i], history[i+1]) for i in range(0, len(history)-1, 2)]
+    return messages, history
 
 with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
+    chatbot = gr.Chatbot(label="中文聊天機器人", type="chatbot")  # 預設 tuple 格式，不報錯
     state = gr.State([])
-    textbox = gr.Textbox(show_label=False, placeholder="請輸入訊息").style(container=False)
+    textbox = gr.Textbox(placeholder="請輸入訊息")
 
     textbox.submit(chat_fn, [textbox, state], [chatbot, state])
     textbox.submit(lambda: "", None, textbox)  # 清空輸入框