Update app.py

app.py

@@ -21,6 +21,11 @@ import time
 import matplotlib.pyplot as plt
 from pathlib import Path
 import plotly.express as px
+from groq import Groq
+import streamlit.components.v1 as components
+import html
+from streamlit_chat import message
+import uuid
 
 # Set page config
 st.set_page_config(
@@ -137,6 +142,13 @@ def analyze_image(image, analysis_types):
 
 def display_results(annotated_img, labels, objects, text):
     """Display analysis results in a clean format"""
+    # Store results in session state for chatbot context
+    st.session_state.analysis_results = {
+        "labels": labels,
+        "objects": objects,
+        "text": text
+    }
+    
     col1, col2 = st.columns([3, 2])
     
     with col1:
@@ -785,7 +797,7 @@ def process_video_file(video_file, analysis_types):
         os.unlink(output_path)
         
         # Return both the video and the detection statistics
-        return processed_video_bytes, detection_stats
+        return processed_video_bytes, {"detection_stats": detection_stats}
         
     except Exception as e:
         # Clean up on error
@@ -826,6 +838,198 @@ def load_bigquery_table(dataset_id, table_id, limit=1000):
         "schema": [field.name for field in table.schema]
     }
 
+def setup_groq_client():
+    """Setup GROQ client with API key from Streamlit secrets"""
+    if 'GROQ_API_KEY' in st.secrets:
+        return Groq(api_key=st.secrets["GROQ_API_KEY"])
+    else:
+        st.sidebar.warning("GROQ API key not found in secrets. Chatbot functionality will be limited.")
+        return None
+
+def get_assistant_response(client, prompt, context=None, model="llama3-70b-8192"):
+    """Get response from GROQ assistant with context awareness"""
+    if client is None:
+        return "I'm unable to connect to my knowledge base right now. Please check the API configuration."
+    
+    # Build a context-aware prompt
+    if context:
+        full_prompt = f"""You are an AI assistant for the Cosmick Cloud AI Analyzer application.
+        
+Current application context: {context}
+
+User question: {prompt}
+
+Please provide a helpful, accurate response based on the current application context. 
+If you need more specific information to answer correctly, please ask for it."""
+    else:
+        full_prompt = f"""You are an AI assistant for the Cosmick Cloud AI Analyzer application.
+        
+The application has the following tools:
+1. Image Analysis: Analyzes images for labels, objects, text, and faces
+2. Video Analysis: Processes videos to detect objects, faces, and text
+3. Document Analysis: Extracts text and structure from documents
+4. Data Analysis: Uploads, queries, and visualizes data in BigQuery
+
+User question: {prompt}
+
+Please provide a helpful response to guide the user on using these tools."""
+
+    # Call GROQ API
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "system",
+                    "content": "You are a helpful, knowledgeable assistant for the Cosmick Cloud AI Analyzer application."
+                },
+                {
+                    "role": "user",
+                    "content": full_prompt
+                }
+            ],
+            model=model,
+            temperature=0.5,
+            max_tokens=1024,
+            top_p=1,
+            stream=False,
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        return f"I encountered an error: {str(e)}. Please try again or check the API configuration."
+
+def chatbot_interface():
+    """Create a chatbot interface at the bottom of the app"""
+    # Initialize chat history
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+    
+    # Initialize GROQ client
+    if "groq_client" not in st.session_state:
+        st.session_state.groq_client = setup_groq_client()
+    
+    # Get current context
+    current_context = ""
+    if "current_tool" in st.session_state:
+        current_context += f"Current tool: {st.session_state.current_tool}\n"
+    if "analysis_results" in st.session_state:
+        current_context += f"Analysis results: {st.session_state.analysis_results}\n"
+    
+    # Create chatbot container
+    chat_container = st.container()
+    
+    with chat_container:
+        # Create a custom expandable chat interface
+        st.markdown("""
+        <style>
+        .chat-container {
+            position: fixed;
+            bottom: 0;
+            right: 20px;
+            width: 400px;
+            background-color: white;
+            border-radius: 10px 10px 0 0;
+            box-shadow: 0 0 10px rgba(0,0,0,0.2);
+            z-index: 1000;
+        }
+        .chat-header {
+            padding: 10px 15px;
+            background-color: #4285F4;
+            color: white;
+            border-radius: 10px 10px 0 0;
+            font-weight: bold;
+            cursor: pointer;
+        }
+        .chat-messages {
+            max-height: 300px;
+            overflow-y: auto;
+            padding: 10px;
+        }
+        .chat-input {
+            padding: 10px;
+            border-top: 1px solid #eee;
+        }
+        </style>
+        
+        <div class="chat-container" id="chat-container">
+            <div class="chat-header" onclick="toggleChat()">
+                💬 Cosmick AI Assistant - Click to toggle
+            </div>
+            <div id="chat-content">
+                <div class="chat-messages" id="chat-messages">
+                    <div id="messages-container"></div>
+                </div>
+                <div class="chat-input">
+                    <input type="text" id="user-input" placeholder="Ask me anything...">
+                    <button onclick="sendMessage()">Send</button>
+                </div>
+            </div>
+        </div>
+        
+        <script>
+        function toggleChat() {
+            const content = document.getElementById('chat-content');
+            if (content.style.display === 'none') {
+                content.style.display = 'block';
+            } else {
+                content.style.display = 'none';
+            }
+        }
+        
+        function sendMessage() {
+            const input = document.getElementById('user-input');
+            const message = input.value;
+            if (message.trim() === '') return;
+            
+            // Clear input
+            input.value = '';
+            
+            // Add message to UI
+            const messagesContainer = document.getElementById('messages-container');
+            messagesContainer.innerHTML += `<div style="margin-bottom: 10px;"><strong>You:</strong> ${message}</div>`;
+            
+            // Send message to Streamlit
+            window.parent.postMessage({
+                type: 'streamlit:message',
+                data: {
+                    type: 'chatbot_message',
+                    message: message
+                }
+            }, '*');
+        }
+        </script>
+        """, unsafe_allow_html=True)
+        
+        # Create a container for the actual chat interface using Streamlit components
+        chat_placeholder = st.empty()
+        
+        with chat_placeholder:
+            # Display chat messages
+            for message in st.session_state.messages:
+                if message["role"] == "user":
+                    st.markdown(f"**You:** {message['content']}")
+                else:
+                    st.markdown(f"**Assistant:** {message['content']}")
+            
+            # Chat input
+            user_input = st.chat_input("Ask me anything about image, video, document or data analysis...")
+            
+            if user_input:
+                # Add user message to history
+                st.session_state.messages.append({"role": "user", "content": user_input})
+                
+                # Get assistant response
+                response = get_assistant_response(
+                    st.session_state.groq_client,
+                    user_input,
+                    context=current_context
+                )
+                
+                # Add assistant response to history
+                st.session_state.messages.append({"role": "assistant", "content": response})
+                
+                # Rerun to update chat display
+                st.rerun()
+
 def main():
     # Header - Updated title
     st.markdown('<div class="main-header">Cosmick Cloud AI Analyzer</div>', unsafe_allow_html=True)
@@ -840,6 +1044,9 @@ def main():
         orientation="horizontal",
     )
     
+    # Store current tool in session state for context
+    st.session_state.current_tool = selected
+    
     if selected == "Image Analysis":
         # Sidebar controls
         with st.sidebar:
@@ -1552,16 +1759,8 @@ def main():
         
         st.info("Note: Make sure your Google Cloud credentials are properly set up to use this application.")
 
-    # At the end of the main function, add this footer
-    # Add footer with attribution
-    st.markdown("""
-    <div style='position: fixed; bottom: 0; width: 100%; background-color: #f8f9fa; 
-         text-align: center; padding: 10px; border-top: 1px solid #e9ecef;'>
-        <p style='color: #6c757d; font-size: 14px;'>
-            Powered by Google Cloud with additional tools
-        </p>
-    </div>
-    """, unsafe_allow_html=True)
+    # Add the chatbot interface at the bottom of the page
+    chatbot_interface()
 
 if __name__ == "__main__":
     # Use GOOGLE_CREDENTIALS directly - no need for file or GOOGLE_APPLICATION_CREDENTIALS