youtube video scan

app.py

@@ -3,6 +3,7 @@ from detector.face import verify_faces, analyze_face
 from detector.voice import verify_voices
 from detector.video import verify_faces_in_video
 from reports.pdf_report import generate_pdf_report
+from utils.youtube_utils import download_youtube_video
 
 # Holds latest results
 last_face_result = None
@@ -50,9 +51,16 @@ def compare_voices(audio1, audio2):
         return result_text
 
 
-def scan_video(video_path, ref_img):
+def scan_video(video_file, ref_img, youtube_url=""):
     global last_video_results
-    results = verify_faces_in_video(video_path, ref_img)
+    
+    if youtube_url:
+        try:
+            video_file = download_youtube_video(youtube_url)
+        except Exception as e:
+            return f"❌ Error downloading YouTube video: {str(e)}"
+    
+    results = verify_faces_in_video(video_file, ref_img)
     report = ""
     last_video_results = results 
     for r in results:
@@ -66,6 +74,23 @@ def scan_video(video_path, ref_img):
             # last_video_results.append(report)
     return report
 
+
+# def scan_video(video_path, ref_img):
+#     global last_video_results
+#     results = verify_faces_in_video(video_path, ref_img)
+#     report = ""
+#     last_video_results = results 
+#     for r in results:
+#         if "error" in r:
+#             report += f"\n⚠️ Frame {r['frame']}: {r['error']}"
+#             # last_video_results.append(report)
+#         else:
+#             status = "✅ Match" if r["verified"] else "❌ Mismatch"
+#             report += f"\n🖼 Frame {r['frame']}: {status} (Distance: {r['distance']})"
+
+#             # last_video_results.append(report)
+#     return report
+
 def generate_report():
     return generate_pdf_report(last_face_result, last_voice_result, last_video_results)
 
@@ -75,7 +100,6 @@ with gr.Blocks(title="Deepfake Watchdog") as demo:
     gr.Markdown("## Upload your image and/or voice to scan for deepfake misuse online.")
 
     # Face Verification
-    gr.Markdown("### 📷 Face Verification")
     with gr.Tab("Face Verification"):
         image1 = gr.Image(label="Upload your face", type="filepath")
         # audio = gr.Audio(label="Upload your voice (optional)", type="filepath")
@@ -92,7 +116,6 @@ with gr.Blocks(title="Deepfake Watchdog") as demo:
         )
 
     # Voice Verification
-    gr.Markdown("### 🎤 Voice Verification")
     with gr.Tab("🎤 Voice Verification"):
         gr.Markdown("Upload two audio files to check if the voices match.")
 
@@ -105,19 +128,29 @@ with gr.Blocks(title="Deepfake Watchdog") as demo:
         voice_btn.click(compare_voices, inputs=[audio1, audio2], outputs=voice_output)
         
     # Video DeepFake Scan
-    gr.Markdown("### 📹 Video Deepfake Scan")
-    with gr.Tab("📹 Video Deepfake Scan"):
-        gr.Markdown("Upload a video and a reference image. We'll scan for deepfake face mismatches.")
+    # gr.Markdown("### 📹 Video Deepfake Scan")
+    # with gr.Tab("📹 Video Deepfake Scan"):
+    #     gr.Markdown("Upload a video and a reference image. We'll scan for deepfake face mismatches.")
+
+    #     ref_img = gr.Image(type="filepath", label="Reference Face")
+    #     video_input = gr.Video(label="Video File")
+    #     scan_btn = gr.Button("Scan Video")
+    #     scan_output = gr.Textbox(label="Scan Results", lines=10)
 
+    #     scan_btn.click(scan_video, inputs=[video_input, ref_img], outputs=scan_output)
+    
+    with gr.Tab("📹 Video Deepfake Scan"):
+        gr.Markdown("🔍 Upload a video or paste a YouTube link and we'll analyze it for deepfake face swaps.")
+        
         ref_img = gr.Image(type="filepath", label="Reference Face")
-        video_input = gr.Video(label="Video File")
+        video_input = gr.Video(label="Video File (optional)")
+        youtube_url = gr.Textbox(label="YouTube URL (optional)")
         scan_btn = gr.Button("Scan Video")
         scan_output = gr.Textbox(label="Scan Results", lines=10)
 
-        scan_btn.click(scan_video, inputs=[video_input, ref_img], outputs=scan_output)
+        scan_btn.click(scan_video, inputs=[video_input, ref_img, youtube_url], outputs=scan_output)    
 
 
-    gr.Markdown("### 📄 Generate Report")
     with gr.Tab("📄 Generate Report"):
         report_btn = gr.Button("Generate PDF Report")
         report_output = gr.File(label="Download Report")

utils/youtube_utils.py

@@ -0,0 +1,21 @@
+import yt_dlp
+import os
+import uuid
+
+
+def download_youtube_video(url: str, output_dir: str = "/tmp"):
+    """Downloads the YouTube video and returns the local file path."""
+    video_id = str(uuid.uuid4())[:8]
+    output_path = os.path.join(output_dir, f"{video_id}.mp4")
+
+    ydl_opts = {
+        "format": "best[ext=mp4]/best",
+        "outtmpl": output_path,
+        "quiet": True,
+        "noplaylist": True,
+    }
+
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        ydl.download([url])
+
+    return output_path