Update streamlit_app.py

streamlit_app.py

@@ -346,6 +346,9 @@ def main() -> None:
         if not st.session_state.get("video_path"):
             st.info("Load a video first.", icon="ℹ️")
 
+    # ------------------------------------------------------------------
+    # Generation button handling (unchanged up to the end of the try/except)
+    # ------------------------------------------------------------------
     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"):
@@ -357,14 +360,14 @@ def main() -> None:
                 st.session_state["busy"] = True
                 genai.configure(api_key=api_key)
 
-                # Optional compression
+                # ----- optional compression -----
                 with st.spinner("Checking video size…"):
                     video_path, was_compressed = _maybe_compress(
                         Path(st.session_state["video_path"]),
                         st.session_state["compress_mb"],
                     )
 
-                # Generation
+                # ----- generation -----
                 with st.spinner("Generating analysis…"):
                     raw_out = generate_report(
                         video_path,
@@ -374,7 +377,7 @@ def main() -> None:
                     )
                     st.session_state["raw_output"] = raw_out
 
-                # Clean up compressed temporary file
+                # ----- clean up compressed file -----
                 if was_compressed:
                     try:
                         video_path.unlink()
@@ -383,6 +386,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.success("Analysis generated.")
                 st.markdown(cleaned or "*(no output)*")
 
@@ -397,25 +401,28 @@ def main() -> None:
             finally:
                 st.session_state["busy"] = False
 
-            # ---------- Results ----------
-            if st.session_state.get("analysis_out"):
-                st.subheader("📝 Analysis")
-                st.markdown(st.session_state["analysis_out"])
-
-            # Full Gemini output – collapsed by default, expanded on error
-            if st.session_state.get("raw_output"):
-                if st.session_state.get("show_raw_on_error"):
-                    st.subheader("🔎 Full Gemini output")
-                    st.code(st.session_state["raw_output"], language="text")
-                else:
-                    with st.expander("🔎 Full Gemini output (collapsed)"):
-                        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")
+    # ------------------------------------------------------------------
+    # Results – displayed **once** after a 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
+    if st.session_state.get("raw_output"):
+        if st.session_state.get("show_raw_on_error"):
+            st.subheader("🔎 Full Gemini output")
+            st.code(st.session_state["raw_output"], language="text")
+        else:
+            with st.expander("🔎 Full Gemini output (collapsed)"):
+                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")
 
 if __name__ == "__main__":
-    main()
+    main()