Update app.py

chat. history. journal

app.py

@@ -9,19 +9,17 @@ from datetime import datetime
 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'}")
-
 # 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"""
+def groq_with_memory(message: str, topic: str = "general") -> tuple:
+    """Groq API call with chat history"""
     
     if not API_KEY:
-        return "❌ No API Key found"
+        return "❌ No API Key found", ""
     
     if not message.strip():
-        return "❌ Empty message"
+        return "❌ Empty message", ""
     
     try:
         headers = {
@@ -29,35 +27,28 @@ def groq_with_memory(message: str, topic: str = "general", summarize: bool = Fal
             "Content-Type": "application/json"
         }
         
-        # Add current message to history with timestamp and topic
+        # Add current message to history
         chat_history.append({
             "role": "user",
             "content": message.strip(),
             "topic": topic,
-            "timestamp": datetime.now().isoformat()
+            "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M")
         })
         
-        # Limit history to max_history messages
-        recent_history = chat_history[-max_history:]
-        
-        # Prepare messages for API call
+        # Prepare messages for API call (last 10 messages)
+        recent_history = chat_history[-10:]
         messages = [{"role": msg["role"], "content": msg["content"]} for msg in recent_history]
         
-        if summarize:
-            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 = {
             "model": "gemma2-9b-it",
             "messages": messages,
-            "max_tokens":2000,
-            "temperature":0.9
+            "max_tokens": 2000,
+            "temperature": 0.7
         }
         
         response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
         
-        if response.status_code ==200:
+        if response.status_code == 200:
             result = response.json()
             if "choices" in result and result["choices"]:
                 response_content = result["choices"][0]["message"]["content"]
@@ -65,129 +56,268 @@ def groq_with_memory(message: str, topic: str = "general", summarize: bool = Fal
                     "role": "assistant",
                     "content": response_content,
                     "topic": topic,
-                    "timestamp": datetime.now().isoformat()
+                    "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M")
                 })
-                return response_content
-            return f"❌ No choices in response: {result}"
-        return f"❌ HTTP {response.status_code}: {response.text}"
+                return response_content, ""  # Clear input after send
+            return f"❌ No response: {result}", ""
+        return f"❌ HTTP {response.status_code}: {response.text}", ""
             
     except Exception as e:
-        return f"❌ Error: {str(e)}"
-
-def clear_history():
-    """Clear the chat history"""
-    global chat_history
-    chat_history.clear()
-    return "✅ Chat history cleared", ""
+        return f"❌ Error: {str(e)}", ""
 
-def view_history(topic: str = None):
-    """View chat history with optional topic filter"""
+def get_chat_summary(topic_filter: str = None) -> str:
+    """Generate conversation summary"""
     if not chat_history:
-        return "❌ No chat history available"
+        return "❌ No chat history to summarize"
     
-    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']}"
-        for msg in filtered
-    )
-
-# Gradio Interface
-with gr.Blocks(title="Groq Chat Interface", theme=gr.themes.Soft(), css="""
-    .gradio-container {
-        max-width:100% !important;
-        padding-left:0px !important;
-        padding-right:0px !important;
-    }
-    .main {
-        max-width:100% !important;
-        padding:0px !important;
-    }
-    .block {
-        max-width:100% !important;
-    }
-""") as demo:
-    gr.Markdown("## 💬 Groq API Chat Interface")
-    
-    with gr.Tab("💬 Chat"):
-        # Response area at top
-        response_output = gr.Textbox(
-            label="🤖 AI Response", 
-            lines=15,
-            interactive=False,
-            placeholder="AI responses will appear here...",
-            show_copy_button=True
-        )
+    filtered = [msg for msg in chat_history if not topic_filter or msg.get("topic") == topic_filter]
+    
+    if not filtered:
+        return f"❌ No messages found for topic: {topic_filter}"
+    
+    # Group by topic
+    topics = {}
+    for msg in filtered:
+        topic = msg.get("topic", "general")
+        if topic not in topics:
+            topics[topic] = []
+        topics[topic].append(msg)
+    
+    summary = "📋 **Chat Summary**\n\n"
+    for topic, messages in topics.items():
+        user_msgs = [m for m in messages if m["role"] == "user"]
+        ai_msgs = [m for m in messages if m["role"] == "assistant"]
         
-        gr.Markdown("---")
+        summary += f"**🏷️ Topic: {topic}**\n"
+        summary += f"- Messages: {len(user_msgs)} user, {len(ai_msgs)} AI\n"
+        summary += f"- Time span: {messages[0]['timestamp']} - {messages[-1]['timestamp']}\n"
         
-        # Input area at bottom
-        with gr.Row():
-            with gr.Column(scale=4):
-                msg_input = gr.Textbox(
-                    label="✍️ Your Message", 
-                    placeholder="Type your message here...",
-                    lines=3,
-                    max_lines=10
-                )
-                    
-            with gr.Column(scale=1):
-                topic_input = gr.Textbox(
-                    label="🏷️ Topic", 
-                    value="general",
-                    placeholder="conversation topic"
-                )
-                summarize_cb = gr.Checkbox(
-                    label="📋 Summarize conversation",
-                    value=False
-                )
-                send_btn = gr.Button("📤 Send Message", variant="primary")
+        if user_msgs:
+            summary += f"- Key topics discussed: {', '.join(msg['content'][:50] + '...' for msg in user_msgs[:3])}\n"
+        summary += "\n"
+    
+    return summary
+
+def get_full_history(topic_filter: str = None) -> str:
+    """Get full chat history with optional topic filter"""
+    if not chat_history:
+        return "❌ No chat history available"
+    
+    filtered = [msg for msg in chat_history if not topic_filter or msg.get("topic") == topic_filter]
+    
+    if not filtered:
+        return f"❌ No messages found for topic: {topic_filter}"
+    
+    history_text = f"📚 **Chat History** ({len(filtered)} messages)\n\n"
+    
+    current_topic = None
+    for msg in filtered:
+        # Add topic header when topic changes
+        if msg.get("topic") != current_topic:
+            current_topic = msg.get("topic")
+            history_text += f"\n**🏷️ Topic: {current_topic}**\n"
+            history_text += "---\n"
         
-        with gr.Row():
-            clear_btn = gr.Button("🗑️ Clear Response", variant="secondary")
-            history_btn = gr.Button("📚 View History", variant="secondary")
-            history_output = gr.Textbox(
-                label="📚 Chat History", 
-                lines=15,
+        role_icon = "👤" if msg["role"] == "user" else "🤖"
+        history_text += f"{role_icon} **{msg['timestamp']}**\n"
+        history_text += f"{msg['content']}\n\n"
+    
+    return history_text
+
+def clear_all_history():
+    """Clear all chat history"""
+    global chat_history
+    chat_history.clear()
+    return "✅ All chat history cleared", "", ""
+
+def get_topics_list() -> List[str]:
+    """Get list of unique topics"""
+    topics = list(set(msg.get("topic", "general") for msg in chat_history))
+    return ["All Topics"] + sorted(topics)
+
+# Custom CSS for better UI
+custom_css = """
+.gradio-container {
+    max-width: 1200px !important;
+    margin: 0 auto !important;
+}
+
+.main-chat {
+    border: 1px solid #e0e0e0;
+    border-radius: 12px;
+    padding: 20px;
+    background: #fafafa;
+}
+
+.input-area {
+    background: white;
+    border-radius: 8px;
+    padding: 15px;
+    margin-top: 10px;
+}
+
+.history-area {
+    background: white;
+    border-radius: 8px;
+    padding: 15px;
+    margin-top: 10px;
+}
+
+.tab-nav {
+    background: #f5f5f5;
+    border-radius: 8px 8px 0 0;
+}
+
+/* Copy button styling */
+.copy-button {
+    background: #4CAF50 !important;
+    color: white !important;
+    border-radius: 4px !important;
+}
+"""
+
+# Main Gradio Interface
+with gr.Blocks(
+    title="🤖 AI Journal Chat", 
+    theme=gr.themes.Soft(),
+    css=custom_css
+) as demo:
+    
+    gr.Markdown("# 📝 AI Journal Chat Interface")
+    gr.Markdown("*Write, chat, and keep track of your thoughts with AI assistance*")
+    
+    with gr.Tabs() as tabs:
+        # MAIN CHAT TAB
+        with gr.Tab("💬 Chat", elem_classes="main-chat"):
+            # AI Response Area (Top)
+            ai_response = gr.Textbox(
+                label="🤖 AI Response",
+                lines=12,
+                max_lines=20,
                 interactive=False,
-                placeholder="Chat history will appear here..."
+                placeholder="AI responses will appear here...",
+                show_copy_button=True,
+                elem_classes="response-area"
             )
+            
+            # Input Area (Bottom)
+            with gr.Group(elem_classes="input-area"):
+                with gr.Row():
+                    user_input = gr.Textbox(
+                        label="✍️ Your Message",
+                        placeholder="Type your thoughts, questions, or journal entry here...",
+                        lines=4,
+                        max_lines=10,
+                        scale=3
+                    )
+                    with gr.Column(scale=1):
+                        topic_input = gr.Textbox(
+                            label="🏷️ Topic",
+                            value="journal",
+                            placeholder="e.g., work, personal, ideas"
+                        )
+                        send_btn = gr.Button("📤 Send", variant="primary", size="lg")
+                
+                with gr.Row():
+                    clear_response_btn = gr.Button("🗑️ Clear Response", variant="secondary")
+                    new_topic_btn = gr.Button("🆕 New Topic", variant="secondary")
         
-        # Event handlers - Fixed indentation
-        send_btn.click(
-            lambda message, topic, summarize: groq_with_memory(message, topic, summarize),
-            inputs=[msg_input, topic_input, summarize_cb],
-            outputs=[response_output]
-        )
-        
-        msg_input.submit(
-            lambda message, topic, summarize: groq_with_memory(message, topic, summarize),
-            inputs=[msg_input, topic_input, summarize_cb],
-            outputs=[response_output]
-        )
-        
-        clear_btn.click(
-            lambda: ("", ""),
-            outputs=[response_output, msg_input]
-        )
-        
-        history_btn.click(
-            lambda topic: view_history(topic),
-            inputs=[topic_input],
-            outputs=[history_output]
-        )
-  
-
-# Launch message
-print("🚀 Starting Groq Chat Interface...")
+        # HISTORY TAB
+        with gr.Tab("📚 Chat History"):
+            with gr.Group(elem_classes="history-area"):
+                with gr.Row():
+                    topic_filter = gr.Dropdown(
+                        label="🔍 Filter by Topic",
+                        choices=["All Topics"],
+                        value="All Topics",
+                        interactive=True
+                    )
+                    refresh_topics_btn = gr.Button("🔄 Refresh Topics", variant="secondary")
+                
+                with gr.Row():
+                    show_history_btn = gr.Button("📖 Show Full History", variant="primary")
+                    show_summary_btn = gr.Button("📋 Show Summary", variant="secondary")
+                    clear_history_btn = gr.Button("🗑️ Clear All History", variant="stop")
+                
+                history_display = gr.Textbox(
+                    label="📚 History & Summary",
+                    lines=20,
+                    max_lines=30,
+                    interactive=False,
+                    show_copy_button=True,
+                    placeholder="Chat history and summaries will appear here..."
+                )
+    
+    # Event Handlers
+    def send_message(message, topic):
+        response, cleared_input = groq_with_memory(message, topic)
+        return response, cleared_input
+    
+    def refresh_topic_choices():
+        return gr.Dropdown(choices=get_topics_list())
+    
+    def filter_and_show_history(topic_filter):
+        filter_topic = None if topic_filter == "All Topics" else topic_filter
+        return get_full_history(filter_topic)
+    
+    def filter_and_show_summary(topic_filter):
+        filter_topic = None if topic_filter == "All Topics" else topic_filter
+        return get_chat_summary(filter_topic)
+    
+    # Button Events
+    send_btn.click(
+        send_message,
+        inputs=[user_input, topic_input],
+        outputs=[ai_response, user_input]
+    )
+    
+    user_input.submit(
+        send_message,
+        inputs=[user_input, topic_input],
+        outputs=[ai_response, user_input]
+    )
+    
+    clear_response_btn.click(
+        lambda: "",
+        outputs=[ai_response]
+    )
+    
+    new_topic_btn.click(
+        lambda: "",
+        outputs=[topic_input]
+    )
+    
+    refresh_topics_btn.click(
+        refresh_topic_choices,
+        outputs=[topic_filter]
+    )
+    
+    show_history_btn.click(
+        filter_and_show_history,
+        inputs=[topic_filter],
+        outputs=[history_display]
+    )
+    
+    show_summary_btn.click(
+        filter_and_show_summary,
+        inputs=[topic_filter],
+        outputs=[history_display]
+    )
+    
+    clear_history_btn.click(
+        clear_all_history,
+        outputs=[ai_response, user_input, history_display]
+    )
+
+# Launch configuration
+print("🚀 Starting AI Journal Chat Interface...")
 print(f"🌐 Access at: http://localhost:7860")
-if API_KEY:
-    print("✅ API Key detected - Ready to chat!")
-else:
-    print("⚠️ No API Key found - Please set GROQ_API_KEY environment variable")
+print(f"🔑 API Key: {'✅ Found' if API_KEY else '❌ Missing'}")
 
 if __name__ == "__main__":
     demo.launch(
-        server_name="0.0.0.0", 
+        server_name="0.0.0.0",
         server_port=7860,
-        show_error=True
-    )
+        show_error=True,
+        share=False
+    )