Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 import requests
 import os
@@ -7,143 +8,154 @@ from datetime import datetime
 
 # Groq API Configuration
 API_URL = "https://api.groq.com/openai/v1/chat/completions"
-API_KEY = os.getenv("GROQ_API_KEY") or "your-api-key-here"  # Fallback for testing
+API_KEY = os.getenv("GROQ_API_KEY")
 
-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
+print(f"🔑 API Key Found: {'Yes' if API_KEY else 'No'}")
 
 # In-memory chat history storage
 chat_history: List[Dict[str, str]] = []
 
-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"
+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"
     
     if not message.strip():
-        return "❌ Please enter a message"
+        return "❌ Empty message"
     
     try:
-        # Add to history
+        headers = {
+            "Authorization": f"Bearer {API_KEY}",
+            "Content-Type": "application/json"
+        }
+        
+        # Add current message to history with timestamp and topic
         chat_history.append({
             "role": "user",
-            "content": message,
+            "content": message.strip(),
             "topic": topic,
             "timestamp": datetime.now().isoformat()
         })
         
-        # Prepare messages (last 10)
-        messages = [{"role": msg["role"], "content": msg["content"]} 
-                   for msg in chat_history[-10:]]
+        # Limit history to max_history messages
+        recent_history = chat_history[-max_history:]
+        
+        # Prepare messages for API call
+        messages = [{"role": msg["role"], "content": msg["content"]} for msg in recent_history]
         
         if summarize:
-            messages.append({
-                "role": "user",
-                "content": f"Summarize our conversation about {topic}"
-            })
+            summary_prompt = f"Summarize this conversation about '{topic}':\n\n"
+            summary_prompt += "\n".join([f"{msg['role']}: {msg['content']}" for msg in recent_history])
+            messages.append({"role": "user", "content": summary_prompt})
         
-        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
-        )
+        payload = {
+            "model": "deepseek-r1-distill-llama-70b",
+            "messages": messages,
+            "max_tokens": 512,
+            "temperature": 0.7
+        }
+        
+        response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
         
         if response.status_code == 200:
-            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}"
-    
+            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}"
+            
     except Exception as e:
-        return f"❌ Connection error: {str(e)}"
+        return f"❌ Error: {str(e)}"
 
-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 clear_history():
+    """Clear the chat history"""
+    chat_history.clear()
+    return "✅ Chat history cleared"
 
-def view_history(topic: str = None) -> str:
-    """View raw chat history"""
+def view_history(topic: str = None):
+    """View chat history with optional topic filter"""
     if not chat_history:
-        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)
+        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
+    )
 
-def clear_all_history():
-    """Clear all chat history"""
-    chat_history.clear()
-    return "✅ History cleared", []
+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_message(message: str, topic: str, summarize: bool):
+    """Process message and clear input"""
+    response = groq_with_memory(message, topic, summarize)
+    return response, ""
 
 # Gradio Interface
-with gr.Blocks(title="Groq Chat", theme=gr.themes.Soft()) as demo:
-    with gr.Tabs():
-        # Chat Tab
-        with gr.Tab("💬 Chat"):
-            chatbot = gr.Chatbot(height=400)
-            with gr.Row():
-                message = gr.Textbox(placeholder="Type your message...", scale=7)
-                send = gr.Button("Send", variant="primary")
-            
-            with gr.Accordion("Settings", open=False):
-                topic = gr.Textbox(label="Conversation Topic", value="general")
-                gr.Markdown("**Note:** History is preserved until you clear it")
-            
-            clear_chat = gr.Button("Clear Chat UI")
-            clear_history = gr.Button("⚠️ Clear ALL History", variant="stop")
-            status = gr.Textbox(label="Status")
+with gr.Blocks(title="Groq Chat Debugger") as demo:
+    gr.Markdown("## 💬 Groq API Chat Interface")
+    
+    with gr.Tab("Chat"):
+        with gr.Row():
+            with gr.Column():
+                msg_input = gr.Textbox(
+                    label="Your Message", 
+                    value="Hello, how can you help me today?"
+                )
+                topic_input = gr.Textbox(
+                    label="Topic", 
+                    value="general"
+                )
+                summarize_cb = gr.Checkbox(label="Summarize conversation")
+                send_btn = gr.Button("Send", variant="primary")
+                
+            with gr.Column():
+                response_output = gr.Textbox(
+                    label="Response", 
+                    lines=10,
+                    interactive=False
+                )
         
-        # History Tab
-        with gr.Tab("📜 History"):
-            history_topic = gr.Textbox(label="Filter by topic")
-            view_btn = gr.Button("View History")
-            raw_history = gr.Textbox(label="Raw History", lines=15)
-            view_btn.click(view_history, history_topic, raw_history)
+        send_btn.click(
+            process_message,
+            inputs=[msg_input, topic_input, summarize_cb],
+            outputs=[response_output, msg_input]
+        )
+    
+    with gr.Tab("History"):
+        history_topic = gr.Textbox(label="Filter by Topic")
+        view_btn = gr.Button("View History")
+        clear_btn = gr.Button("Clear History", variant="stop")
+        history_output = gr.Textbox(label="Chat History", lines=10)
         
-        # API Info Tab
-        with gr.Tab("ℹ️ API Info"):
-            gr.Markdown(f"""
-            **API Status:** {'✅ Ready' if verify_api_key() else '❌ Not Configured'}
-            **Endpoint:** `{API_URL}`
-            **Models Available:**
-            - `llama3-8b-8192`
-            - `llama3-70b-8192`
-            - `mixtral-8x7b-32768`
-            
-            **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"
-               ```
-            """)
+        view_btn.click(view_history, inputs=[history_topic], outputs=[history_output])
+        clear_btn.click(clear_history, outputs=[history_output])
     
-    # 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])
+    with gr.Tab("API Tools"):
+        gr.Markdown(f"""
+        **API Status:** {'✅ Ready' if API_KEY else '❌ No API Key'}
+        **Endpoint:** `{API_URL}`
+        **Models:** `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])
 
 if __name__ == "__main__":
-    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")
+    demo.launch(server_name="0.0.0.0", server_port=7860)