Update app.py

app.py

@@ -2,14 +2,14 @@ import gradio as gr
 import os
 import json
 import requests
-from typing import List, Dict
+from typing import List, Tuple
 
 # 環境變數設置
 XAI_API_KEY = os.getenv("XAI_API_KEY")
 if not XAI_API_KEY:
     raise ValueError("Please set XAI_API_KEY environment variable")
 
-def call_grok_api(messages: List[Dict]) -> str:
+def call_grok_api(messages: List[dict]) -> str:
     """調用Grok API"""
     headers = {
         "Content-Type": "application/json",
@@ -20,7 +20,7 @@ def call_grok_api(messages: List[Dict]) -> str:
         "messages": messages,
         "model": "grok-beta",
         "stream": False,
-        "temperature": 0.7  # 可以調整此值來控制回答的創造性
+        "temperature": 0.7
     }
     
     try:
@@ -34,8 +34,8 @@ def call_grok_api(messages: List[Dict]) -> str:
     except requests.exceptions.RequestException as e:
         return f"Error calling Grok API: {str(e)}"
 
-def format_history(history: List[Dict]) -> List[Dict]:
-    """將Gradio的歷史記錄格式化為Grok API需要的格式"""
+def format_history(history: List[Tuple[str, str]]) -> List[dict]:
+    """將聊天歷史格式化為Grok API需要的格式"""
     formatted_messages = [
         {
             "role": "system",
@@ -43,27 +43,30 @@ def format_history(history: List[Dict]) -> List[Dict]:
         }
     ]
     
-    for msg in history:
-        formatted_messages.append({
-            "role": "user" if msg["role"] == "user" else "assistant",
-            "content": msg["content"]
-        })
+    for user_msg, assistant_msg in history:
+        if user_msg:
+            formatted_messages.append({
+                "role": "user",
+                "content": user_msg
+            })
+        if assistant_msg:
+            formatted_messages.append({
+                "role": "assistant",
+                "content": assistant_msg
+            })
     
     return formatted_messages
 
-def respond(message: str, history: List[Dict]) -> Dict:
+def respond(message: str, history: List[Tuple[str, str]]) -> List[Tuple[str, str]]:
     """處理用戶輸入並生成回應"""
-    history.append({"role": "user", "content": message})
-    
-    # 格式化消息歷史
-    formatted_history = format_history(history)
+    # 格式化歷史記錄並添加新消息
+    history_for_api = format_history(history + [(message, None)])
     
     # 獲取Grok的回應
-    response = call_grok_api(formatted_history)
-    
-    # 添加助手回應到歷史記錄
-    history.append({"role": "assistant", "content": response})
+    response = call_grok_api(history_for_api)
     
+    # 返回更新後的歷史記錄
+    history.append((message, response))
     return history
 
 # 創建Gradio界面
@@ -75,8 +78,6 @@ with gr.Blocks() as demo:
     
     chatbot = gr.Chatbot(
         value=[],
-        bubble_full_width=False,
-        avatar_images=(None, "🤖"),
         height=500
     )