Update app.py

app.py

@@ -275,7 +275,7 @@ def detect_deepfake_video(video_path):
     
     return {"label": final_label, "score": confidence}
 
-# Download video from direct MP4 URL
+# Download direct MP4 video
 def download_video(url):
     try:
         temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
@@ -290,58 +290,34 @@ 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  # Return local file path
-    except Exception as e:
-        return None
-
-# Select Video Source
+# ✅ Display Video (YouTube Embed or Local)
 video_path = None
-video_display_url = None
 
-# Process Uploaded Video
 if uploaded_video is not None:
     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
     with open(temp_file.name, "wb") as f:
         f.write(uploaded_video.read())
     video_path = temp_file.name  # Set video path for detection
-    video_display_url = temp_file.name
+    st.video(video_path)  # Show uploaded video
 
-# Process Video from URL
 elif video_url:
     if "youtube.com" in video_url or "youtu.be" in video_url:
-        st.write("⏳ Downloading YouTube Video...")
-        video_path = download_youtube_video(video_url)  # Download YouTube video
+        st.markdown(
+            f'<iframe width="560" height="315" src="{video_url.replace("watch?v=", "embed/")}" frameborder="0" allowfullscreen></iframe>',
+            unsafe_allow_html=True,
+        )
+    else:
+        video_path = download_video(video_url)  # Download MP4
         if video_path:
-            st.success("✅ YouTube video downloaded successfully!")
-            video_display_url = video_path
+            st.video(video_path)  # Show downloaded MP4
         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
+            st.warning("⚠️ Invalid MP4 URL.")
 
-# ✅ Display Video Properly
-if video_display_url:
-    st.markdown("### 🎬 Video Preview")
-    st.video(video_display_url)  # Play Video
+# ✅ "Analyze Video" Button (Only for Local/MP4)
+if uploaded_video or (video_url and not "youtube.com" in video_url):
+    analyze_button = st.button("Analyze Video")
 
-# ✅ "Analyze Video" بٹن ہمیشہ شو ہوگا
-analyze_button = st.button("Analyze Video")
-
-# 🚀 اگر ویڈیو موجود ہے تو تجزیہ کریں
-if analyze_button:
-    if video_path and os.path.exists(video_path):
+    if analyze_button and video_path:
         st.write("🔍 Processing... Please wait.")
         result = detect_deepfake_video(video_path)
         
@@ -351,7 +327,5 @@ if analyze_button:
             st.success(f"✅ This video appears to be REAL. (Confidence: {1 - result['score']:.2f})")
         else:
             st.warning("⚠️ Unable to analyze the video. Please try a different file.")
-    else:
-        st.warning("⚠️ Invalid video. Please check the file or URL.")
             
 st.markdown("🔹 **Developed for Fake News & Deepfake Detection**")