Update streamlit_app.py

streamlit_app.py

@@ -13,7 +13,7 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-# Try imports (phi and google SDK)
+# Optional SDKs
 try:
     from phi.agent import Agent
     from phi.model.google import Gemini
@@ -26,11 +26,10 @@ except Exception:
 try:
     import google.generativeai as genai
     from google.generativeai import upload_file, get_file, ResponsesClient
-    GENAI = genai
     HAS_GENAI = True
 except Exception:
+    genai = None
     upload_file = get_file = ResponsesClient = None
-    GENAI = None
     HAS_GENAI = False
 
 st.set_page_config(page_title="Generate the story of videos:", layout="wide")
@@ -110,16 +109,10 @@ def file_name_or_id(file_obj):
         return file_obj.get("name") or file_obj.get("id")
     return getattr(file_obj, "name", None) or getattr(file_obj, "id", None) or getattr(file_obj, "fileId", None)
 
-# Ensure SDK symbols are defined consistently
-if not HAS_GENAI:
-    upload_file = None
-    get_file = None
-    ResponsesClient = None
-
 # configure GENAI if available
 if API_KEY and HAS_GENAI:
     try:
-        GENAI.configure(api_key=API_KEY)
+        genai.configure(api_key=API_KEY)
     except Exception:
         st.sidebar.warning("Failed to configure google.generativeai with provided API key.")
 else:
@@ -184,7 +177,6 @@ def upload_video_sdk(filepath: str):
 
 def wait_for_processed(file_obj, timeout=180):
     if not HAS_GENAI or get_file is None:
-        # If SDK not available, return the original object and let later code handle it.
         return file_obj
     start = time.time()
     name = file_name_or_id(file_obj)
@@ -229,7 +221,6 @@ if st.button("Generate the story", type="primary"):
                         raise RuntimeError("Uploaded file missing name/id")
 
                     client = ResponsesClient()
-                    # ResponsesClient.generate expects a request dict; adapt to SDK
                     request = {
                         "model": model_id if not model_id.startswith("models/") else model_id.split("/", 1)[1],
                         "input": [{"text": prompt_text, "files": [{"name": fname}]}],
@@ -238,7 +229,6 @@ if st.button("Generate the story", type="primary"):
                     }
                     res = client.generate(**request)
 
-                    # extract text from response
                     out = ""
                     for item in getattr(res, "output", []) or res.get("output", []):
                         for c in getattr(item, "content", []) or item.get("content", []):