Update app.py

app.py

@@ -6,13 +6,6 @@ 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 "AIzaSyChkOeyjQb4wkjMfgROQKAkPIxP0GFpqqc"  # Warning: Hardcoded key
-
 # VideoProcessor class
 class VideoProcessor:
     def __init__(self, api_key):
@@ -54,6 +47,8 @@ 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" not in st.session_state:
+    st.session_state.api_key = ""
 
 # Main app function
 def main():
@@ -62,16 +57,77 @@ def main():
     st.markdown("---")
 
     # Step 1: API Key input
-    st.subheader("Step 1: Enter your Gemini API key")
-    api_key = get_api_key()
+    st.subheader("Step 1: API Configuration")
+    
+    # Check for environment variable first
+    env_api_key = os.environ.get("GOOGLE_API_KEY", "")
+    
+    if env_api_key:
+        st.info("Using API key from environment variable")
+        st.session_state.api_key = env_api_key
+        api_key = env_api_key
+    else:
+        # Create columns for better layout
+        col1, col2 = st.columns([3, 1])
+        
+        with col1:
+            # API key input field
+            api_key_input = st.text_input(
+                "Gemini API Key",
+                value=st.session_state.api_key,
+                type="password",
+                placeholder="AIzaSy...",
+                help="Get your API key from https://makersuite.google.com/app/apikey"
+            )
+        
+        with col2:
+            # Save button
+            st.markdown("<br>", unsafe_allow_html=True)  # Add spacing
+            if st.button("Save API Key", type="primary"):
+                if api_key_input:
+                    st.session_state.api_key = api_key_input
+                    st.success("API key saved successfully")
+                    # Reinitialize processor with new key
+                    st.session_state.video_processor = None
+                else:
+                    st.error("Please enter an API key")
+        
+        api_key = st.session_state.api_key
+        
+        # Show instructions if no API key
+        if not api_key:
+            st.warning("Please enter your Gemini API key to continue")
+            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 and click "Save API Key"
+                
+                **Important:** Keep your API key secure and do not share it publicly.
+                """)
 
     if not api_key:
-        st.error("Please enter your API key to proceed.")
+        st.error("API key required. Please enter your API key to proceed.")
         st.stop()
 
+    # Initialize processor with the API key if not already initialized
+    if api_key and not st.session_state.video_processor:
+        try:
+            st.session_state.video_processor = VideoProcessor(api_key)
+            st.success("API key validated and processor initialized")
+        except Exception as e:
+            st.error(f"Invalid API key or initialization error: {str(e)}")
+            st.session_state.api_key = ""
+            st.session_state.video_processor = None
+            st.stop()
+
+    st.markdown("---")
+
     # Step 2: Upload Video
-    st.subheader("Step 2: Upload your video file")
-    uploaded_file = st.file_uploader("Upload a video", type=['mp4', 'mov', 'avi', 'mkv', 'webm'])
+    st.subheader("Step 2: Video Upload")
+    uploaded_file = st.file_uploader("Select a video file", type=['mp4', 'mov', 'avi', 'mkv', 'webm'])
 
     if uploaded_file:
         # Validate video file
@@ -82,30 +138,26 @@ def main():
             # Display file info
             col1, col2 = st.columns(2)
             with col1:
-                st.info(f"📁 File: {uploaded_file.name}")
+                st.info(f"File: {uploaded_file.name}")
             with col2:
-                st.info(f"💾 Size: {file_size:.2f} MB")
+                st.info(f"Size: {file_size:.2f} MB")
             
             # Check file size limit (200MB for Gemini)
             if file_size > 200:
-                st.error("❌ File too large! Maximum size is 200MB")
+                st.error("File size exceeds limit. Maximum file size is 200MB.")
                 st.stop()
 
             # Process new video if different from current
             if st.session_state.video_name != uploaded_file.name:
                 tmp_path = None
                 try:
-                    # 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..."):
+                    with st.spinner("Uploading and processing video..."):
                         progress_bar = st.progress(0)
                         progress_bar.progress(25, text="Uploading video...")
                         
@@ -113,19 +165,19 @@ def main():
                         progress_bar.progress(50, text="Processing video...")
                         
                         processed_file = st.session_state.video_processor.wait_for_processing(video_file)
-                        progress_bar.progress(100, text="Complete!")
+                        progress_bar.progress(100, text="Processing 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!")
+                        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)}")
+                    st.error(f"Error processing video: {str(e)}")
                     st.session_state.video_file = None
                     st.session_state.video_name = None
                     
@@ -137,23 +189,32 @@ def main():
             # Display video player
             st.video(uploaded_file.getvalue())
         else:
-            st.error("❌ Please upload a valid video file")
+            st.error("Please upload a valid video file")
 
     # Control buttons
-    col1, col2 = st.columns(2)
+    col1, col2, col3 = st.columns(3)
     with col1:
-        if st.button("🔄 Reset Chat", disabled=not st.session_state.messages):
+        if st.button("Reset Chat", disabled=not st.session_state.messages):
             st.session_state.messages = []
             st.rerun()
     
     with col2:
-        if st.button("🗑️ Reset All", disabled=not st.session_state.video_file):
+        if st.button("Reset Video", disabled=not st.session_state.video_file):
+            st.session_state.video_file = None
+            st.session_state.video_name = None
+            st.session_state.messages = []
+            st.rerun()
+    
+    with col3:
+        if st.button("Clear API Key"):
             for key in list(st.session_state.keys()):
                 del st.session_state[key]
             st.rerun()
 
+    st.markdown("---")
+
     # Step 3: Chat about Video
-    st.subheader("Step 3: Chat with your video")
+    st.subheader("Step 3: Video Analysis Chat")
 
     if st.session_state.video_file:
         # Display chat history
@@ -172,19 +233,19 @@ def main():
             # Generate and display assistant response
             with st.chat_message("assistant"):
                 placeholder = st.empty()
-                with st.spinner("🤔 Thinking..."):
+                with st.spinner("Processing..."):
                     try:
                         response = st.session_state.video_processor.chat_with_video(
                             st.session_state.video_file, 
                             user_question
                         )
                     except Exception as e:
-                        response = f"❌ Error: {str(e)}"
+                        response = f"Error: {str(e)}"
 
                 placeholder.markdown(response)
                 st.session_state.messages.append({"role": "assistant", "content": response})
     else:
-        st.info("📹 Please upload a video in Step 2 to start chatting.")
+        st.info("Please upload a video in Step 2 to begin analysis.")
 
 if __name__ == "__main__":
     main()