Update app.py

app.py

@@ -7,168 +7,143 @@ from datetime import datetime
 
 # Groq API Configuration
 API_URL = "https://api.groq.com/openai/v1/chat/completions"
-API_KEY = os.getenv("GROQ_API_KEY")
+API_KEY = os.getenv("GROQ_API_KEY") or "your-api-key-here"  # Fallback for testing
 
-print(f"🔑 API Key Found: {'Yes' if API_KEY else 'No'}")
+def verify_api_key():
+    if not API_KEY or API_KEY == "your-api-key-here":
+        print("❌ No valid API key found")
+        print("Please set GROQ_API_KEY environment variable")
+        return False
+    return True
 
 # In-memory chat history storage
 chat_history: List[Dict[str, str]] = []
 
-def groq_with_memory(message: str, topic: str = "general", summarize: bool = False, max_history: int = 10) -> str:
-    """Groq API call with chat history and optional summarization"""
-    if not API_KEY:
-        return "❌ No API Key found"
+def groq_with_memory(message: str, topic: str = "general", summarize: bool = False) -> str:
+    """Groq API call with chat history"""
+    if not verify_api_key():
+        return "❌ API key not configured - please set GROQ_API_KEY environment variable"
     
     if not message.strip():
-        return "❌ Empty message"
+        return "❌ Please enter a message"
     
     try:
-        headers = {
-            "Authorization": f"Bearer {API_KEY}",
-            "Content-Type": "application/json"
-        }
-        
-        # Add current message to history
+        # Add to history
         chat_history.append({
             "role": "user",
-            "content": message.strip(),
+            "content": message,
             "topic": topic,
             "timestamp": datetime.now().isoformat()
         })
         
-        # Prepare messages for API call
+        # Prepare messages (last 10)
         messages = [{"role": msg["role"], "content": msg["content"]} 
-                   for msg in chat_history[-max_history:]]
+                   for msg in chat_history[-10:]]
         
         if summarize:
-            summary_prompt = f"Summarize this conversation about '{topic}':\n\n"
-            summary_prompt += "\n".join([f"{msg['role']}: {msg['content']}" 
-                                       for msg in chat_history if msg["topic"] == topic])
-            messages.append({"role": "user", "content": summary_prompt})
-        
-        payload = {
-            "model": "llama3-8b-8192",
-            "messages": messages,
-            "max_tokens": 512,
-            "temperature": 0.7
-        }
+            messages.append({
+                "role": "user",
+                "content": f"Summarize our conversation about {topic}"
+            })
         
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
+        response = requests.post(
+            API_URL,
+            headers={
+                "Authorization": f"Bearer {API_KEY}",
+                "Content-Type": "application/json"
+            },
+            json={
+                "model": "llama3-8b-8192",
+                "messages": messages,
+                "temperature": 0.7
+            },
+            timeout=30
+        )
         
         if response.status_code == 200:
-            result = response.json()
-            if "choices" in result and result["choices"]:
-                response_content = result["choices"][0]["message"]["content"]
-                chat_history.append({
-                    "role": "assistant",
-                    "content": response_content,
-                    "topic": topic,
-                    "timestamp": datetime.now().isoformat()
-                })
-                return response_content
-            return f"❌ No choices in response: {result}"
-        return f"❌ HTTP {response.status_code}: {response.text}"
-            
+            reply = response.json()["choices"][0]["message"]["content"]
+            chat_history.append({
+                "role": "assistant",
+                "content": reply,
+                "topic": topic,
+                "timestamp": datetime.now().isoformat()
+            })
+            return reply
+        return f"❌ Error {response.status_code}: {response.text}"
+    
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        return f"❌ Connection error: {str(e)}"
 
-def clear_history():
-    """Clear the chat history"""
-    chat_history.clear()
-    return "✅ Chat history cleared"
+def process_chat(message: str, chat_ui: list, topic: str) -> tuple:
+    """Process chat message and update UI"""
+    if not message.strip():
+        return "", chat_ui
+    reply = groq_with_memory(message, topic)
+    return "", chat_ui + [(message, reply)]
 
-def view_history(topic: str = None):
-    """View chat history with optional topic filter"""
+def view_history(topic: str = None) -> str:
+    """View raw chat history"""
     if not chat_history:
-        return "❌ No chat history available"
-    
-    filtered = [msg for msg in chat_history if topic is None or msg["topic"] == topic]
-    return "\n".join(
-        f"[{msg['timestamp']}] {msg['role'].capitalize()}: {msg['content']}"
-        for msg in filtered
-    )
+        return "No history yet"
+    filtered = [msg for msg in chat_history if not topic or msg["topic"] == topic]
+    return "\n".join(f"{msg['timestamp']} - {msg['role']}: {msg['content']}" for msg in filtered)
 
-def manual_curl_example():
-    """Generate example curl command"""
-    if not API_KEY:
-        return "❌ No API key available"
-    return f"""curl -X POST "{API_URL}" \\
-  -H "Authorization: Bearer {API_KEY}" \\
-  -H "Content-Type: application/json" \\
-  -d '{{"model": "llama3-8b-8192", "messages": [{{"role": "user", "content": "Hello"}}]}}'
-"""
-
-def process_chat(message: str, chat_history: list, topic: str, summarize: bool):
-    """Process chat message and update history"""
-    response = groq_with_memory(message, topic, summarize)
-    chat_history.append((message, response))
-    return "", chat_history
+def clear_all_history():
+    """Clear all chat history"""
+    chat_history.clear()
+    return "✅ History cleared", []
 
-# Gradio Interface with Chatbox Style
-with gr.Blocks(title="Groq Chat Interface") as demo:
+# Gradio Interface
+with gr.Blocks(title="Groq Chat", theme=gr.themes.Soft()) as demo:
     with gr.Tabs():
-        with gr.TabItem("💬 Chat"):
-            chatbot = gr.Chatbot(height=500)
+        # Chat Tab
+        with gr.Tab("💬 Chat"):
+            chatbot = gr.Chatbot(height=400)
             with gr.Row():
-                msg_input = gr.Textbox(
-                    placeholder="Type your message here...",
-                    show_label=False,
-                    container=False,
-                    scale=7
-                )
-                with gr.Column(scale=1):
-                    send_btn = gr.Button("Send", variant="primary")
-                    clear_btn = gr.Button("Clear", variant="secondary")
+                message = gr.Textbox(placeholder="Type your message...", scale=7)
+                send = gr.Button("Send", variant="primary")
             
-            with gr.Accordion("⚙️ Settings", open=False):
-                topic_input = gr.Textbox(
-                    label="Conversation Topic",
-                    value="general"
-                )
-                summarize_cb = gr.Checkbox(
-                    label="Summarize conversation",
-                    value=False
-                )
+            with gr.Accordion("Settings", open=False):
+                topic = gr.Textbox(label="Conversation Topic", value="general")
+                gr.Markdown("**Note:** History is preserved until you clear it")
             
-            # Chat interaction
-            msg_input.submit(
-                process_chat,
-                [msg_input, chatbot, topic_input, summarize_cb],
-                [msg_input, chatbot]
-            )
-            send_btn.click(
-                process_chat,
-                [msg_input, chatbot, topic_input, summarize_cb],
-                [msg_input, chatbot]
-            )
-            clear_btn.click(lambda: [], None, chatbot)
+            clear_chat = gr.Button("Clear Chat UI")
+            clear_history = gr.Button("⚠️ Clear ALL History", variant="stop")
+            status = gr.Textbox(label="Status")
         
-        with gr.TabItem("📜 History"):
-            history_topic = gr.Textbox(label="Filter by Topic")
+        # History Tab
+        with gr.Tab("📜 History"):
+            history_topic = gr.Textbox(label="Filter by topic")
             view_btn = gr.Button("View History")
-            clear_history_btn = gr.Button("Clear All History", variant="stop")
-            history_output = gr.Textbox(label="Raw History", lines=15)
-            
-            view_btn.click(
-                view_history,
-                inputs=[history_topic],
-                outputs=[history_output]
-            )
-            clear_history_btn.click(
-                clear_history,
-                outputs=[history_output]
-            )
+            raw_history = gr.Textbox(label="Raw History", lines=15)
+            view_btn.click(view_history, history_topic, raw_history)
         
-        with gr.TabItem("🔧 API Tools"):
+        # API Info Tab
+        with gr.Tab("ℹ️ API Info"):
             gr.Markdown(f"""
-            **API Status:** {'✅ Ready' if API_KEY else '❌ No API Key'}
+            **API Status:** {'✅ Ready' if verify_api_key() else '❌ Not Configured'}
             **Endpoint:** `{API_URL}`
-            **Models:** `llama3-8b-8192`, `llama3-70b-8192`, `mixtral-8x7b-32768`
-            """)
+            **Models Available:**
+            - `llama3-8b-8192`
+            - `llama3-70b-8192`
+            - `mixtral-8x7b-32768`
             
-            curl_btn = gr.Button("Generate cURL Example")
-            curl_output = gr.Code(label="cURL Command", language="shell")
-            curl_btn.click(manual_curl_example, outputs=[curl_output])
+            **How to set up:**
+            1. Get API key from [Groq Cloud](https://console.groq.com/)
+            2. Set environment variable:
+               ```bash
+               export GROQ_API_KEY="your-api-key-here"
+               ```
+            """)
+    
+    # Event handlers
+    message.submit(process_chat, [message, chatbot, topic], [message, chatbot])
+    send.click(process_chat, [message, chatbot, topic], [message, chatbot])
+    clear_chat.click(lambda: [], None, chatbot)
+    clear_history.click(clear_all_history, outputs=[status, chatbot])
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    if verify_api_key():
+        demo.launch(server_name="0.0.0.0", server_port=7860)
+    else:
+        print("Cannot start app - no valid API key found")