Update streamlit_app.py

streamlit_app.py

@@ -250,6 +250,19 @@ def _strip_prompt_echo(prompt: str, text: str, threshold: float = 0.68) -> str:
 # ----------------------------------------------------------------------
 # Streamlit UI
 # ----------------------------------------------------------------------
+import os
+import traceback
+from pathlib import Path
+
+import streamlit as st
+import genai  # your Gemini wrapper
+
+# import your helpers:
+# download_video, _convert_to_mp4, _maybe_compress,
+# generate_report, _strip_prompt_echo, MODEL_OPTIONS,
+# DEFAULT_MODEL, DEFAULT_PROMPT, DATA_DIR
+
+
 def main() -> None:
     st.set_page_config(page_title="Video Analysis", layout="wide")
 
@@ -257,59 +270,30 @@ def main() -> None:
     st.sidebar.header("Video Input")
     st.sidebar.text_input("Video URL", key="url", placeholder="https://")
 
-    # **Load Video button – now directly under the URL field**
+    # ---- Load Video button (directly under the URL) ----
     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))
+                mp4_path = _convert_to_mp4(Path(raw_path))   # always MP4
             st.session_state["video_path"] = str(mp4_path)   # guaranteed MP4
             st.session_state["last_error"] = ""
             st.success("Video loaded successfully.")
+            # Force Streamlit to rerun so the preview appears immediately
+            st.experimental_rerun()
         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.get("video_path"):
-            try:
-                mp4 = _convert_to_mp4(Path(st.session_state["video_path"]))
-                st.sidebar.video(str(mp4))
-            except Exception:
-                st.sidebar.write("Preview unavailable")
-
-            if st.sidebar.button("Clear Video"):
-                # delete every file in DATA_DIR (both raw and converted)
-                for f in DATA_DIR.iterdir():
-                    try:
-                        f.unlink()
-                    except Exception:
-                        pass
-                # reset session state, including the URL field
-                st.session_state.update(
-                    {
-                        "url": "",
-                        "video_path": "",
-                        "analysis_out": "",
-                        "last_error": "",
-                        "busy": False,
-                        "show_raw_on_error": False,
-                    }
-                )
-                st.success("Session cleared.")
-
     # ---------- Settings ----------
     with st.sidebar.expander("Settings", expanded=False):
         model = st.selectbox(
             "Model", MODEL_OPTIONS, index=MODEL_OPTIONS.index(DEFAULT_MODEL)
         )
         if model == "custom":
-            model = st.text_input(
-                "Custom model ID", value=DEFAULT_MODEL, key="custom_model"
-            )
+            model = st.text_input("Custom model ID", value=DEFAULT_MODEL, key="custom_model")
         st.session_state["model_input"] = model
 
         # API key handling
@@ -318,11 +302,8 @@ def main() -> None:
             st.session_state["api_key"] = secret_key
         st.text_input("Google API Key", key="api_key", type="password")
 
-        st.text_area(
-            "Analysis prompt", value=DEFAULT_PROMPT, key="prompt", height=140
-        )
+        st.text_area("Analysis prompt", value=DEFAULT_PROMPT, key="prompt", height=140)
         st.text_input("Video password (if needed)", key="video_password", type="password")
-
         st.number_input(
             "Compress if > (MB)",
             min_value=10,
@@ -332,12 +313,43 @@ def main() -> None:
             key="compress_mb",
         )
 
+    # ---------- Preview & clear (shown only when a video is loaded) ----------
+    if st.session_state.get("video_path"):
+        try:
+            mp4 = _convert_to_mp4(Path(st.session_state["video_path"]))
+            st.sidebar.video(str(mp4))
+        except Exception:
+            st.sidebar.write("Preview unavailable")
+
+        if st.sidebar.button("Clear Video"):
+            # delete every file in DATA_DIR (both raw and converted)
+            for f in DATA_DIR.iterdir():
+                try:
+                    f.unlink()
+                except Exception:
+                    pass
+            # reset session state (including the URL field)
+            st.session_state.update(
+                {
+                    "url": "",
+                    "video_path": "",
+                    "analysis_out": "",
+                    "last_error": "",
+                    "busy": False,
+                    "show_raw_on_error": False,
+                }
+            )
+            st.success("Session cleared.")
+            st.experimental_rerun()
+
     # ---------- Generation ----------
     col1, col2 = st.columns([1, 3])
 
     with col1:
         generate_now = st.button(
-            "Generate analysis", type="primary", disabled=st.session_state.get("busy", False)
+            "Generate analysis",
+            type="primary",
+            disabled=st.session_state.get("busy", False),
         )
 
     with col2:
@@ -400,10 +412,15 @@ def main() -> None:
                 st.subheader("📝 Analysis")
                 st.markdown(st.session_state["analysis_out"])
 
-            # Show raw Gemini output only when it exists
+            # Full Gemini output – collapsed by default, expanded only on error
             if st.session_state.get("raw_output"):
-                st.subheader("🔎 Full Gemini output")
-                st.code(st.session_state["raw_output"], language="text")
+                if st.session_state.get("show_raw_on_error"):
+                    st.subheader("🔎 Full Gemini output")
+                    st.code(st.session_state["raw_output"], language="text")
+                else:
+                    # collapsed expander, starts closed
+                    with st.expander("🔎 Full Gemini output (collapsed)"):
+                        st.code(st.session_state["raw_output"], language="text")
 
             # ---------- Errors ----------
             if st.session_state.get("last_error"):