Update app.py
Browse files
app.py
CHANGED
|
@@ -407,6 +407,24 @@ def detect_deepfake_video(video_path):
|
|
| 407 |
|
| 408 |
return {"label": final_label, "score": confidence}
|
| 409 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 410 |
# Process Uploaded Video
|
| 411 |
if uploaded_video is not None:
|
| 412 |
st.video(uploaded_video) # Show uploaded video
|
|
@@ -418,10 +436,7 @@ if uploaded_video is not None:
|
|
| 418 |
# Process Video from URL
|
| 419 |
elif video_url:
|
| 420 |
st.video(video_url) # Show the video directly from URL
|
| 421 |
-
video_path = video_url #
|
| 422 |
-
|
| 423 |
-
else:
|
| 424 |
-
video_path = None
|
| 425 |
|
| 426 |
# Analyze Button
|
| 427 |
if video_path and st.button("Analyze Video"):
|
|
|
|
| 407 |
|
| 408 |
return {"label": final_label, "score": confidence}
|
| 409 |
|
| 410 |
+
# Download video from URL
|
| 411 |
+
def download_video(url):
|
| 412 |
+
try:
|
| 413 |
+
temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
|
| 414 |
+
response = requests.get(url, stream=True)
|
| 415 |
+
if response.status_code == 200:
|
| 416 |
+
with open(temp_file.name, "wb") as f:
|
| 417 |
+
for chunk in response.iter_content(chunk_size=1024):
|
| 418 |
+
f.write(chunk)
|
| 419 |
+
return temp_file.name
|
| 420 |
+
else:
|
| 421 |
+
return None
|
| 422 |
+
except Exception as e:
|
| 423 |
+
return None
|
| 424 |
+
|
| 425 |
+
# Select Video Source
|
| 426 |
+
video_path = None
|
| 427 |
+
|
| 428 |
# Process Uploaded Video
|
| 429 |
if uploaded_video is not None:
|
| 430 |
st.video(uploaded_video) # Show uploaded video
|
|
|
|
| 436 |
# Process Video from URL
|
| 437 |
elif video_url:
|
| 438 |
st.video(video_url) # Show the video directly from URL
|
| 439 |
+
video_path = download_video(video_url) # Download the video for analysis
|
|
|
|
|
|
|
|
|
|
| 440 |
|
| 441 |
# Analyze Button
|
| 442 |
if video_path and st.button("Analyze Video"):
|