Update streamlit_app.py

streamlit_app.py

@@ -1,8 +1,8 @@
 # streamlit_app.py
 import os
-import string
 import time
 import json
+import string
 from glob import glob
 from pathlib import Path
 
@@ -10,30 +10,41 @@ import yt_dlp
 import ffmpeg
 import streamlit as st
 from dotenv import load_dotenv
-import requests
 
 load_dotenv()
 
-# Optional Google SDK helpers
+# Optional SDK imports — we'll try to import and set flags
+HAS_GENAI = False
+GENAI = None
 try:
     import google.generativeai as genai
     from google.generativeai import upload_file, get_file
     GENAI = genai
     HAS_GENAI = True
 except Exception:
-    GENAI = None
-    HAS_GENAI = False
+    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", "")
-# default loop unchecked as requested
 st.session_state.setdefault("loop_video", False)
 
-# UI - Sidebar
+# Sidebar UI
 st.sidebar.header("Video Input")
 st.sidebar.text_input("Video URL (or local .mp4 path)", key="url", placeholder="Enter Video URL or path")
 
@@ -41,7 +52,6 @@ settings_exp = st.sidebar.expander("Settings", expanded=False)
 env_api_key = os.getenv("GOOGLE_API_KEY", "")
 API_KEY = settings_exp.text_input("Google API Key", value=env_api_key, placeholder="Set GOOGLE_API_KEY in .env or enter here")
 
-# Keep model_id as the short model name (e.g. "gemini-2.0-flash-lite")
 raw_model = settings_exp.text_input("Gemini Model (model name)", "gemini-2.0-flash-lite")
 model_id = raw_model.strip() or "gemini-2.0-flash-lite"
 
@@ -51,7 +61,7 @@ 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="⚠️")
 
-# Basic safety settings (kept OFF by default)
+# Safety settings (kept OFF by default)
 safety_settings = [
     {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "OFF"},
     {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "OFF"},
@@ -59,7 +69,7 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "OFF"},
 ]
 
-# Utilities
+# Helpers
 def sanitize_filename(path_str: str):
     name = Path(path_str).name
     name = name.lower().translate(str.maketrans("", "", string.punctuation)).replace(" ", "_")
@@ -82,7 +92,7 @@ def download_video_ytdlp(url: str, save_dir: str, video_password: str = None) ->
     # local file
     if os.path.exists(url) and os.path.isfile(url):
         return convert_video_to_mp4(url)
-    outtmpl = os.path.join(save_dir, "%(id)s.%(ext)s")
+    outtmpl = str(Path(save_dir) / "%(id)s.%(ext)s")
     ydl_opts = {"outtmpl": outtmpl, "format": "best"}
     if video_password:
         ydl_opts["videopassword"] = video_password
@@ -96,7 +106,18 @@ 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])
 
-# Load video action
+# 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
+
+# Download action
 if st.sidebar.button("Load Video", use_container_width=True):
     try:
         video_password = st.session_state.get("video-password", "")
@@ -138,13 +159,13 @@ if st.session_state["videos"]:
 
     st.sidebar.write("Title:", Path(st.session_state["videos"]).name)
 
-# Upload helpers
+# 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 upload (legacy v1beta2 upload used by file-based multimodal flows)
+    # 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}
@@ -182,7 +203,7 @@ def poll_file_processed(file_obj, timeout=180):
             raise TimeoutError("File processing timed out")
         time.sleep(2)
 
-# Call model (streamlined, using generativeapis host path format)
+# 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")
@@ -193,34 +214,26 @@ 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 responses.generate if available
-    if HAS_GENAI:
-        try:
-            if hasattr(GENAI, "responses") and hasattr(GENAI.responses, "generate"):
-                request = {
-                    "model": f"models/{model_id}",
-                    "input": [{"text": prompt_text, "files": [{"name": file_ref_name}]}],
-                    "safetySettings": safety_settings,
-                    "maxOutputTokens": 1000,
-                }
-                return GENAI.responses.generate(**request)
-        except Exception:
-            pass
-
-    # Use generativeapis.googleapis.com with correct path: /v1/models/{model}:generate
-    endpoint = f"https://generativeapis.googleapis.com/v1/models/{model_id}:generate"
+    # Prefer SDK
+    if HAS_GENAI and hasattr(GENAI, "responses") and hasattr(GENAI.responses, "generate"):
+        request = {
+            "model": f"models/{model_id}",
+            "input": [{"text": prompt_text, "files": [{"name": file_ref_name}]}],
+            "safetySettings": safety_settings,
+            "maxOutputTokens": 1000,
+        }
+        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 = {
-        "input": [
-            {
-                "text": prompt_text,
-                "files": [{"name": file_ref_name}],
-            }
-        ],
+        "input": [{"text": prompt_text, "files": [{"name": file_ref_name}]}],
         "safetySettings": safety_settings,
         "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()