Update streamlit_app.py

streamlit_app.py

@@ -9,7 +9,7 @@ Features
 * Download videos from direct links, Twitter, or any site supported by yt‑dlp.
 * Convert to MP4 (ffmpeg) and compress if larger than a user‑defined threshold.
 * Send the video (base64‑encoded) + a custom prompt to Gemini‑Flash models.
-* Simple sidebar UI with clear‑video handling.
+* Simple sidebar UI with clear video handling.
 """
 
 # ----------------------------------------------------------------------
@@ -22,7 +22,7 @@ import string
 import traceback
 from pathlib import Path
 from typing import Tuple, Optional
-from difflib import SequenceMatcher  # <-- needed for prompt‑echo stripping
+from difflib import SequenceMatcher
 
 # ----------------------------------------------------------------------
 # Third‑party libraries
@@ -67,23 +67,27 @@ DEFAULT_PROMPT = (
 # ----------------------------------------------------------------------
 # Session‑state defaults (run once per session)
 # ----------------------------------------------------------------------
-DEFAULT_STATE = {
-    "url": "",
-    "video_path": "",
-    "model_input": DEFAULT_MODEL,
-    "prompt": DEFAULT_PROMPT,
-    "api_key": os.getenv("GOOGLE_API_KEY", "AIzaSyBiAW2GQLid0HGe9Vs_ReKwkwsSVNegNzs"),
-    "video_password": "",
-    "compress_mb": 200,
-    "busy": False,
-    "last_error": "",
-    "analysis_out": "",
-    "raw_output": "",
-    "last_error_detail": "",
-    "show_raw_on_error": False,
-}
-for k, v in DEFAULT_STATE.items():
-    st.session_state.setdefault(k, v)
+def _init_state() -> None:
+    defaults = {
+        "url": "",
+        "video_path": "",
+        "model_input": DEFAULT_MODEL,
+        "prompt": DEFAULT_PROMPT,
+        "api_key": os.getenv("GOOGLE_API_KEY", ""),
+        "video_password": "",
+        "compress_mb": 200,
+        "busy": False,
+        "last_error": "",
+        "analysis_out": "",
+        "raw_output": "",
+        "last_error_detail": "",
+        "show_raw_on_error": False,
+        "show_analysis": False,
+    }
+    for k, v in defaults.items():
+        st.session_state.setdefault(k, v)
+
+_init_state()
 
 # ----------------------------------------------------------------------
 # Helper utilities
@@ -168,11 +172,13 @@ def _download_with_yt_dlp(url: str, dst: Path, password: str = "") -> Path:
     except Exception as e:
         raise RuntimeError(f"yt‑dlp could not download the URL: {e}") from e
 
+    # yt‑dlp may return a dict with the final filename
     if isinstance(info, dict) and "id" in info:
         candidate = dst / f"{info['id']}.{info.get('ext', 'mp4')}"
         if candidate.exists():
             return _convert_to_mp4(candidate)
 
+    # Fallback: pick the newest file in the output folder
     files = list(dst.iterdir())
     if not files:
         raise RuntimeError("yt‑dlp did not produce any files.")
@@ -311,7 +317,9 @@ def main() -> None:
     if st.session_state.get("video_path"):
         try:
             mp4 = _convert_to_mp4(Path(st.session_state["video_path"]))
-            st.sidebar.video(str(mp4))
+            with open(mp4, "rb") as f:
+                video_bytes = f.read()
+            st.sidebar.video(video_bytes)
         except Exception:
             st.sidebar.write("Preview unavailable")
 
@@ -326,9 +334,11 @@ def main() -> None:
                     "url": "",
                     "video_path": "",
                     "analysis_out": "",
+                    "raw_output": "",
                     "last_error": "",
                     "busy": False,
                     "show_raw_on_error": False,
+                    "show_analysis": False,
                 }
             )
             st.success("Session cleared.")
@@ -347,7 +357,7 @@ def main() -> None:
             st.info("Load a video first.", icon="ℹ️")
 
     # ------------------------------------------------------------------
-    # Generation button handling (unchanged up to the end of the try/except)
+    # Generation handling
     # ------------------------------------------------------------------
     if generate_now and not st.session_state.get("busy", False):
         api_key = st.session_state.get("api_key") or os.getenv("GOOGLE_API_KEY")
@@ -386,7 +396,7 @@ def main() -> None:
 
                 cleaned = _strip_prompt_echo(st.session_state["prompt"], raw_out)
                 st.session_state["analysis_out"] = cleaned
-                st.session_state["show_analysis"] = True          # ← new flag
+                st.session_state["show_analysis"] = True
                 st.success("Analysis generated.")
                 st.markdown(cleaned or "*(no output)*")
 
@@ -402,12 +412,11 @@ def main() -> None:
                 st.session_state["busy"] = False
 
     # ------------------------------------------------------------------
-    # Results – displayed **once** after a successful generation
+    # Results display (once per successful generation)
     # ------------------------------------------------------------------
     if st.session_state.get("show_analysis"):
         st.subheader("📝 Analysis")
         st.markdown(st.session_state["analysis_out"])
-        # Reset the flag so the block won’t run on the next automatic rerun
         st.session_state["show_analysis"] = False
 
     # Full Gemini output – collapsed by default, expanded on error
@@ -424,5 +433,6 @@ def main() -> None:
         with st.expander("❗️ Error details"):
             st.code(st.session_state["last_error_detail"], language="text")
 
+
 if __name__ == "__main__":
-    main()
+    main()