Update app.py

app.py

@@ -89,6 +89,7 @@ import numpy as np
 import cv2
 import tempfile
 import requests
+from pytube import YouTube
 import os
 from PIL import Image
 import tensorflow as tf
@@ -373,8 +374,8 @@ st.subheader("🎥 Deepfake Video Detection")
 # Upload video file
 uploaded_video = st.file_uploader("Upload a Video", type=["mp4", "avi", "mov"])
 
-# URL Input for Video
-video_url = st.text_input("Enter Video URL (MP4 only)")
+# URL Input for Video (MP4 Direct Link or YouTube URL)
+video_url = st.text_input("Enter Video URL (YouTube or MP4 Link)")
 
 # Function to detect deepfake in video
 def detect_deepfake_video(video_path):
@@ -407,7 +408,7 @@ def detect_deepfake_video(video_path):
     
     return {"label": final_label, "score": confidence}
 
-# Download video from URL
+# Download video from direct MP4 URL
 def download_video(url):
     try:
         temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
@@ -422,6 +423,19 @@ def download_video(url):
     except Exception as e:
         return None
 
+# Download YouTube video
+def download_youtube_video(youtube_url):
+    try:
+        yt = YouTube(youtube_url)
+        video_stream = yt.streams.filter(file_extension="mp4", res="360p").first()
+        if not video_stream:
+            video_stream = yt.streams.get_lowest_resolution()
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+        video_stream.download(filename=temp_file.name)
+        return temp_file.name
+    except Exception as e:
+        return None
+
 # Select Video Source
 video_path = None
 
@@ -433,10 +447,14 @@ if uploaded_video is not None:
         f.write(uploaded_video.read())
     video_path = temp_file.name  # Set video path for detection
 
-# Process Video from URL
+# Process Video from URL (Check if YouTube or Direct Link)
 elif video_url:
-    st.video(video_url)  # Show the video directly from URL
-    video_path = download_video(video_url)  # Download the video for analysis
+    if "youtube.com" in video_url or "youtu.be" in video_url:
+        st.video(video_url)  # Show YouTube video
+        video_path = download_youtube_video(video_url)  # Download YouTube video
+    else:
+        st.video(video_url)  # Show direct MP4 video
+        video_path = download_video(video_url)  # Download direct MP4 video
 
 # Ensure video_path is valid before analysis
 if video_path and os.path.exists(video_path):