Update app.py

app.py

@@ -6,6 +6,13 @@ import time
 import mimetypes
 from pathlib import Path
 
+# --- Get API key from environment variable or user input ---
+def get_api_key():
+    GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY", "")
+    if not GOOGLE_API_KEY:
+        GOOGLE_API_KEY = st.text_input("Enter your Gemini API key", type="password")
+    return GOOGLE_API_KEY or "AIzaSyA8TTu9s6fJDG9RlMwOyHFxg270xLgpiyE"  # Warning: Hardcoded key
+
 # VideoProcessor class
 class VideoProcessor:
     def __init__(self, api_key):
@@ -39,268 +46,123 @@ class VideoProcessor:
             return f"Error generating response: {str(e)}"
 
 # Initialize session state properly
-def init_session_state():
-    if "messages" not in st.session_state:
-        st.session_state.messages = []
-    if "video_processor" not in st.session_state:
-        st.session_state.video_processor = None
-    if "video_file" not in st.session_state:
-        st.session_state.video_file = None
-    if "video_name" not in st.session_state:
-        st.session_state.video_name = None
-    if "api_key_validated" not in st.session_state:
-        st.session_state.api_key_validated = False
-    if "debug_mode" not in st.session_state:
-        st.session_state.debug_mode = False
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+if "video_processor" not in st.session_state:
+    st.session_state.video_processor = None
+if "video_file" not in st.session_state:
+    st.session_state.video_file = None
+if "video_name" not in st.session_state:
+    st.session_state.video_name = None
 
 # Main app function
 def main():
-    st.set_page_config(page_title="Video Retrieval-Augmented Generation", page_icon="🎥", layout="wide")
-    st.header("🎥 Video Retrieval-Augmented Generation - Gemini 2.0")
-    
-    # Initialize session state
-    init_session_state()
-    
-    # Debug mode toggle in sidebar
-    with st.sidebar:
-        st.session_state.debug_mode = st.checkbox("Debug Mode", value=st.session_state.debug_mode)
-        if st.session_state.debug_mode:
-            st.write("### Session State:")
-            st.write(f"API Key Validated: {st.session_state.api_key_validated}")
-            st.write(f"Video File: {'Yes' if st.session_state.video_file else 'No'}")
-            st.write(f"Video Name: {st.session_state.video_name}")
-            st.write(f"Messages Count: {len(st.session_state.messages)}")
-    
+    st.set_page_config(page_title="Video Retrieval-Augmented Generation", page_icon="🎬", layout="wide")
+    st.header("🎬 Video Retrieval-Augmented Generation - Gemini 2.0")
     st.markdown("---")
 
     # Step 1: API Key input
-    st.subheader("📌 Step 1: Enter Your API Key")
-    
-    # Check for environment variable first
-    env_api_key = os.environ.get("GOOGLE_API_KEY", "")
-    api_key = None
-    
-    if env_api_key:
-        st.info("✅ Using API key from environment variable")
-        api_key = env_api_key
-        # Auto-validate environment API key
-        if not st.session_state.api_key_validated:
-            try:
-                st.session_state.video_processor = VideoProcessor(api_key)
-                st.session_state.api_key_validated = True
-                st.success("✅ API key validated successfully")
-            except Exception as e:
-                st.error(f"❌ Invalid API key from environment: {str(e)}")
-                api_key = None
-    else:
-        # Direct API key input
-        api_key = st.text_input(
-            "Enter your Gemini API key below:",
-            type="password",
-            placeholder="AIzaSy...",
-            help="Get your API key from https://makersuite.google.com/app/apikey",
-            key="api_key_input"
-        )
-        
-        if api_key:
-            # Add a button to validate the API key
-            col1, col2 = st.columns([1, 4])
-            with col1:
-                validate_button = st.button("Validate Key", type="primary")
-            
-            if validate_button or (api_key and not st.session_state.api_key_validated):
-                try:
-                    with st.spinner("Validating API key..."):
-                        test_processor = VideoProcessor(api_key)
-                        st.session_state.video_processor = test_processor
-                        st.session_state.api_key_validated = True
-                    st.success("✅ API key validated successfully!")
-                    st.balloons()
-                    time.sleep(1)
-                    st.rerun()
-                except Exception as e:
-                    st.error(f"❌ Invalid API key: {str(e)}")
-                    st.session_state.api_key_validated = False
-                    st.session_state.video_processor = None
-        else:
-            st.info("ℹ️ Please enter your Gemini API key above to proceed")
-            with st.expander("📖 How to obtain your API key"):
-                st.markdown("""
-                1. Navigate to [Google AI Studio](https://makersuite.google.com/app/apikey)
-                2. Sign in with your Google account
-                3. Click on "Get API key" or "Create API key"
-                4. Copy the generated API key
-                5. Paste it in the field above
-                
-                **Important:** Keep your API key secure and do not share it publicly.
-                """)
+    #st.subheader("Step 1: Enter your Gemini API key")
+    api_key = get_api_key()
 
-    # Check if we can proceed to next steps
-    if not st.session_state.api_key_validated:
-        st.warning("⚠️ Please enter a valid API key to proceed to Steps 2 and 3")
-        return  # Changed from st.stop() to return
-
-    st.markdown("---")
+    if not api_key:
+        st.error("Please enter your API key to proceed.")
+        st.stop()
 
     # Step 2: Upload Video
-    st.subheader("📹 Step 2: Video Upload")
-    
-    # Show current status
-    if st.session_state.video_name:
-        st.success(f"✅ Current video: {st.session_state.video_name}")
-    
-    uploaded_file = st.file_uploader(
-        "Select a video file", 
-        type=['mp4', 'mov', 'avi', 'mkv', 'webm'],
-        help="Maximum file size: 200MB"
-    )
+    st.subheader("Step 2: Upload your video file")
+    uploaded_file = st.file_uploader("Upload a video", type=['mp4', 'mov', 'avi', 'mkv', 'webm'])
 
     if uploaded_file:
         # Validate video file
         mime_type = mimetypes.guess_type(uploaded_file.name)[0]
-        
-        if st.session_state.debug_mode:
-            st.info(f"Debug: MIME type detected: {mime_type}")
-        
         if mime_type and mime_type.startswith("video"):
-            file_size_mb = len(uploaded_file.getvalue()) / (1024**2)
+            file_size = len(uploaded_file.getvalue()) / (1024**2)
             
             # Display file info
-            col1, col2, col3 = st.columns(3)
+            col1, col2 = st.columns(2)
             with col1:
-                st.metric("File", uploaded_file.name)
+                st.info(f"📁 File: {uploaded_file.name}")
             with col2:
-                st.metric("Size", f"{file_size_mb:.2f} MB")
-            with col3:
-                st.metric("Type", mime_type.split('/')[-1].upper())
+                st.info(f"💾 Size: {file_size:.2f} MB")
             
-            # Check file size limit
-            if file_size_mb > 200:
-                st.error("❌ File size exceeds 200MB limit")
-                return
-            
-            # Process button
-            process_video = False
-            if st.session_state.video_name != uploaded_file.name:
-                if st.button("🚀 Process Video", type="primary", use_container_width=True):
-                    process_video = True
-            else:
-                st.info("ℹ️ This video has already been processed")
+            # Check file size limit (200MB for Gemini)
+            if file_size > 200:
+                st.error("❌ File too large! Maximum size is 200MB")
+                st.stop()
 
-            # Process new video
-            if process_video:
+            # Process new video if different from current
+            if st.session_state.video_name != uploaded_file.name:
                 tmp_path = None
-                progress_container = st.container()
-                
                 try:
-                    with progress_container:
-                        # Create temporary file
-                        with tempfile.NamedTemporaryFile(delete=False, suffix=Path(uploaded_file.name).suffix) as tmp:
-                            tmp.write(uploaded_file.getvalue())
-                            tmp_path = tmp.name
-                        
-                        if st.session_state.debug_mode:
-                            st.info(f"Debug: Temp file created at {tmp_path}")
-                        
-                        # Upload and process with detailed progress
+                    # Initialize processor if needed
+                    if not st.session_state.video_processor:
+                        st.session_state.video_processor = VideoProcessor(api_key)
+                    
+                    # Create temporary file
+                    with tempfile.NamedTemporaryFile(delete=False, suffix=Path(uploaded_file.name).suffix) as tmp:
+                        tmp.write(uploaded_file.getvalue())
+                        tmp_path = tmp.name
+
+                    # Upload and process with progress indication
+                    with st.spinner("📤 Uploading and processing video..."):
                         progress_bar = st.progress(0)
-                        status_text = st.empty()
+                        progress_bar.progress(25, text="Uploading video...")
                         
-                        # Step 1: Upload
-                        status_text.text("📤 Uploading video to Gemini...")
-                        progress_bar.progress(25)
                         video_file = st.session_state.video_processor.upload_video(tmp_path, uploaded_file.name)
+                        progress_bar.progress(50, text="Processing video...")
                         
-                        if st.session_state.debug_mode:
-                            st.info(f"Debug: Video uploaded, state: {video_file.state.name}")
-                        
-                        # Step 2: Process
-                        status_text.text("⏳ Processing video (this may take a minute)...")
-                        progress_bar.progress(50)
                         processed_file = st.session_state.video_processor.wait_for_processing(video_file)
-                        
-                        if st.session_state.debug_mode:
-                            st.info(f"Debug: Video processed, final state: {processed_file.state.name}")
-                        
-                        # Step 3: Complete
-                        progress_bar.progress(100)
-                        status_text.text("✅ Video ready for analysis!")
+                        progress_bar.progress(100, text="Complete!")
                         
                         # Update session state
                         st.session_state.video_file = processed_file
                         st.session_state.video_name = uploaded_file.name
                         st.session_state.messages = []  # Clear previous conversation
                         
-                        st.success("🎉 Video processed successfully! You can now ask questions below.")
-                        time.sleep(2)
-                        st.rerun()  # Refresh to show chat interface
-                        
+                        st.success("✅ Video processed successfully!")
+                        time.sleep(1)  # Show success message briefly
+                        progress_bar.empty()  # Clear progress bar
+
                 except Exception as e:
                     st.error(f"❌ Error processing video: {str(e)}")
-                    if st.session_state.debug_mode:
-                        st.exception(e)
                     st.session_state.video_file = None
                     st.session_state.video_name = None
                     
                 finally:
                     # Clean up temporary file
                     if tmp_path and os.path.exists(tmp_path):
-                        try:
-                            os.unlink(tmp_path)
-                            if st.session_state.debug_mode:
-                                st.info("Debug: Temp file cleaned up")
-                        except:
-                            pass
+                        os.unlink(tmp_path)
 
             # Display video player
-            if uploaded_file:
-                st.video(uploaded_file.getvalue())
+            st.video(uploaded_file.getvalue())
         else:
             st.error("❌ Please upload a valid video file")
 
     # Control buttons
-    st.markdown("---")
-    col1, col2, col3 = st.columns(3)
+    col1, col2 = st.columns(2)
     with col1:
-        if st.button("🔄 Reset Chat", disabled=not st.session_state.messages, use_container_width=True):
+        if st.button("🔄 Reset Chat", disabled=not st.session_state.messages):
             st.session_state.messages = []
-            st.success("Chat history cleared!")
-            time.sleep(1)
             st.rerun()
     
     with col2:
-        if st.button("📹 Reset Video", disabled=not st.session_state.video_file, use_container_width=True):
-            st.session_state.video_file = None
-            st.session_state.video_name = None
-            st.session_state.messages = []
-            st.success("Video reset!")
-            time.sleep(1)
-            st.rerun()
-    
-    with col3:
-        if st.button("🗑️ Clear All", type="secondary", use_container_width=True):
+        if st.button("🗑️ Reset All", disabled=not st.session_state.video_file):
             for key in list(st.session_state.keys()):
                 del st.session_state[key]
-            st.success("All data cleared!")
-            time.sleep(1)
             st.rerun()
 
-    st.markdown("---")
-
     # Step 3: Chat about Video
-    st.subheader("💬 Step 3: Video Analysis Chat")
+    st.subheader("Step 3: Chat with your video")
 
     if st.session_state.video_file:
-        st.success(f"✅ Ready to analyze: {st.session_state.video_name}")
-        
         # Display chat history
         for msg in st.session_state.messages:
             with st.chat_message(msg["role"]):
                 st.markdown(msg["content"])
 
         # Chat input
-        user_question = st.chat_input("Ask a question about the video...", key="chat_input")
-        
+        user_question = st.chat_input("Ask a question about the video...")
         if user_question:
             # Add user message
             st.session_state.messages.append({"role": "user", "content": user_question})
@@ -309,38 +171,20 @@ def main():
 
             # Generate and display assistant response
             with st.chat_message("assistant"):
-                with st.spinner("🤔 Analyzing video..."):
+                placeholder = st.empty()
+                with st.spinner("🤔 Thinking..."):
                     try:
                         response = st.session_state.video_processor.chat_with_video(
                             st.session_state.video_file, 
                             user_question
                         )
-                        
-                        if st.session_state.debug_mode:
-                            st.info(f"Debug: Response length: {len(response)} characters")
-                            
                     except Exception as e:
                         response = f"❌ Error: {str(e)}"
-                        if st.session_state.debug_mode:
-                            st.exception(e)
 
-                st.markdown(response)
+                placeholder.markdown(response)
                 st.session_state.messages.append({"role": "assistant", "content": response})
-        
-        # Sample questions
-        with st.expander("💡 Sample Questions"):
-            st.markdown("""
-            - What is happening in this video?
-            - Can you describe the main events?
-            - What objects or people can you see?
-            - What is the mood or atmosphere?
-            - Can you provide a summary of the video?
-            - What are the key moments in the video?
-            """)
     else:
-        st.info("📤 Please upload and process a video in Step 2 to begin analysis.")
-        if st.session_state.debug_mode:
-            st.warning("Debug: No video file in session state")
+        st.info("📹 Please upload a video in Step 2 to start chatting.")
 
 if __name__ == "__main__":
     main()