Update streamlit_app.py

streamlit_app.py

@@ -285,22 +285,24 @@ def main() -> None:
             key="compress_mb",
         )
 
-    # ---------- Load video ----------
-    if st.sidebar.button("Load Video"):
-        try:
-            with st.spinner("Downloading video…"):
-                path = download_video(
-                    st.session_state["url"], DATA_DIR, st.session_state["video_password"]
-                )
-            st.session_state["video_path"] = str(path)
-            st.session_state["last_error"] = ""
-            st.success("Video loaded successfully.")
-        except Exception as e:
-            st.session_state["last_error"] = f"Download failed: {e}"
-            st.sidebar.error(st.session_state["last_error"])
+# ---------- Load video ----------
+if st.sidebar.button("Load Video"):
+    try:
+        with st.spinner("Downloading video…"):
+            raw_path = download_video(
+                st.session_state["url"], DATA_DIR, st.session_state["video_password"]
+            )
+            # ALWAYS convert to MP4 before storing
+            mp4_path = _convert_to_mp4(Path(raw_path))
+        st.session_state["video_path"] = str(mp4_path)   # guaranteed MP4
+        st.session_state["last_error"] = ""
+        st.success("Video loaded successfully.")
+    except Exception as e:
+        st.session_state["last_error"] = f"Download failed: {e}"
+        st.sidebar.error(st.session_state["last_error"])
 
     # ---------- Preview & clear ----------
-    if st.session_state["video_path"]:
+    if st.session_state.get("video_path"):
         try:
             mp4 = _convert_to_mp4(Path(st.session_state["video_path"]))
             st.sidebar.video(str(mp4))
@@ -308,13 +310,13 @@ def main() -> None:
             st.sidebar.write("Preview unavailable")
 
         if st.sidebar.button("Clear Video"):
-            # delete files
+            # delete every file in DATA_DIR (both raw and converted)
             for f in DATA_DIR.iterdir():
                 try:
                     f.unlink()
                 except Exception:
                     pass
-            # reset state, including URL field
+            # reset session state, including the URL field
             st.session_state.update(
                 {
                     "url": "",
@@ -322,23 +324,27 @@ def main() -> None:
                     "analysis_out": "",
                     "last_error": "",
                     "busy": False,
+                    "show_raw_on_error": False,
                 }
             )
             st.success("Session cleared.")
 
+
 # ---------- Generation ----------
 col1, col2 = st.columns([1, 3])
+
 with col1:
     generate_now = st.button(
-        "Generate analysis", type="primary", disabled=st.session_state["busy"]
+        "Generate analysis", type="primary", disabled=st.session_state.get("busy", False)
     )
+
 with col2:
-    if not st.session_state["video_path"]:
+    if not st.session_state.get("video_path"):
         st.info("Load a video first.", icon="ℹ️")
 
-if generate_now and not st.session_state["busy"]:
-    api_key = st.session_state["api_key"] or os.getenv("GOOGLE_API_KEY")
-    if not st.session_state["video_path"]:
+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")
+    if not st.session_state.get("video_path"):
         st.error("No video loaded.")
     elif not api_key:
         st.error("Google API key missing.")
@@ -376,31 +382,35 @@ if generate_now and not st.session_state["busy"]:
             st.session_state["analysis_out"] = out
             st.success("Analysis generated.")
             st.markdown(out or "*(no output)*")
+
         except Exception as exc:
             tb = traceback.format_exc()
             # keep both traceback and whatever raw output we might have
-            st.session_state["last_error_detail"] = f"{tb}\n\nRaw Gemini output:\n{st.session_state.get('raw_output','')}"
+            st.session_state["last_error_detail"] = (
+                f"{tb}\n\nRaw Gemini output:\n{st.session_state.get('raw_output', '')}"
+            )
             st.session_state["last_error"] = f"Generation error: {exc}"
             # indicate that raw output should be shown in the main area
             st.session_state["show_raw_on_error"] = True
             st.error("An error occurred during generation.")
+
         finally:
             st.session_state["busy"] = False
 
-                # ---------- Results ----------
-                if st.session_state.get("analysis_out"):
-                    st.subheader("📝 Analysis")
-                    st.markdown(st.session_state["analysis_out"])
+        # ---------- Results ----------
+        if st.session_state.get("analysis_out"):
+            st.subheader("📝 Analysis")
+            st.markdown(st.session_state["analysis_out"])
 
-                # Show raw Gemini output only when it exists
-                if st.session_state.get("raw_output"):
-                    st.subheader("🔎 Full Gemini output")
-                    st.code(st.session_state["raw_output"], language="text")
+        # Show raw Gemini output only when it exists
+        if st.session_state.get("raw_output"):
+            st.subheader("🔎 Full Gemini output")
+            st.code(st.session_state["raw_output"], language="text")
 
-                # ---------- Errors ----------
-                if st.session_state.get("last_error"):
-                    with st.expander("❗️ Error details"):
-                        st.code(st.session_state["last_error_detail"], language="text")
+        # ---------- Errors ----------
+        if st.session_state.get("last_error"):
+            with st.expander("❗️ Error details"):
+                st.code(st.session_state["last_error_detail"], language="text")
 
 
 if __name__ == "__main__":