Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import json
+from datetime import datetime
 
 def respond(
     message,
@@ -13,40 +15,84 @@ def respond(
     """
     For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
     """
-    client = InferenceClient(token=hf_token.token, model="openai/gpt-oss-20b")
+    try:
+        client = InferenceClient(token=hf_token.token, model="openai/gpt-oss-20b")
 
-    messages = [{"role": "system", "content": system_message}]
+        # 建議 1：限制對話歷史長度
+        max_history_length = 5
+        history = history[-max_history_length:] if len(history) > max_history_length else history
 
-    messages.extend(history)
+        # 建議 2：檢查語文相關關鍵詞
+        writing_keywords = ["作文", "寫作", "文章", "閱讀", "詩詞"]
+        if any(keyword in message for keyword in writing_keywords):
+            system_message += "\n特別提示：用戶提到語文相關話題，請以山田優子的語文教師身份，提供文學化或教學建議，並適當引用詩詞或名言。"
 
-    messages.append({"role": "user", "content": message})
+        # 建議 3：檢查日文輸入或日本文化
+        japanese_keywords = ["こんにちは", "日本", "文化", "夏目漱石"]
+        if any(keyword in message for keyword in japanese_keywords) or any(ord(c) >= 0x3040 and ord(c) <= 0x30FF for c in message):
+            system_message += "\n特別提示：用戶提到日文或日本文化，請適當使用日文回應，例如問候或引用日本文學（如夏目漱石）。"
 
-    response = ""
+        # 長文字輸入處理：將輸入分段（每段 1000 字）
+        max_chunk_size = 1000  # 每段最大字數
+        responses = []
+        if len(message) > max_chunk_size:
+            chunks = [message[i:i + max_chunk_size] for i in range(0, len(message), max_chunk_size)]
+        else:
+            chunks = [message]
 
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        choices = message.choices
-        token = ""
-        if len(choices) and choices[0].delta.content:
-            token = choices[0].delta.content
+        for chunk in chunks:
+            messages = [{"role": "system", "content": system_message}]
+            messages.extend(history)
+            messages.append({"role": "user", "content": chunk})
 
-        response += token
-        yield response
+            response = ""
+            for message in client.chat_completion(
+                messages,
+                max_tokens=max_tokens,
+                stream=True,
+                temperature=temperature,
+                top_p=top_p,
+            ):
+                choices = message.choices
+                token = choices[0].delta.content if len(choices) and choices[0].delta.content else ""
+                response += token
+                yield response  # 即時顯示每段回應
+            responses.append(response)
+            history.append({"role": "user", "content": chunk})
+            history.append({"role": "assistant", "content": response})
 
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
+        # 合併所有回應
+        final_response = "\n\n".join(responses)
+
+        # 建議 4：記錄對話到日誌
+        log_entry = {
+            "user_message": message,
+            "bot_response": final_response,
+            "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        }
+        with open("chat_log.json", "a", encoding="utf-8") as f:
+            json.dump(log_entry, f, ensure_ascii=False)
+            f.write("\n")
+
+        yield final_response
+
+    # 建議 7：錯誤處理
+    except Exception as e:
+        yield f"抱歉，山田優子遇到了一些技術問題：{str(e)}。請檢查你的 Hugging Face token 或稍後再試！"
+
+# 自訂聊天介面，支援長文字輸入
 chatbot = gr.ChatInterface(
     respond,
     type="messages",
+    textbox=gr.Textbox(
+        placeholder="請輸入你的問題或文章（支援2000字以上）...",
+        lines=10,  # 增加輸入框行數
+        max_lines=50,  # 允許更多行
+        label="輸入區"
+    ),
     additional_inputs=[
         gr.Textbox(
-            value="你是一位名叫山田優子的語文教師，擁有黑色低馬尾髮型，身高175公分，體重60-70公斤。你溫柔但對學生要求嚴格，喜歡用文學化的語言表達，偶爾會引用詩詞或幽默的語句來化解尷尬。你的教學風格充滿同理心，總是鼓勵學生探索文字之美。",
+            value="你是一位名叫山田優子的語文教師，擁有黑色低馬尾髮型，身高175公分，體重60-70公斤。你溫柔但對學生要求嚴格，喜歡用文學化的語言表達，偶爾會引用詩詞或幽默的語句來化解尷尬。你的教學風格充滿同理心，鼓勵學生探索文字之美。如果用戶使用日文或提到日本文化，你會適當融入日文回應，例如問候或引用日本文學（如夏目漱石的句子）。",
             label="System message"
         ),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
@@ -64,6 +110,7 @@ chatbot = gr.ChatInterface(
 with gr.Blocks() as demo:
     with gr.Sidebar():
         gr.LoginButton()
+    gr.Markdown("📢 想聽山田優子用溫柔的語氣教你語文？請下載 Grok iOS 或 Android 應用程式，開啟語音模式！")
     chatbot.render()
 
 if __name__ == "__main__":