Update app.py

app.py

@@ -8,7 +8,8 @@ from datetime import datetime
 # Groq API Configuration
 API_URL = "https://api.groq.com/openai/v1/chat/completions"
 API_KEY = os.getenv("GROQ_API_KEY")
-print(f"API Key Found: {'Yes' if API_KEY else 'No'}")
+
+print(f"🔑 API Key Found: {'Yes' if API_KEY else 'No'}")
 
 # In-memory chat history storage
 chat_history: List[Dict[str, str]] = []
@@ -16,18 +17,19 @@ 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"
+    
     if not message.strip():
-        return "Empty message"
+        return "❌ Empty message"
     
     try:
-        if not API_KEY:
-            return "No API Key found"
-
         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.strip(),
@@ -35,7 +37,10 @@ def groq_with_memory(message: str, topic: str = "general", summarize: bool = Fal
             "timestamp": datetime.now().isoformat()
         })
         
+        # 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:
@@ -44,7 +49,7 @@ def groq_with_memory(message: str, topic: str = "general", summarize: bool = Fal
             messages.append({"role": "user", "content": summary_prompt})
         
         payload = {
-            "model": "llama3-70b-8192",
+            "model": "llama-3.3-70b-versatile",
             "messages": messages,
             "max_tokens": 5000,
             "temperature": 0.7
@@ -63,86 +68,168 @@ def groq_with_memory(message: str, topic: str = "general", summarize: bool = Fal
                     "timestamp": datetime.now().isoformat()
                 })
                 return response_content
-            return f"No choices in response: {result}"
-        return f"HTTP {response.status_code}: {response.text}"
-        
+            return f"❌ No choices in response: {result}"
+        return f"❌ HTTP {response.status_code}: {response.text}"
+            
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"❌ Error: {str(e)}"
 
-def clear_history():
+def chat_with_groq(message: str, history: List[List[str]], topic: str = "general", summarize: bool = False) -> tuple:
+    """
+    Function for Gradio Chatbot
+    Returns: (updated_history, empty_string_for_textbox)
+    """
+    if not message.strip():
+        return history, ""
+    
+    # Get response from Groq
+    response = groq_with_memory(message, topic, summarize)
+    
+    # Add to gradio chat history format
+    history.append([message, response])
+    
+    return history, ""
+
+def clear_chat():
+    """Clear both chat histories"""
     chat_history.clear()
-    return [], "Chat history cleared"
+    return [], "✅ Chat history cleared"
 
 def view_history(topic: str = None):
+    """View chat history with optional topic filter"""
     if not chat_history:
-        return "No chat history available"
+        return "❌ No chat history available"
     
-    filtered = [msg for msg in chat_history if topic is None or msg["topic"] == topic]
+    filtered = [msg for msg in chat_history if topic is None or msg.get("topic") == topic]
     return "\n".join(
-        f"[{msg['timestamp']}] {msg['role'].capitalize()}: {msg['content']}"
+        f"[{msg.get('timestamp', 'N/A')}] {msg['role'].capitalize()}: {msg['content']}"
         for msg in filtered
     )
 
 def manual_curl_example():
+    """Generate example curl command"""
     if not API_KEY:
-        return "No API key available"
+        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"}}]}}'
+  -d '{{"model": "llama-3.3-70b-versatile", "messages": [{{"role": "user", "content": "Hello"}}], "max_tokens": 1000}}'
 """
 
-def process_message(message: str, topic: str, summarize: bool, chatbot):
-    if not API_KEY:
-        chatbot.append(("user", message))
-        chatbot.append(("assistant", "No API Key found."))
-        return chatbot, ""
-    
-    response = groq_with_memory(message, topic, summarize)
-    chatbot.append(("user", message))
-    chatbot.append(("assistant", response))
-    return chatbot, ""
-
 # Gradio Interface
-with gr.Blocks(title="Groq Chat UI") as demo:
-    gr.Markdown("## Groq Chat UI")
-
-    with gr.Tab("Chat"):
-        chatbot = gr.Chatbot(label="Conversation")
-        
-        with gr.Row():
-            msg_input = gr.Textbox(placeholder="Tulis pesan di sini...", label="Your Message", lines=2)
-            send_btn = gr.Button("Send", variant="primary")
-        
+with gr.Blocks(title="Groq Chat Debugger", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("## 💬 Groq API Chat Interface")
+    
+    with gr.Tab("💬 Chat"):
         with gr.Row():
-            topic_input = gr.Textbox(value="general", label="Topic")
-            summarize_cb = gr.Checkbox(label="Summarize conversation")
+            with gr.Column(scale=3):
+                # Main chatbot interface
+                chatbot = gr.Chatbot(
+                    label="Chat with Groq",
+                    height=500,
+                    show_copy_button=True
+                )
+                
+                with gr.Row():
+                    msg_input = gr.Textbox(
+                        label="Your Message",
+                        placeholder="Ketik pesan Anda di sini...",
+                        scale=4
+                    )
+                    send_btn = gr.Button("Send 📤", variant="primary", scale=1)
+                
+            with gr.Column(scale=1):
+                # Settings panel
+                gr.Markdown("### ⚙️ Settings")
+                topic_input = gr.Textbox(
+                    label="Topic", 
+                    value="general",
+                    placeholder="general"
+                )
+                summarize_cb = gr.Checkbox(
+                    label="📋 Summarize conversation",
+                    value=False
+                )
+                clear_btn = gr.Button("🗑️ Clear Chat", variant="stop")
+                
+                # Stats
+                gr.Markdown("### 📊 Stats")
+                stats_output = gr.Textbox(
+                    label="Chat Stats",
+                    value=f"API Status: {'✅ Ready' if API_KEY else '❌ No Key'}",
+                    interactive=False,
+                    lines=3
+                )
         
+        # Event handlers
         send_btn.click(
-            process_message,
-            inputs=[msg_input, topic_input, summarize_cb, chatbot],
+            chat_with_groq,
+            inputs=[msg_input, chatbot, topic_input, summarize_cb],
             outputs=[chatbot, msg_input]
         )
+        
+        msg_input.submit(  # Enter key support
+            chat_with_groq,
+            inputs=[msg_input, chatbot, topic_input, summarize_cb],
+            outputs=[chatbot, msg_input]
+        )
+        
+        clear_btn.click(
+            clear_chat,
+            outputs=[chatbot, stats_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)
+    with gr.Tab("📜 History"):
+        gr.Markdown("### View and manage your chat history")
+        with gr.Row():
+            history_topic = gr.Textbox(
+                label="Filter by Topic",
+                placeholder="Leave empty to see all"
+            )
+            view_btn = gr.Button("👀 View History", variant="primary")
+        
+        history_output = gr.Textbox(
+            label="Chat History", 
+            lines=15,
+            interactive=False
+        )
         
-        view_btn.click(view_history, inputs=[history_topic], outputs=[history_output])
-        clear_btn.click(clear_history, outputs=[chatbot, history_output])
+        view_btn.click(
+            view_history, 
+            inputs=[history_topic], 
+            outputs=[history_output]
+        )
     
-    with gr.Tab("API Tools"):
+    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
+        ### API Configuration
+        **Status:** {'✅ Connected' if API_KEY else '❌ No API Key Set'}  
+        **Endpoint:** `{API_URL}`  
+        **Available Models:** 
+        - `llama-3.3-70b-versatile` (Default)
+        - `llama3-8b-8192`
+        - `mixtral-8x7b-32768`
         """)
         
-        curl_btn = gr.Button("Generate cURL Example")
-        curl_output = gr.Code(label="cURL Command", language="shell")
+        curl_btn = gr.Button("📋 Generate cURL Example")
+        curl_output = gr.Code(
+            label="cURL Command", 
+            language="bash",
+            lines=8
+        )
         curl_btn.click(manual_curl_example, outputs=[curl_output])
+        
+        if not API_KEY:
+            gr.Markdown("""
+            ### ⚠️ Setup Instructions
+            1. Get your API key from [Groq Console](https://console.groq.com/)
+            2. Set environment variable: `export GROQ_API_KEY="your-key-here"`
+            3. Restart the application
+            """)
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(
+        server_name="0.0.0.0", 
+        server_port=7860,
+        share=False  # Set True if you want public link
+    )