Update app.py

app.py

@@ -6,18 +6,15 @@ import time
 # --- VirusTotal helper functions ---
 def scan_url_vt(url, api_key):
     headers = {"x-apikey": api_key}
-    # submit URL
     resp = requests.post(
         "https://www.virustotal.com/api/v3/urls", headers=headers, data={"url": url}
     )
     resp.raise_for_status()
     analysis_id = resp.json()["data"]["id"]
-    # poll until complete
     while True:
         time.sleep(5)
         status_resp = requests.get(
-            f"https://www.virustotal.com/api/v3/analyses/{analysis_id}",
-            headers=headers
+            f"https://www.virustotal.com/api/v3/analyses/{analysis_id}", headers=headers
         )
         status_resp.raise_for_status()
         attr = status_resp.json()["data"]["attributes"]
@@ -29,14 +26,13 @@ def scan_url_vt(url, api_key):
 def fetch_clean_videos(keywords, api_key, scan_enabled):
     query = " OR ".join([f"{kw.strip().replace(' ', '+')}" for kw in keywords.split(",")])
     ia_query = f"mediatype:(movies) AND ({query})"
-    # fetch default batch and cap at 50 items
     results = list(search_items(ia_query))[:50]
     clean_urls = []
     for res in results:
         item = get_item(res['identifier'])
         for f in item.files:
             fmt = f.get('format','').lower()
-            if fmt.startswith(('mpeg','mp4','avi','mov','webm')):
+            if fmt.startswith(('mpeg','mp4','avi','mov','webm', 'm4v')):
                 url = f"https://archive.org/download/{res['identifier']}/{f['name']}"
                 if scan_enabled and api_key:
                     try:
@@ -50,22 +46,26 @@ def fetch_clean_videos(keywords, api_key, scan_enabled):
     return clean_urls
 
 # --- Gradio UI setup ---
-def gui(keywords, api_key, scan_enabled):
-    """Perform search and optional VT scan, return list of clean URLs."""
-    urls = fetch_clean_videos(keywords, api_key, scan_enabled)
-    if not urls:
-        return "No clean videos found."
-    return "\n".join(urls)
-
 with gr.Blocks() as demo:
-    gr.Markdown("# 📼 IA Drone‑Strike Video Finder with VT Scan")
+    gr.Markdown("# 📼 IA Drone‑Strike Video Browser with VT Scan")
     with gr.Row():
         kw_input = gr.Textbox(label="Search keywords (comma-separated)", value="drone strike, military uav, kamikaze drone")
         key_input = gr.Textbox(label="VirusTotal API Key", type="password")
     scan_toggle = gr.Checkbox(label="Enable VirusTotal scan", value=True)
     run_btn = gr.Button("Search & Scan")
-    output = gr.Textbox(label="Results (URLs)", lines=10)
-    run_btn.click(fn=gui, inputs=[kw_input, key_input, scan_toggle], outputs=output)
+
+    url_dropdown = gr.Dropdown(label="Clean Video URLs", choices=[], interactive=True)
+    video_player = gr.Video(label="Video Player")
+
+    def search_and_populate(keywords, api_key, scan_enabled):
+        urls = fetch_clean_videos(keywords, api_key, scan_enabled)
+        return urls if urls else []
+
+    def update_player(selected_url):
+        return selected_url or None
+
+    run_btn.click(fn=search_and_populate, inputs=[kw_input, key_input, scan_toggle], outputs=url_dropdown)
+    url_dropdown.change(fn=update_player, inputs=url_dropdown, outputs=video_player)
 
 if __name__ == "__main__":
-    demo.launch()  # Hugging Face Spaces binds to 0.0.0.0, no file downloads
+    demo.launch()  # HF Spaces deployment