Update streamlit_app.py

streamlit_app.py

@@ -274,26 +274,85 @@ if st.sidebar.button("Load Video"):
         st.session_state["last_error"] = f"Download failed: {e}"
         st.sidebar.error(st.session_state["last_error"])
 
-st.sidebar.markdown("---")
-st.sidebar.subheader("🔎 Extract video from a Tweet")
-tweet_url = st.sidebar.text_input(
-    "Tweet URL (e.g. https://twitter.com/user/status/1234567890)",
-    key="tweet_url",
-)
+# --------------------------------------------------------------
+# 📥 Twitter video extractor – hidden inside an expander
+# --------------------------------------------------------------
+with st.sidebar.expander("🔎 Extract video(s) from a Tweet", expanded=False):
+    tweet_url = st.text_input(
+        "Tweet URL (e.g. https://twitter.com/user/status/1234567890)",
+        key="tweet_url",
+    )
+
+    # ------------------------------------------------------------------
+    # Step 1 – Scrape the tweet and collect every video URL
+    # ------------------------------------------------------------------
+    if st.button("Find videos in tweet"):
+        if not tweet_url:
+            st.error("Paste a tweet URL first.")
+        else:
+            try:
+                tweet_id = tweet_url.split("/")[-1].split("?")[0]
+                video_urls: list[str] = []
+
+                for tweet in sntwitter.TwitterTweetScraper(tweet_id).get_items():
+                    # media objects (video, gif, etc.)
+                    if getattr(tweet, "media", None):
+                        for m in tweet.media:
+                            if getattr(m, "video_url", None):
+                                video_urls.append(m.video_url)
+
+                    # expanded URLs that already end with a video extension
+                    for u in getattr(tweet, "urls", []):
+                        if u.expandedUrl.lower().endswith(
+                            (".mp4", ".mov", ".webm", ".mkv", ".avi", ".flv")
+                        ):
+                            video_urls.append(u.expandedUrl)
+
+                # deduplicate while preserving order
+                video_urls = list(dict.fromkeys(video_urls))
+
+                if not video_urls:
+                    raise RuntimeError("No video URLs detected in this tweet.")
+
+                # store a friendly label + raw URL for the selector
+                st.session_state["tweet_video_options"] = [
+                    (f"Video {i+1} – {url.split('/')[-1][:30]}...", url)
+                    for i, url in enumerate(video_urls)
+                ]
+                st.success(f"Found {len(video_urls)} video(s). Choose one below.")
+            except Exception as e:
+                st.session_state["tweet_video_options"] = []
+                st.error(f"Tweet scrape failed: {e}")
+
+    # ------------------------------------------------------------------
+    # Step 2 – Let the user pick one of the discovered videos
+    # ------------------------------------------------------------------
+    if st.session_state.get("tweet_video_options"):
+        display_names, raw_urls = zip(*st.session_state["tweet_video_options"])
+        sel_idx = st.selectbox(
+            "Select video to download",
+            options=range(len(display_names)),
+            format_func=lambda i: display_names[i],
+            key="tweet_video_select",
+        )
 
-if st.sidebar.button("Fetch video from tweet"):
-    if not tweet_url:
-        st.sidebar.error("Please paste a tweet URL.")
+        # ------------------------------------------------------------------
+        # Step 3 – Download the chosen video
+        # ------------------------------------------------------------------
+        if st.button("Download selected video"):
+            chosen_url = raw_urls[sel_idx]
+            try:
+                path = download_video(chosen_url, str(DATA_DIR))
+                st.session_state["videos"] = path
+                st.session_state["last_error"] = ""
+                st.success("Video downloaded and loaded.")
+            except Exception as e:
+                st.session_state["last_error"] = f"Download failed: {e}"
+                st.error(st.session_state["last_error"])
     else:
-        try:
-            # The same download_video function will recognise the tweet URL
-            path = download_video(tweet_url, str(DATA_DIR))
-            st.session_state["videos"] = path
-            st.session_state["last_error"] = ""
-            st.sidebar.success("Video extracted from tweet and loaded.")
-        except Exception as e:
-            st.session_state["last_error"] = f"Tweet extraction failed: {e}"
-            st.sidebar.error(st.session_state["last_error"])
+        st.info(
+            "Paste a tweet URL and click **Find videos in tweet** to discover available videos."
+        )
 
 if st.session_state["videos"]:
     try: