Update app.py

app.py

@@ -1,30 +1,25 @@
 from transformers import pipeline
 import gradio as gr
 
-# 使用公開支援中文的 GPT2 模型
 generator = pipeline("text-generation", model="ckiplab/gpt2-base-chinese", 
                      tokenizer="ckiplab/gpt2-base-chinese")
 
 def chat_fn(message, history):
     history = history or []
     input_text = "\n".join(history + [f"你: {message}", "AI:"])
-    
     output = generator(input_text, max_new_tokens=80, pad_token_id=0)[0]["generated_text"]
     response = output.split("AI:")[-1].strip().split("你:")[0].strip()
-    
     history.append(f"你: {message}")
     history.append(f"AI: {response}")
-    
-    # 將歷史轉換成 [(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(label="中文聊天機器人", type="chatbot")  # 預設 tuple 格式，不報錯
+    chatbot = gr.Chatbot(label="中文聊天機器人", type="tuples")
     state = gr.State([])
     textbox = gr.Textbox(placeholder="請輸入訊息")
 
     textbox.submit(chat_fn, [textbox, state], [chatbot, state])
-    textbox.submit(lambda: "", None, textbox)  # 清空輸入框
+    textbox.submit(lambda: "", None, textbox)
 
 demo.launch()