Update src/streamlit_app.py

src/streamlit_app.py

@@ -34,7 +34,15 @@ logger = logging.getLogger(__name__)
 
 def configure_gemini():
     """Configure Gemini API with backend key"""
-    genai.configure(api_key=GEMINI_API_KEY)
+    if not GEMINI_API_KEY:
+        st.error("GEMINI_KEY not found in environment variables")
+        return False
+    try:
+        genai.configure(api_key=GEMINI_API_KEY)
+        return True
+    except Exception as e:
+        st.error(f"Failed to configure Gemini API: {str(e)}")
+        return False
 
 # Enhanced system prompt with timestamp-based improvements
 SYSTEM_PROMPT = f"""{os.getenv("SYS_PROMPT")}"""
@@ -57,7 +65,8 @@ def analyze_video_and_generate_script(
             tmp_file_path = tmp_file.name
         
         # Configure Gemini
-        configure_gemini()
+        if not configure_gemini():
+            return None
         
         # Show upload progress
         upload_progress = st.progress(0)
@@ -121,8 +130,12 @@ def analyze_video_and_generate_script(
         IMPORTANT: Return only valid JSON in the exact format specified in the system prompt. Analyze the video second-by-second for maximum detail."""
 
         # Generate response
-        model = genai.GenerativeModel("gemini-1.5-flash")
-        response = model.generate_content([video_file_obj, user_prompt + "\n\n" + SYSTEM_PROMPT])
+        try:
+            model = genai.GenerativeModel("gemini-2.0-flash")
+            response = model.generate_content([video_file_obj, user_prompt + "\n\n" + SYSTEM_PROMPT])
+        except Exception as e:
+            st.error(f"Error generating content with Gemini: {str(e)}")
+            return None
         
         upload_progress.progress(100)
         upload_status.success("Analysis complete!")