Update streamlit_app.py

streamlit_app.py

@@ -14,7 +14,7 @@ import requests
 
 load_dotenv()
 
-# Try to import google.generativeai upload/get helpers
+# Optional Google SDK helpers
 try:
     import google.generativeai as genai
     from google.generativeai import upload_file, get_file
@@ -32,7 +32,7 @@ DATA_DIR.mkdir(exist_ok=True)
 st.session_state.setdefault("videos", "")
 st.session_state.setdefault("loop_video", True)
 
-# Sidebar controls
+# UI - Sidebar
 st.sidebar.header("Video Input")
 st.sidebar.text_input("Video URL (or local .mp4 path)", key="url", placeholder="Enter Video URL or path")
 
@@ -52,6 +52,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 = [
     {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "OFF"},
     {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "OFF"},
@@ -59,6 +60,7 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "OFF"},
 ]
 
+# Utilities
 def sanitize_filename(path_str: str):
     name = Path(path_str).name
     name = name.lower().translate(str.maketrans("", "", string.punctuation)).replace(" ", "_")
@@ -78,6 +80,7 @@ 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 = os.path.join(save_dir, "%(id)s.%(ext)s")
@@ -94,6 +97,7 @@ 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
 if st.sidebar.button("Load Video", use_container_width=True):
     try:
         video_password = st.session_state.get("video-password", "")
@@ -102,6 +106,7 @@ 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", True))
@@ -134,11 +139,13 @@ if st.session_state["videos"]:
 
     st.sidebar.write("Title:", Path(st.session_state["videos"]).name)
 
+# Upload helpers
 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)
     url = "https://generative.googleapis.com/v1beta2/files?uploadType=multipart"
     headers = {"Authorization": f"Bearer {API_KEY}"}
     metadata = {"mimeType": "video/mp4", "displayName": Path(filepath).name}
@@ -176,6 +183,7 @@ def poll_file_processed(file_obj, timeout=180):
             raise TimeoutError("File processing timed out")
         time.sleep(2)
 
+# Call model (streamlined, using generativeapis host to avoid HF Spaces cert issue)
 def call_model_with_file(prompt_text: str, uploaded_file_obj):
     if not API_KEY:
         raise RuntimeError("No API key provided")
@@ -200,8 +208,8 @@ def call_model_with_file(prompt_text: str, uploaded_file_obj):
         except Exception:
             pass
 
-    # Use the v1 Responses API host and normalized model_id (models/...)
-    endpoint = f"https://api.generativeai.googleapis.com/v1/{model_id}:generate"
+    # Use generativeapis.googleapis.com to avoid hostname mismatch in some environments (Hugging Face Spaces)
+    endpoint = f"https://generativeapis.googleapis.com/v1/{model_id}:generate"
     body = {
         "input": [
             {
@@ -218,6 +226,7 @@ def call_model_with_file(prompt_text: str, uploaded_file_obj):
     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.")
@@ -234,6 +243,7 @@ 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: