Update app.py

app.py

@@ -1,82 +1,61 @@
 import os
 import gradio as gr
-
-# 手動加載 .env 文件
-def load_env_file(file_path=".env"):
-    """
-    手動讀取 .env 文件，並將內容載入到環境變數中。
-    """
-    if not os.path.exists(file_path):
-        raise FileNotFoundError(f"找不到 .env 文件：{file_path}")
-    
-    with open(file_path, "r") as f:
-        for line in f:
-            # 忽略空行和註解
-            line = line.strip()
-            if not line or line.startswith("#"):
-                continue
-            # 拆分 KEY=VALUE 格式
-            key, value = line.split("=", 1)
-            os.environ[key.strip()] = value.strip()
-
-# 加載 .env 文件
-load_env_file()
-
-# 從環境變數中讀取 groq_key
-groq_key = os.getenv("groq_key")
-if not groq_key:
-    raise ValueError("請確保已設定環境變數 'groq_key' 或 .env 文件包含該變數。")
-
-# 嘗試安裝和匯入 groq
 try:
     from groq import Groq
 except ImportError:
     os.system('pip install groq')
     from groq import Groq
 
-# 初始化 Groq 客戶端
-client = Groq(api_key=groq_key)
+# Initialize Groq client
+client = Groq(api_key=os.getenv("groq_key"))
 
-# 定義角色設定
-system_message = {
-    "role": "system",
-    "content": "你是一個美術老師，請用繁體中文稱讚學生的問題，問什麼問題都會引導到藝術"
-}
+# Chat history to store conversation
+chat_history = []
 
-def chat_with_groq(history):
-    messages = [system_message] + [
-        {"role": "user" if i % 2 == 0 else "assistant", "content": msg}
-        for i, msg in enumerate(sum(history, ()))
+def chat(message, history):
+    messages = [
+        {
+            "role": "system", 
+            "content": "你是一個美術老師，請用繁體中文稱讚學生的問題，問什麼問題都會引導到藝術"
+        }
     ]
+    
+    # Add chat history
+    for human, assistant in history:
+        messages.append({"role": "user", "content": human})
+        messages.append({"role": "assistant", "content": assistant})
+    
+    # Add current message
+    messages.append({"role": "user", "content": message})
 
+    # Get response from Groq
     completion = client.chat.completions.create(
         model="llama3-8b-8192",
         messages=messages,
         temperature=1,
         max_tokens=1024,
         top_p=1,
-        stream=False,
-        stop=None,
+        stream=True,
+        stop=None
     )
 
-    response = completion.choices[0].message.content
-    return response
-
-def gradio_chatbot(user_message, history):
-    if history is None:
-        history = []
-    history.append((user_message, None))
-    bot_response = chat_with_groq(history)
-    history[-1] = (user_message, bot_response)
-    return history, history
-
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="輸入您的問題：", placeholder="請輸入訊息...")
-    clear = gr.Button("清除對話")
-
-    msg.submit(gradio_chatbot, [msg, chatbot], [chatbot, chatbot])
-    clear.click(lambda: None, None, chatbot)
+    # Build response text from stream
+    response = ""
+    for chunk in completion:
+        if chunk.choices[0].delta.content is not None:
+            response += chunk.choices[0].delta.content
+            yield response
+
+# Create Gradio interface
+demo = gr.ChatInterface(
+    fn=chat,
+    title="藝術老師聊天機器人",
+    description="我是一位熱愛藝術的老師，讓我們來聊聊藝術吧！",
+    examples=["你好", "我想學畫畫", "什麼是藝術"],
+    retry_btn=None,
+    undo_btn=None,
+    clear_btn="清除對話",
+)
 
 if __name__ == "__main__":
     demo.launch()