Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import requests
 import os
 import json
+from typing import List, Dict
+from datetime import datetime
 
 # Groq API Configuration
 API_URL = "https://api.groq.com/openai/v1/chat/completions"
@@ -9,8 +11,11 @@ API_KEY = os.getenv("GROQ_API_KEY")
 
 print(f"🔑 API Key Found: {'Yes' if API_KEY else 'No'}")
 
-def simple_groq_test(message: str) -> str:
-    """Simple Groq API test without chat history"""
+# 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"
@@ -24,11 +29,32 @@ def simple_groq_test(message: str) -> str:
             "Content-Type": "application/json"
         }
         
+        # Add current message to history with timestamp and topic
+        chat_history.append({
+            "role": "user",
+            "content": message.strip(),
+            "topic": topic,
+            "timestamp": datetime.now().isoformat()
+        })
+        
+        # Limit history to max_history messages to avoid token overflow
+        recent_history = chat_history[-max_history:]
+        
+        # Prepare messages for API call
+        messages = recent_history.copy()
+        
+        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."
+            messages.append({"role": "user", "content": summary_prompt})
+        
         payload = {
             "model": "llama3-8b-8192",
-            "messages": [
-                {"role": "user", "content": message.strip()}
-            ],
+            "messages": messages,
             "max_tokens": 512,
             "temperature": 0.7
         }
@@ -50,7 +76,15 @@ def simple_groq_test(message: str) -> str:
         if response.status_code == 200:
             result = response.json()
             if "choices" in result and len(result["choices"]) > 0:
-                return result["choices"][0]["message"]["content"]
+                response_content = result["choices"][0]["message"]["content"]
+                # Add assistant's response to history
+                chat_history.append({
+                    "role": "assistant",
+                    "content": response_content,
+                    "topic": topic,
+                    "timestamp": datetime.now().isoformat()
+                })
+                return response_content
             else:
                 return f"❌ No choices in response: {json.dumps(result, indent=2)}"
         else:
@@ -65,6 +99,22 @@ def simple_groq_test(message: str) -> str:
     except Exception as e:
         return f"❌ Unexpected error: {str(e)}"
 
+def clear_history():
+    """Clear the chat history"""
+    chat_history.clear()
+    return "✅ Chat history cleared"
+
+def view_history(topic: str = None):
+    """View the chat history, optionally filtered by topic"""
+    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
+
 def manual_curl_example():
     """Generate curl command for manual testing"""
     if not API_KEY:
@@ -83,29 +133,63 @@ def manual_curl_example():
 '''
     return curl_cmd
 
-# Create simple interface
-with gr.Blocks(title="Groq Debug") as demo:
-    gr.Markdown("# 🔍 Groq API Debug Tool")
+# Create Gradio interface
+with gr.Blocks(title="Groq Debug with Memory") as demo:
+    gr.Markdown("# 🔍 Groq API Debug Tool with Memory")
     
-    with gr.Tab("Simple Test"):
-        gr.Markdown("Test single message without chat history")
+    with gr.Tab("Chat with Memory"):
+        gr.Markdown("Chat with Groq, maintain history, and summarize by topic")
         
-        test_input = gr.Textbox(
-            label="Test Message",
-            placeholder="Enter a simple test message",
-            value="Hello, how are you?"
+        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
         )
-        test_button = gr.Button("Test API", variant="primary")
-        test_output = gr.Textbox(
+        chat_button = gr.Button("Send Message", variant="primary")
+        chat_output = gr.Textbox(
             label="Response",
             lines=10,
             max_lines=20
         )
         
-        test_button.click(
-            simple_groq_test,
-            inputs=[test_input],
-            outputs=[test_output]
+        chat_button.click(
+            groq_with_memory,
+            inputs=[chat_input, topic_input, summarize_checkbox],
+            outputs=[chat_output]
+        )
+    
+    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"):
@@ -138,7 +222,7 @@ with gr.Blocks(title="Groq Debug") as demo:
         
         **Expected Models:**
         - `llama3-8b-8192`
-        - `llama3-70b-8192` 
+        - `llama3-70b-8192`
         - `mixtral-8x7b-32768`
         - `gemma-7b-it`
         """)