Update app.py

app.py

@@ -37,19 +37,15 @@ def groq_with_memory(message: str, topic: str = "general", summarize: bool = Fal
             "timestamp": datetime.now().isoformat()
         })
         
-        # Limit history to max_history messages to avoid token overflow
+        # Limit history to max_history messages
         recent_history = chat_history[-max_history:]
         
-        # Prepare messages for API call, excluding timestamp and topic
+        # Prepare messages for API call
         messages = [{"role": msg["role"], "content": msg["content"]} for msg in recent_history]
         
         if summarize:
-            # Create a prompt to summarize the conversation
-            summary_prompt = f"Summarize the following conversation history for the topic '{topic}':\n\n"
-            for msg in recent_history:
-                if msg["topic"] == topic:
-                    summary_prompt += f"{msg['role'].capitalize()}: {msg['content']}\n"
-            summary_prompt += "\nProvide a concise summary of the key points discussed."
+            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})
         
         payload = {
@@ -59,25 +55,12 @@ def groq_with_memory(message: str, topic: str = "general", summarize: bool = Fal
             "temperature": 0.7
         }
         
-        print(f"🚀 Sending request to: {API_URL}")
-        print(f"📝 Payload: {json.dumps(payload, indent=2)}")
-        
-        response = requests.post(
-            API_URL,
-            headers=headers,
-            json=payload,
-            timeout=30
-        )
-        
-        print(f"📊 Status Code: {response.status_code}")
-        print(f"📋 Response Headers: {dict(response.headers)}")
-        print(f"📄 Response Text (first 300 chars): {response.text[:300]}")
+        response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
         
         if response.status_code == 200:
             result = response.json()
-            if "choices" in result and len(result["choices"]) > 0:
+            if "choices" in result and result["choices"]:
                 response_content = result["choices"][0]["message"]["content"]
-                # Add assistant's response to history
                 chat_history.append({
                     "role": "assistant",
                     "content": response_content,
@@ -85,19 +68,11 @@ def groq_with_memory(message: str, topic: str = "general", summarize: bool = Fal
                     "timestamp": datetime.now().isoformat()
                 })
                 return response_content
-            else:
-                return f"❌ No choices in response: {json.dumps(result, indent=2)}"
-        else:
-            return f"❌ HTTP {response.status_code}: {response.text}"
+            return f"❌ No choices in response: {result}"
+        return f"❌ HTTP {response.status_code}: {response.text}"
             
-    except requests.exceptions.Timeout:
-        return "❌ Request timeout (30s)"
-    except requests.exceptions.ConnectionError:
-        return "❌ Connection error - cannot reach Groq API"
-    except json.JSONDecodeError:
-        return f"❌ Invalid JSON response: {response.text}"
     except Exception as e:
-        return f"❌ Unexpected error: {str(e)}"
+        return f"❌ Error: {str(e)}"
 
 def clear_history():
     """Clear the chat history"""
@@ -105,127 +80,81 @@ def clear_history():
     return "✅ Chat history cleared"
 
 def view_history(topic: str = None):
-    """View the chat history, optionally filtered by topic"""
+    """View chat history with optional topic filter"""
     if not chat_history:
         return "❌ No chat history available"
     
-    output = "📜 Chat History:\n\n"
-    for msg in chat_history:
-        if topic is None or msg["topic"] == topic:
-            output += f"[{msg['timestamp']}] {msg['role'].capitalize()} ({msg['topic']}): {msg['content']}\n"
-    return output
+    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 manual_curl_example():
-    """Generate curl command for manual testing"""
+    """Generate example curl command"""
     if not API_KEY:
-        return "❌ No API key to generate curl"
-    
-    curl_cmd = f'''curl -X POST "{API_URL}" \\
+        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, test message"}}
-    ],
-    "max_tokens": 100
-  }}'
-'''
-    return curl_cmd
+  -d '{{"model": "llama3-8b-8192", "messages": [{{"role": "user", "content": "Hello"}}]}}'
+"""
 
-# Create Gradio interface
-with gr.Blocks(title="Groq Debug with Memory") as demo:
-    gr.Markdown("# 🔍 Groq API Debug Tool with Memory")
+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 Debugger") as demo:
+    gr.Markdown("## 💬 Groq API Chat Interface")
     
-    with gr.Tab("Chat with Memory"):
-        gr.Markdown("Chat with Groq, maintain history, and summarize by topic")
-        
-        chat_input = gr.Textbox(
-            label="Message",
-            placeholder="Enter your message",
-            value="Hello, how can you help me today?"
-        )
-        topic_input = gr.Textbox(
-            label="Topic",
-            placeholder="Enter topic (e.g., 'coding', 'general')",
-            value="general"
-        )
-        summarize_checkbox = gr.Checkbox(
-            label="Summarize conversation for this topic",
-            value=False
-        )
-        chat_button = gr.Button("Send Message", variant="primary")
-        chat_output = gr.Textbox(
-            label="Response",
-            lines=10,
-            max_lines=20
-        )
-        
-        chat_button.click(
-            groq_with_memory,
-            inputs=[chat_input, topic_input, summarize_checkbox],
-            outputs=[chat_output]
+    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
+                )
+        
+        send_btn.click(
+            process_message,
+            inputs=[msg_input, topic_input, summarize_cb],
+            outputs=[response_output, msg_input]
         )
     
-    with gr.Tab("View History"):
-        gr.Markdown("View or clear chat history")
-        
-        history_topic = gr.Textbox(
-            label="Filter by Topic (optional)",
-            placeholder="Enter topic to filter (leave blank for all)"
-        )
-        history_button = gr.Button("View History")
-        clear_button = gr.Button("Clear History", variant="secondary")
-        history_output = gr.Textbox(
-            label="Chat History",
-            lines=10,
-            max_lines=20
-        )
-        
-        history_button.click(
-            view_history,
-            inputs=[history_topic],
-            outputs=[history_output]
-        )
-        clear_button.click(
-            clear_history,
-            outputs=[history_output]
-        )
-    
-    with gr.Tab("Manual Test"):
-        gr.Markdown("Copy this curl command and run it in terminal")
-        
-        curl_button = gr.Button("Generate Curl Command")
-        curl_output = gr.Code(
-            label="Curl Command",
-            language="shell"
-        )
-        
-        curl_button.click(
-            manual_curl_example,
-            outputs=[curl_output]
-        )
+    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)
+        
+        view_btn.click(view_history, inputs=[history_topic], outputs=[history_output])
+        clear_btn.click(clear_history, outputs=[history_output])
     
-    with gr.Tab("API Info"):
+    with gr.Tab("API Tools"):
         gr.Markdown(f"""
-        **Debug Information:**
-        - API URL: `{API_URL}`
-        - API Key Status: {'✅ Found' if API_KEY else '❌ Missing'}
-        - API Key Preview: `{API_KEY[:10] + '...' if API_KEY else 'None'}`
-        
-        **Common Issues:**
-        1. **Wrong URL** - Make sure using correct Groq endpoint
-        2. **Invalid Model** - Check if model name is correct
-        3. **Malformed Request** - Check JSON structure
-        4. **Rate Limiting** - Wait between requests
-        5. **Network Issues** - Check HF Spaces connectivity
-        
-        **Expected Models:**
-        - `llama3-8b-8192`
-        - `llama3-70b-8192`
-        - `mixtral-8x7b-32768`
-        - `gemma-7b-it`
+        **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="bash")
+        curl_btn.click(manual_curl_example, outputs=[curl_output])
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)