Update streamlit_app.py

streamlit_app.py

@@ -1,3 +1,4 @@
+# streamlit_app.py
 import os
 import time
 import string
@@ -60,7 +61,8 @@ st.session_state.setdefault("loop_video", False)
 st.session_state.setdefault("uploaded_file", None)
 st.session_state.setdefault("processed_file", None)
 st.session_state.setdefault("busy", False)
-st.session_state.setdefault("last_loaded_path", "")
+st.session_state.setdefault("last_url", "")
+st.session_state.setdefault("last_local_path", "")
 st.session_state.setdefault("analysis_out", "")
 st.session_state.setdefault("last_error", "")
 st.session_state.setdefault("file_hash", None)
@@ -188,7 +190,7 @@ def clear_all_video_state():
     st.session_state.pop("uploaded_file", None)
     st.session_state.pop("processed_file", None)
     st.session_state["videos"] = ""
-    st.session_state["last_loaded_path"] = ""
+    st.session_state["last_local_path"] = ""
     st.session_state["analysis_out"] = ""
     st.session_state["last_error"] = ""
     st.session_state["file_hash"] = None
@@ -198,11 +200,13 @@ def clear_all_video_state():
         except Exception:
             pass
 
-# Reset when URL changes
+# Reset when URL changes (compare against last_url only)
 current_url = st.session_state.get("url", "")
-if current_url != st.session_state.get("last_loaded_path"):
-    clear_all_video_state()
-    st.session_state["last_loaded_path"] = current_url
+if current_url != st.session_state.get("last_url"):
+    # avoid clearing on first load when last_url is empty
+    if st.session_state.get("last_url"):
+        clear_all_video_state()
+    st.session_state["last_url"] = current_url
 
 # ---- Sidebar UI ----
 st.sidebar.header("Video Input")
@@ -337,16 +341,25 @@ def compress_video_if_large(local_path: str, threshold_mb: int = 200):
 # ---- Simple layout ----
 col1, col2 = st.columns([1, 3])
 with col1:
-    generate_now = st.button("Generate the story", type="primary", disabled=not bool(get_effective_api_key()))
+    generate_now = st.button(
+        "Generate the story",
+        type="primary",
+        disabled=not (bool(get_effective_api_key()) and bool(st.session_state.get("videos")) and not st.session_state.get("busy"))
+    )
 with col2:
-    st.write("")
+    # small UX note column
+    if not st.session_state.get("videos"):
+        st.info("Load a video first (sidebar) to enable generation.", icon="ℹ️")
+    else:
+        st.write("")
 
 if st.sidebar.button("Load Video", use_container_width=True):
     try:
         vpw = st.session_state.get("video-password", "")
         path = download_video_ytdlp(st.session_state.get("url", ""), str(DATA_DIR), vpw)
         st.session_state["videos"] = path
-        st.session_state["last_loaded_path"] = path
+        st.session_state["last_local_path"] = path
+        # keep last_url intact — it tracks the input URL
         st.session_state.pop("uploaded_file", None)
         st.session_state.pop("processed_file", None)
         try:
@@ -389,7 +402,7 @@ if st.session_state["videos"]:
     except Exception:
         pass
 
-# ---- Generation flow (minimal) ----
+# ---- Generation flow (minimal, robust) ----
 if generate_now and not st.session_state.get("busy"):
     if not st.session_state.get("videos"):
         st.error("No video loaded. Use 'Load Video' in the sidebar.")
@@ -421,7 +434,8 @@ if generate_now and not st.session_state.get("busy"):
                 reupload_needed = True
                 uploaded_file = st.session_state.get("uploaded_file")
                 uploaded_name = file_name_or_id(uploaded_file)
-                if processed and st.session_state.get("last_loaded_path") == current_path and st.session_state.get("file_hash") == current_hash and uploaded_name:
+                # Use last_local_path to determine if file changed
+                if processed and st.session_state.get("last_local_path") == current_path and st.session_state.get("file_hash") == current_hash and uploaded_name:
                     reupload_needed = False
 
                 if reupload_needed:
@@ -455,7 +469,7 @@ if generate_now and not st.session_state.get("busy"):
 
                     st.session_state["uploaded_file"] = uploaded
                     st.session_state["processed_file"] = processed
-                    st.session_state["last_loaded_path"] = current_path
+                    st.session_state["last_local_path"] = current_path
                     st.session_state["file_hash"] = current_hash
 
                 prompt_text = (analysis_prompt.strip() or DEFAULT_PROMPT).strip()
@@ -598,7 +612,7 @@ if generate_now and not st.session_state.get("busy"):
                 st.caption(f"Est. max tokens: {est_tokens}")
 
             except Exception as e:
-                st.session_state["last_error"] = f"{e}"
+                st.session_state["last_error"] = f"{e}\n{traceback.format_exc()}"
                 st.error("An error occurred while generating the story. You can try Generate again; the uploaded video will be reused.")
             finally:
                 st.session_state["busy"] = False