Update streamlit_app.py

streamlit_app.py

@@ -105,6 +105,15 @@ 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
 if API_KEY and HAS_GENAI:
     try:
@@ -168,7 +177,7 @@ def upload_video_sdk(filepath: str):
 
 def wait_for_processed(file_obj, timeout=180):
     start = time.time()
-    name = getattr(file_obj, "name", None) or file_obj.get("name") or file_obj.get("id")
+    name = file_name_or_id(file_obj)
     if not name:
         return file_obj
     while True:
@@ -201,18 +210,24 @@ if st.button("Generate the story", type="primary"):
             else:
                 # fallback: call GENAI.responses.generate directly
                 with st.spinner("Generating description via SDK..."):
+                    fname = file_name_or_id(processed)
+                    if not fname:
+                        raise RuntimeError("Uploaded file missing name/id")
+
                     request = {
                         "model": f"models/{model_id}",
-                        "input": [{"text": prompt_text, "files": [{"name": getattr(processed, 'name', processed.get('name'))}]}],
+                        "input": [{"text": prompt_text, "files": [{"name": fname}]}],
                         "safetySettings": safety_settings,
                         "maxOutputTokens": 1000,
                     }
                     res = GENAI.responses.generate(**request)
                     out = ""
                     # basic extraction
-                    for item in res.output or res.get("output", []):
-                        for c in (getattr(item, "content", None) or item.get("content", [])):
-                            if getattr(c, "type", None) in ("output_text", "text") or c.get("type") in ("output_text", "text"):
+                    for item in (getattr(res, "output", None) or res.get("output", [])):
+                        contents = getattr(item, "content", None) or item.get("content", [])
+                        for c in contents:
+                            ctype = getattr(c, "type", None) or c.get("type")
+                            if ctype in ("output_text", "text"):
                                 out += getattr(c, "text", "") or c.get("text", "")
                     if not out:
                         out = getattr(res, "text", None) or json.dumps(res, default=str)