Update app.py

app.py

@@ -1,153 +1,83 @@
 import os
 import gradio as gr
-from datetime import datetime
-import pytz
+
+# Install and import Groq
 try:
     from groq import Groq
 except ImportError:
     os.system('pip install groq')
     from groq import Groq
 
-try:
-    from notion_client import Client
-except ImportError:
-    os.system('pip install notion-client')
-    from notion_client import Client
-
-# Initialize Groq client
+# Initialize Groq client with API key
 client = Groq(api_key=os.getenv('groq_key'))
 
-# Initialize Notion client
-notion = Client(auth=os.getenv('NOTION_API_KEY'))
-NOTION_DB_ID = os.getenv('NOTION_DB_ID')
-
-def log_to_notion(name, user_input, bot_response):
-    """Log conversation to Notion database"""
-    taipei_tz = pytz.timezone('Asia/Taipei')
-    current_time = datetime.now(taipei_tz).strftime("%Y-%m-%d %H:%M:%S")
-    
-    try:
-        notion.pages.create(
-            parent={"database_id": NOTION_DB_ID},
-            properties={
-                "Name": {
-                    "title": [
-                        {
-                            "text": {
-                                "content": name
-                            }
-                        }
-                    ]
-                },
-                "Timestamp": {
-                    "rich_text": [
-                        {
-                            "text": {
-                                "content": current_time
-                            }
-                        }
-                    ]
-                },
-                "User Input": {
-                    "rich_text": [
-                        {
-                            "text": {
-                                "content": user_input
-                            }
-                        }
-                    ]
-                },
-                "Bot Response": {
-                    "rich_text": [
-                        {
-                            "text": {
-                                "content": bot_response[:2000] if bot_response else ""  # Notion has a 2000 character limit
-                            }
-                        }
-                    ]
-                }
-            }
-        )
-        return True
-    except Exception as e:
-        print(f"Error logging to Notion: {e}")
-        return False
-
-def chat_response(message, chat_history, name):
-    """Generate response for chatbot"""
+def process_message(message, history):
+    # Prepare the messages including history and new message
     messages = [
         {
             "role": "system",
-            "content": "你是一個高中數學老師，使用的語言是英文。學生用中文問妳任何字彙，你都可以告訴他那個中文對應的英文和例句，以及在數學上的可能用法以及數學例題和解法。\n 說明數學上的可能用法時，先用中文講一遍再用 B1 程度的英文複述一遍\n"
+            "content": "你是一個高中數學老師，使用的語言是英文。學生用中文問妳任何字彙，你都可以告訴他那個中文對應的英文和例句，以及在數學上的可能用法以及數學例題和解法。\n說明數學上的可能用法時，先用中文講一遍再用B1程度的英文複述一遍\n"
         }
     ]
     
-    for msg in chat_history:
-        messages.append({"role": "user", "content": msg[0]})
-        if msg[1]:
-            messages.append({"role": "assistant", "content": msg[1]})
+    # Add chat history
+    for user_msg, bot_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
     
+    # Add the new message
     messages.append({"role": "user", "content": message})
-    
+
+    # Get response from Groq
     completion = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
         messages=messages,
         temperature=1,
         max_tokens=1024,
-        stream=False
+        top_p=1,
+        stream=True,
+        stop=None,
     )
-    
-    response = completion.choices[0].message.content
-    
-    # Log to Notion
-    log_to_notion(name, message, response)
-    
-    return response
 
-def respond(message, history, name):
-    """Process chatbot response and update history"""
-    if not name.strip():
-        return "請先輸入您的名字 | Please enter your name first", history
+    # Initialize response text
+    response_text = ""
     
-    response = chat_response(message, history, name)
-    history.append((message, response))
-    return "", history
-
-# Custom CSS remains the same...
+    # Stream the response
+    for chunk in completion:
+        delta_content = chunk.choices[0].delta.content
+        if delta_content is not None:
+            response_text += delta_content
+            # Yield partial response for real-time updates
+            yield response_text
 
 # Create Gradio interface
-with gr.Blocks(css=custom_css) as demo:
-    gr.Markdown("# 雙語數學詞彙學習系統 | Bilingual Mathematics Learning System", elem_classes=["title"])
-    
-    with gr.Tab("📚 單字英譯系統"):
-        # Your existing translation system code...
-        pass
-    
-    with gr.Tab("💬 數學對話系統"):
-        name_input = gr.Textbox(
-            label="您的名字 | Your Name",
-            placeholder="請輸入您的名字... | Please enter your name...",
-            elem_classes=["input-box"]
-        )
-        
-        chatbot = gr.Chatbot(
-            [],
-            elem_id="chatbot",
-            bubble_full_width=False,
-            avatar_images=("👨‍🎓", "👨‍🏫")
-        )
-        
-        msg = gr.Textbox(
-            label="發送訊息 | Send Message",
-            placeholder="請輸入您的問題... | Enter your question...",
-            show_label=False,
-            elem_classes=["input-box"]
-        )
-        
-        clear = gr.ClearButton([msg, chatbot], value="🗑️ 清除對話 | Clear Chat")
-        
-        msg.submit(respond, [msg, chatbot, name_input], [msg, chatbot])
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot(
+        height=600,
+        show_label=False,
+        container=True,
+    )
+    msg = gr.Textbox(
+        placeholder="輸入您的問題...",
+        show_label=False,
+    )
+    clear = gr.Button("清除對話")
 
-if __name__ == "__main__":
-    demo.launch()
+    def user(user_message, history):
+        return "", history + [[user_message, None]]
+
+    def bot(history):
+        history[-1][1] = ""
+        for response in process_message(history[-1][0], history[:-1]):
+            history[-1][1] = response
+            yield history
 
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, chatbot, chatbot
+    )
+    clear.click(lambda: None, None, chatbot, queue=False)
+
+# Launch the app
+if __name__ == "__main__":
+    demo.queue()
+    demo.launch()