Update streamlit_app.py

streamlit_app.py

@@ -25,7 +25,7 @@ except Exception:
 
 try:
     import google.generativeai as genai
-    from google.generativeai import upload_file, get_file
+    from google.generativeai import upload_file, get_file, ResponsesClient
     GENAI = genai
     HAS_GENAI = True
 except Exception:
@@ -76,7 +76,6 @@ def convert_video_to_mp4(video_path: str) -> str:
     try:
         ffmpeg.input(video_path).output(target_path).run(overwrite_output=True, quiet=True)
     except Exception as e:
-        # If ffmpeg binary missing or conversion fails, re-raise with hint
         raise RuntimeError(f"ffmpeg conversion failed: {e}")
     try:
         os.remove(video_path)
@@ -95,7 +94,6 @@ def download_video_ytdlp(url: str, save_dir: str, video_password: str = None) ->
         ydl_opts["videopassword"] = video_password
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         info = ydl.extract_info(url, download=True)
-    # Prefer returned id/filename
     video_id = info.get("id") if isinstance(info, dict) else None
     if video_id:
         matches = glob(os.path.join(save_dir, f"{video_id}.*"))
@@ -105,13 +103,11 @@ 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])
 
-# Helper: robustly extract name/id from SDK file object or dict
 def file_name_or_id(file_obj):
     if file_obj is None:
         return None
     if isinstance(file_obj, dict):
         return file_obj.get("name") or file_obj.get("id")
-    # some SDKs return an object with .name or .id attributes
     return getattr(file_obj, "name", None) or getattr(file_obj, "id", None) or getattr(file_obj, "fileId", None)
 
 # configure GENAI if available
@@ -208,24 +204,25 @@ if st.button("Generate the story", type="primary"):
                     response = _agent.run(prompt_text, videos=[processed], safety_settings=safety_settings)
                     out = getattr(response, "content", None) or getattr(response, "outputText", None) or str(response)
             else:
-                # fallback: call GENAI.responses.generate directly
-                with st.spinner("Generating description via SDK..."):
+                with st.spinner("Generating description via Responses client..."):
                     fname = file_name_or_id(processed)
                     if not fname:
                         raise RuntimeError("Uploaded file missing name/id")
 
+                    client = ResponsesClient()
+                    # ResponsesClient.generate expects a request dict; adapt to SDK
                     request = {
-                        "model": f"models/{model_id}",
+                        "model": model_id if not model_id.startswith("models/") else model_id.split("/", 1)[1],
                         "input": [{"text": prompt_text, "files": [{"name": fname}]}],
                         "safetySettings": safety_settings,
                         "maxOutputTokens": 1000,
                     }
-                    res = GENAI.responses.generate(**request)
+                    res = client.generate(**request)
+
+                    # extract text from response
                     out = ""
-                    # basic extraction
-                    for item in (getattr(res, "output", None) or res.get("output", [])):
-                        contents = getattr(item, "content", None) or item.get("content", [])
-                        for c in contents:
+                    for item in getattr(res, "output", []) or res.get("output", []):
+                        for c in getattr(item, "content", []) or item.get("content", []):
                             ctype = getattr(c, "type", None) or c.get("type")
                             if ctype in ("output_text", "text"):
                                 out += getattr(c, "text", "") or c.get("text", "")