Update app.py

app.py

@@ -316,9 +316,71 @@ if uploaded_image is not None:
             
             st.error(f"⚠️ Result: This image is a Deepfake. (Confidence: {result['score']:.2f})")
 
+# # ---- Deepfake Video Detection Section ----
+# st.subheader("🎥 Deepfake Video Detection")
+# uploaded_video = st.file_uploader("Upload a Video", type=["mp4", "avi", "mov"])
+
+# def detect_deepfake_video(video_path):
+#     cap = cv2.VideoCapture(video_path)
+#     frame_scores = []
+#     frame_count = 0
+    
+#     while cap.isOpened():
+#         ret, frame = cap.read()
+#         if not ret:
+#             break
+        
+#         if frame_count % 10 == 0:  # ہر 10ویں فریم کا تجزیہ کریں
+#             frame_path = "temp_frame.jpg"
+#             cv2.imwrite(frame_path, frame)
+#             result = detect_deepfake_image(frame_path)
+#             frame_scores.append(result["score"])
+#             os.remove(frame_path)
+        
+#         frame_count += 1
+    
+#     cap.release()
+    
+#     if not frame_scores:
+#         return {"label": "UNKNOWN", "score": 0.0}  # اگر کوئی فریم پراسیس نہ ہو سکے
+    
+#     avg_score = np.mean(frame_scores)
+#     confidence = round(float(avg_score), 2)
+#     final_label = "FAKE" if avg_score > 0.5 else "REAL"
+    
+#     return {"label": final_label, "score": confidence}
+
+# if uploaded_video is not None:
+#     st.video(uploaded_video)
+#     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+#     with open(temp_file.name, "wb") as f:
+#         f.write(uploaded_video.read())
+    
+#     if st.button("Analyze Video"):
+#         st.write("🔍 Processing... Please wait.")
+#         result = detect_deepfake_video(temp_file.name)
+        
+#         if result["label"] == "FAKE":
+#             st.error(f"⚠️ Deepfake Detected! This video appears to be FAKE. (Confidence: {result['score']:.2f})")
+#         elif result["label"] == "REAL":
+#             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.")
 # ---- Deepfake Video Detection Section ----
 st.subheader("🎥 Deepfake Video Detection")
+
 uploaded_video = st.file_uploader("Upload a Video", type=["mp4", "avi", "mov"])
+video_url = st.text_input("Or enter a video URL")
+
+def download_video(url):
+    temp_video = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+    response = requests.get(url, stream=True)
+    if response.status_code == 200:
+        with open(temp_video.name, "wb") as f:
+            for chunk in response.iter_content(chunk_size=8192):
+                f.write(chunk)
+        return temp_video.name
+    return None
 
 def detect_deepfake_video(video_path):
     cap = cv2.VideoCapture(video_path)
@@ -350,15 +412,25 @@ def detect_deepfake_video(video_path):
     
     return {"label": final_label, "score": confidence}
 
-if uploaded_video is not None:
-    st.video(uploaded_video)
-    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
-    with open(temp_file.name, "wb") as f:
-        f.write(uploaded_video.read())
-    
-    if st.button("Analyze Video"):
+if uploaded_video is not None or video_url:
+    if uploaded_video is not None:
+        st.video(uploaded_video)
+        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
+    elif video_url:
+        st.write("📥 Downloading video from URL...")
+        video_path = download_video(video_url)
+        if video_path:
+            st.video(video_path)
+        else:
+            st.error("⚠️ Failed to download video. Please check the URL and try again.")
+            video_path = None
+
+    if video_path and st.button("Analyze Video"):
         st.write("🔍 Processing... Please wait.")
-        result = detect_deepfake_video(temp_file.name)
+        result = detect_deepfake_video(video_path)
         
         if result["label"] == "FAKE":
             st.error(f"⚠️ Deepfake Detected! This video appears to be FAKE. (Confidence: {result['score']:.2f})")