Add video download

app.py

@@ -1,12 +1,28 @@
+import os
+import gdown
 import gradio as gr
 from weaviate_explorer import WeaviateExplorer
 
 weaviate_explorer = WeaviateExplorer()
 video_player = gr.Video(label="Search Results")
 
+downloads_dir = "downloads"  # Directory to store downloaded videos
+
+# Create the downloads directory if it doesn't exist
+os.makedirs(downloads_dir, exist_ok=True)
+
+def remove_downloaded_videos():
+    """Remove all downloaded videos from the 'downloads' directory."""
+    for file_name in os.listdir(downloads_dir):
+        file_path = os.path.join(downloads_dir, file_name)
+        if os.path.isfile(file_path):
+            os.remove(file_path)
+
 def search(query):
+    remove_downloaded_videos()  # Remove previously downloaded videos
     shorts = weaviate_explorer.explore(query)
-    video_player.value = shorts[0]["link"]
+    video_path = gdown.download(shorts[0]["link"], output=downloads_dir, quiet=False)
+    video_player.value = video_path
     return video_player, "\n\n".join([f"{r['title']}\n{r['link']}" for r in shorts])
 
 iface = gr.Interface(fn=search,