Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from google import genai
+import google.generativeai as genai
 import tempfile
 import os
 import time
@@ -34,7 +34,7 @@ logger = logging.getLogger(__name__)
 
 def configure_gemini():
     """Configure Gemini API with backend key"""
-    return genai.Client(api_key=GEMINI_API_KEY)
+    genai.configure(api_key=GEMINI_API_KEY)
 
 # Enhanced system prompt with timestamp-based improvements
 SYSTEM_PROMPT = f"""{os.getenv("SYS_PROMPT")}"""
@@ -57,7 +57,7 @@ def analyze_video_and_generate_script(
             tmp_file_path = tmp_file.name
         
         # Configure Gemini
-        client = configure_gemini()
+        configure_gemini()
         
         # Show upload progress
         upload_progress = st.progress(0)
@@ -67,13 +67,13 @@ def analyze_video_and_generate_script(
         upload_progress.progress(20)
         
         # Upload video to Gemini
-        video_file_obj = client.files.upload(file=tmp_file_path)
+        video_file_obj = genai.upload_file(tmp_file_path)
         upload_progress.progress(40)
         
         upload_status.text("Processing video...")
         while video_file_obj.state.name == "PROCESSING":
             time.sleep(2)
-            video_file_obj = client.files.get(name=video_file_obj.name)
+            video_file_obj = genai.get_file(video_file_obj.name)
             upload_progress.progress(60)
         
         if video_file_obj.state.name == "FAILED":
@@ -121,10 +121,8 @@ 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
-        response = client.models.generate_content(
-            model="gemini-2.0-flash", 
-            contents=[video_file_obj, user_prompt + "\n\n" + SYSTEM_PROMPT]
-        )
+        model = genai.GenerativeModel("gemini-1.5-flash")
+        response = model.generate_content([video_file_obj, user_prompt + "\n\n" + SYSTEM_PROMPT])
         
         upload_progress.progress(100)
         upload_status.success("Analysis complete!")
@@ -341,16 +339,6 @@ def check_token(user_token):
 def main():
     """Main application function"""
 
-    st.set_page_config(
-        page_title="Video Analyser and Script Generator",
-        page_icon="🎥",
-        layout="wide",
-        initial_sidebar_state="expanded"
-    )
-
-    st.title("Video Analyser and Script Generator")
-    st.divider()
-
     if "authenticated" not in st.session_state:
         st.session_state["authenticated"] = False