Update streamlit_app.py

streamlit_app.py

@@ -10,10 +10,11 @@ import yt_dlp
 import ffmpeg
 import streamlit as st
 from dotenv import load_dotenv
+import requests
 
 load_dotenv()
 
-# Optional SDK imports — we'll try to import and set flags
+# Try to import the Google SDK
 HAS_GENAI = False
 GENAI = None
 try:
@@ -25,22 +26,11 @@ except Exception:
     upload_file = None
     get_file = None
 
-# Optional phi Agent (useful if phi is installed)
-HAS_PHI_AGENT = False
-try:
-    from phi.agent import Agent
-    from phi.model.google import Gemini
-    from phi.tools.duckduckgo import DuckDuckGo
-    HAS_PHI_AGENT = True
-except Exception:
-    Agent = Gemini = DuckDuckGo = None
-
 st.set_page_config(page_title="Generate the story of videos:", layout="wide")
 
 DATA_DIR = Path("./data")
 DATA_DIR.mkdir(exist_ok=True)
 
-# Session defaults
 st.session_state.setdefault("videos", "")
 st.session_state.setdefault("loop_video", False)
 
@@ -61,7 +51,6 @@ settings_exp.text_input("Video Password", key="video-password", placeholder="Ent
 if not API_KEY:
     settings_exp.warning("No Google API key provided. Uploading/processing will not work.", icon="⚠️")
 
-# Safety settings (kept OFF by default)
 safety_settings = [
     {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "OFF"},
     {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "OFF"},
@@ -69,7 +58,6 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "OFF"},
 ]
 
-# Helpers
 def sanitize_filename(path_str: str):
     name = Path(path_str).name
     name = name.lower().translate(str.maketrans("", "", string.punctuation)).replace(" ", "_")
@@ -89,7 +77,6 @@ def convert_video_to_mp4(video_path: str) -> str:
 def download_video_ytdlp(url: str, save_dir: str, video_password: str = None) -> str:
     if not url:
         raise ValueError("No URL provided")
-    # local file
     if os.path.exists(url) and os.path.isfile(url):
         return convert_video_to_mp4(url)
     outtmpl = str(Path(save_dir) / "%(id)s.%(ext)s")
@@ -106,18 +93,12 @@ def download_video_ytdlp(url: str, save_dir: str, video_password: str = None) ->
         raise FileNotFoundError("Downloaded video not found")
     return convert_video_to_mp4(matches[0])
 
-# Initialize SDK if available and API key present
 if API_KEY and HAS_GENAI:
     try:
         GENAI.configure(api_key=API_KEY)
     except Exception:
-        # older SDK variants may use genai.configure or genai.set_api_key; try both silently
-        try:
-            GENAI.configure(api_key=API_KEY)
-        except Exception:
-            pass
+        pass
 
-# Download action
 if st.sidebar.button("Load Video", use_container_width=True):
     try:
         video_password = st.session_state.get("video-password", "")
@@ -126,7 +107,6 @@ if st.sidebar.button("Load Video", use_container_width=True):
     except Exception as e:
         st.sidebar.error(f"Failed to load video: {e}")
 
-# Sidebar preview & options
 if st.session_state["videos"]:
     try:
         st.sidebar.video(st.session_state["videos"], loop=st.session_state.get("loop_video", False))
@@ -159,13 +139,11 @@ if st.session_state["videos"]:
 
     st.sidebar.write("Title:", Path(st.session_state["videos"]).name)
 
-# Upload and polling using SDK if available, otherwise fallback to REST
 def upload_video(filepath: str):
     if not API_KEY:
         raise RuntimeError("No API key provided")
     if HAS_GENAI and callable(upload_file):
         return upload_file(filepath)
-    # REST fallback (v1beta2 files endpoint)
     url = "https://generative.googleapis.com/v1beta2/files?uploadType=multipart"
     headers = {"Authorization": f"Bearer {API_KEY}"}
     metadata = {"mimeType": "video/mp4", "displayName": Path(filepath).name}
@@ -203,7 +181,6 @@ def poll_file_processed(file_obj, timeout=180):
             raise TimeoutError("File processing timed out")
         time.sleep(2)
 
-# Use SDK generate when available (rest fallback also present but SDK preferred)
 def call_model_with_file(prompt_text: str, uploaded_file_obj):
     if not API_KEY:
         raise RuntimeError("No API key provided")
@@ -214,7 +191,6 @@ def call_model_with_file(prompt_text: str, uploaded_file_obj):
         else uploaded_file_obj.get("name") or uploaded_file_obj.get("id")
     )
 
-    # Prefer SDK
     if HAS_GENAI and hasattr(GENAI, "responses") and hasattr(GENAI.responses, "generate"):
         request = {
             "model": f"models/{model_id}",
@@ -224,7 +200,6 @@ def call_model_with_file(prompt_text: str, uploaded_file_obj):
         }
         return GENAI.responses.generate(**request)
 
-    # REST v1 fallback (use short model id in URL)
     model_short = model_id.split("/")[-1]
     endpoint = f"https://generativeapis.googleapis.com/v1/models/{model_short}:generate"
     body = {
@@ -233,12 +208,10 @@ def call_model_with_file(prompt_text: str, uploaded_file_obj):
         "maxOutputTokens": 1000,
     }
     headers = {"Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json"}
-    import requests
     r = requests.post(endpoint, headers=headers, json=body, timeout=180)
     r.raise_for_status()
     return r.json()
 
-# Main generate action
 if st.button("Generate the story", type="primary"):
     if not st.session_state.get("videos"):
         st.error("No video loaded. Use 'Load Video' in the sidebar.")
@@ -255,7 +228,6 @@ if st.button("Generate the story", type="primary"):
             with st.spinner("Generating description..."):
                 result = call_model_with_file(prompt_text, processed)
 
-            # Extract text from common response shapes
             text_out = ""
             if isinstance(result, dict):
                 if "output" in result: