Update app.py

app.py

@@ -255,10 +255,10 @@ def detect_deepfake_video(video_path):
         if not ret:
             break
         
-        if frame_count % 10 == 0:  # ہر 10ویں فریم کا تجزیہ کریں
+        if frame_count % 10 == 0:
             frame_path = "temp_frame.jpg"
             cv2.imwrite(frame_path, frame)
-            result = detect_deepfake_image(frame_path)  # Deepfake detection function
+            result = detect_deepfake_image(frame_path)
             frame_scores.append(result["score"])
             os.remove(frame_path)
         
@@ -267,7 +267,7 @@ def detect_deepfake_video(video_path):
     cap.release()
     
     if not frame_scores:
-        return {"label": "UNKNOWN", "score": 0.0}  # اگر کوئی فریم پراسیس نہ ہو سکے
+        return {"label": "UNKNOWN", "score": 0.0}
     
     avg_score = np.mean(frame_scores)
     confidence = round(float(avg_score), 2)
@@ -291,19 +291,23 @@ def download_video(url):
         return None
 
 # Download YouTube video
-def get_youtube_embed_code(youtube_url):
-    """Generate an embedded iframe for YouTube videos"""
-    return f"""
-        <iframe width="560" height="315" src="https://www.youtube.com/embed/{youtube_url.split('=')[-1]}" 
-        frameborder="0" allowfullscreen></iframe>
-    """
+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  # Return local file path
+    except Exception as e:
+        return None
 
 # Select Video Source
 video_path = None
 video_display_url = None
-video_title = None
-video_thumbnail = None
-youtube_embed = None
 
 # Process Uploaded Video
 if uploaded_video is not None:
@@ -312,25 +316,25 @@ if uploaded_video is not None:
         f.write(uploaded_video.read())
     video_path = temp_file.name  # Set video path for detection
     video_display_url = temp_file.name
-    video_title = "Uploaded Video"
 
-# Process Video from URL (Check if YouTube or Direct Link)
+# Process Video from URL
 elif video_url:
     if "youtube.com" in video_url or "youtu.be" in video_url:
-        youtube_embed = get_youtube_embed_code(video_url)  # Get embedded iframe
-        video_title = "YouTube Video"
+        st.write("⏳ Downloading YouTube Video...")
+        video_path = download_youtube_video(video_url)  # Download YouTube video
+        if video_path:
+            st.success("✅ YouTube video downloaded successfully!")
+            video_display_url = video_path
+        else:
+            st.error("❌ Failed to download YouTube video. Try another link.")
     else:
         video_path = download_video(video_url)  # Download direct MP4 video
         video_display_url = video_path
-        video_title = "Online MP4 Video"
 
 # ✅ Display Video Properly
-if youtube_embed:
-    st.markdown(f"### 🎬 {video_title}")
-    st.markdown(youtube_embed, unsafe_allow_html=True)  # Show YouTube Video
-elif video_display_url:
-    st.markdown(f"### 🎬 {video_title}")
-    st.video(video_display_url)  # Show Uploaded or MP4 URL Video
+if video_display_url:
+    st.markdown("### 🎬 Video Preview")
+    st.video(video_display_url)  # Play Video
 
 # ✅ "Analyze Video" بٹن ہمیشہ شو ہوگا
 analyze_button = st.button("Analyze Video")