Update streamlit_app.py

streamlit_app.py

@@ -224,11 +224,17 @@ def generate_report(video_path: Path, prompt: str, model_id: str, timeout: int =
 def _strip_prompt_echo(prompt: str, text: str, threshold: float = 0.68) -> str:
     if not prompt or not text:
         return text
+    
+    # Normalize the prompt and the response text
     clean_prompt = " ".join(prompt.lower().split())
-    snippet = " ".join(text.lower().split()[:600])
-    if SequenceMatcher(None, clean_prompt, snippet).ratio() > threshold:
-        cut = max(len(clean_prompt), int(len(prompt) * 0.9))
-        return text[cut:].lstrip(" \n:-")
+    lower_text = text.lower()
+
+    # Check if the start of the response matches the prompt
+    if lower_text.startswith(clean_prompt):
+        # If it matches, remove the prompt section from the start
+        return text[len(prompt):].lstrip(" \n:-")
+    
+    # If there is no significant match, return the text as is
     return text
 
 # ----------------------------------------------------------------------
@@ -360,34 +366,37 @@ def main() -> None:
         )
 
     # ---------- Main panel ----------
-# Run Analysis button
-if st.button("Run Analysis"):
-    if not st.session_state.get("video_path"):
-        st.error("No video loaded – load a video first.")
-    elif not st.session_state.get("api_key"):
-        st.error("Google API key missing – enter it in the sidebar.")
-
-        st.session_state["busy"] = True
-        st.session_state["analysis_out"] = ""
-        st.session_state["raw_output"] = ""
-        st.session_state["last_error"] = ""
-        st.session_state["last_error_detail"] = ""
+    # Run Analysis button (placed after settings for visual flow)
+    if st.button("Run Analysis"):
+        if not st.session_state.get("video_path"):
+            st.error("No video loaded – load a video first.")
+        elif not st.session_state.get("api_key"):
+            st.error("Google API key missing – enter it in the sidebar.")
+        else:
+            # configure Gemini now that we have a key
+            genai.configure(api_key=st.session_state["api_key"])
+
+            st.session_state["busy"] = True
+            st.session_state["analysis_out"] = ""
+            st.session_state["raw_output"] = ""
+            st.session_state["last_error"] = ""
+            st.session_state["last_error_detail"] = ""
 
-        try:
-            with st.spinner("Generating report (this may take a minute)…"):
-                raw = generate_report(
-                    Path(st.session_state["video_path"]),
-                    st.session_state["prompt"],
-                    st.session_state["model_input"],
-                )
-                cleaned = _strip_prompt_echo(st.session_state["prompt"], raw)
-                st.session_state["analysis_out"] = cleaned
-                st.session_state["raw_output"] = raw
-        except Exception as e:
-            st.session_state["last_error"] = f"Analysis failed: {e}"
-            st.session_state["last_error_detail"] = traceback.format_exc()
-        finally:
-            st.session_state["busy"] = False
+            try:
+                with st.spinner("Generating report (this may take a minute)…"):
+                    raw = generate_report(
+                        Path(st.session_state["video_path"]),
+                        st.session_state["prompt"],
+                        st.session_state["model_input"],
+                    )
+                    cleaned = _strip_prompt_echo(st.session_state["prompt"], raw)
+                    st.session_state["analysis_out"] = cleaned
+                    st.session_state["raw_output"] = raw
+            except Exception as e:
+                st.session_state["last_error"] = f"Analysis failed: {e}"
+                st.session_state["last_error_detail"] = traceback.format_exc()
+            finally:
+                st.session_state["busy"] = False
 
     # ---- Layout: analysis first, then video, then errors ----
     if st.session_state.get("analysis_out"):
@@ -412,4 +421,4 @@ if st.button("Run Analysis"):
 # ----------------------------------------------------------------------
 if __name__ == "__main__":
     # No need to call _init_state() here – it is invoked inside main()
-    main()
+    main()