Abraham E. Tavarez
commited on
Commit
Β·
bd55878
1
Parent(s):
56225c5
deepface scanning offloaded to modal cloud function
Browse files- app.py +43 -49
- detector/face.py +13 -0
- pyproject.toml +2 -0
- uv.lock +0 -0
app.py
CHANGED
|
@@ -4,19 +4,30 @@ from detector.voice import verify_voices
|
|
| 4 |
from detector.video import verify_faces_in_video
|
| 5 |
from reports.pdf_report import generate_pdf_report
|
| 6 |
from utils.youtube_utils import download_youtube_video
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
# Holds latest results
|
| 9 |
last_face_result = None
|
| 10 |
last_voice_result = None
|
| 11 |
last_video_results = None
|
| 12 |
|
| 13 |
-
|
| 14 |
-
def
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
|
|
|
|
|
|
| 18 |
global last_face_result
|
| 19 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
result_text = ""
|
| 21 |
|
| 22 |
if "error" in result:
|
|
@@ -26,14 +37,19 @@ def compare_faces(img1_path, img2_path):
|
|
| 26 |
result_text = f"β
Match! Distance: {result['distance']:.4f} (Threshold: {result['threshold']})"
|
| 27 |
last_face_result = result_text
|
| 28 |
return result_text
|
| 29 |
-
|
| 30 |
else:
|
| 31 |
result_text = f"β No Match. Distance: {result['distance']:.4f} (Threshold: {result['threshold']})"
|
| 32 |
last_face_result = result_text
|
| 33 |
return result_text
|
| 34 |
|
| 35 |
|
| 36 |
-
def compare_voices(audio1, audio2):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
global last_voice_result
|
| 38 |
result = verify_voices(audio1, audio2)
|
| 39 |
result_text = ""
|
|
@@ -51,46 +67,35 @@ def compare_voices(audio1, audio2):
|
|
| 51 |
return result_text
|
| 52 |
|
| 53 |
|
| 54 |
-
def scan_video(video_file, ref_img, youtube_url=""):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 55 |
global last_video_results
|
| 56 |
-
|
| 57 |
if youtube_url:
|
| 58 |
try:
|
| 59 |
video_file = download_youtube_video(youtube_url)
|
| 60 |
except Exception as e:
|
| 61 |
return f"β Error downloading YouTube video: {str(e)}"
|
| 62 |
-
|
| 63 |
results = verify_faces_in_video(video_file, ref_img)
|
| 64 |
report = ""
|
| 65 |
-
last_video_results = results
|
| 66 |
for r in results:
|
| 67 |
if "error" in r:
|
| 68 |
report += f"\nβ οΈ Frame {r['frame']}: {r['error']}"
|
| 69 |
-
|
| 70 |
else:
|
| 71 |
status = "β
Match" if r["verified"] else "β Mismatch"
|
| 72 |
report += f"\nπΌ Frame {r['frame']}: {status} (Distance: {r['distance']})"
|
| 73 |
|
| 74 |
-
# last_video_results.append(report)
|
| 75 |
return report
|
| 76 |
|
| 77 |
|
| 78 |
-
# def scan_video(video_path, ref_img):
|
| 79 |
-
# global last_video_results
|
| 80 |
-
# results = verify_faces_in_video(video_path, ref_img)
|
| 81 |
-
# report = ""
|
| 82 |
-
# last_video_results = results
|
| 83 |
-
# for r in results:
|
| 84 |
-
# if "error" in r:
|
| 85 |
-
# report += f"\nβ οΈ Frame {r['frame']}: {r['error']}"
|
| 86 |
-
# # last_video_results.append(report)
|
| 87 |
-
# else:
|
| 88 |
-
# status = "β
Match" if r["verified"] else "β Mismatch"
|
| 89 |
-
# report += f"\nπΌ Frame {r['frame']}: {status} (Distance: {r['distance']})"
|
| 90 |
-
|
| 91 |
-
# # last_video_results.append(report)
|
| 92 |
-
# return report
|
| 93 |
-
|
| 94 |
def generate_report():
|
| 95 |
return generate_pdf_report(last_face_result, last_voice_result, last_video_results)
|
| 96 |
|
|
@@ -111,9 +116,7 @@ with gr.Blocks(title="Deepfake Watchdog") as demo:
|
|
| 111 |
output_text = gr.Textbox(label="Result")
|
| 112 |
# output_gallery = gr.Gallery(label="Matched Results")
|
| 113 |
|
| 114 |
-
run_button.click(
|
| 115 |
-
compare_faces, inputs=[image1, image2], outputs=[output_text]
|
| 116 |
-
)
|
| 117 |
|
| 118 |
# Voice Verification
|
| 119 |
with gr.Tab("π€ Voice Verification"):
|
|
@@ -126,30 +129,21 @@ with gr.Blocks(title="Deepfake Watchdog") as demo:
|
|
| 126 |
voice_output = gr.Textbox(label="Result")
|
| 127 |
|
| 128 |
voice_btn.click(compare_voices, inputs=[audio1, audio2], outputs=voice_output)
|
| 129 |
-
|
| 130 |
-
# Video DeepFake Scan
|
| 131 |
-
# gr.Markdown("### πΉ Video Deepfake Scan")
|
| 132 |
-
# with gr.Tab("πΉ Video Deepfake Scan"):
|
| 133 |
-
# gr.Markdown("Upload a video and a reference image. We'll scan for deepfake face mismatches.")
|
| 134 |
-
|
| 135 |
-
# ref_img = gr.Image(type="filepath", label="Reference Face")
|
| 136 |
-
# video_input = gr.Video(label="Video File")
|
| 137 |
-
# scan_btn = gr.Button("Scan Video")
|
| 138 |
-
# scan_output = gr.Textbox(label="Scan Results", lines=10)
|
| 139 |
|
| 140 |
-
# scan_btn.click(scan_video, inputs=[video_input, ref_img], outputs=scan_output)
|
| 141 |
-
|
| 142 |
with gr.Tab("πΉ Video Deepfake Scan"):
|
| 143 |
-
gr.Markdown(
|
| 144 |
-
|
|
|
|
|
|
|
| 145 |
ref_img = gr.Image(type="filepath", label="Reference Face")
|
| 146 |
video_input = gr.Video(label="Video File (optional)")
|
| 147 |
youtube_url = gr.Textbox(label="YouTube URL (optional)")
|
| 148 |
scan_btn = gr.Button("Scan Video")
|
| 149 |
scan_output = gr.Textbox(label="Scan Results", lines=10)
|
| 150 |
|
| 151 |
-
scan_btn.click(
|
| 152 |
-
|
|
|
|
| 153 |
|
| 154 |
with gr.Tab("π Generate Report"):
|
| 155 |
report_btn = gr.Button("Generate PDF Report")
|
|
|
|
| 4 |
from detector.video import verify_faces_in_video
|
| 5 |
from reports.pdf_report import generate_pdf_report
|
| 6 |
from utils.youtube_utils import download_youtube_video
|
| 7 |
+
import modal
|
| 8 |
+
verify_faces_remote = modal.Function.lookup("deepface-agent", "verify_faces_remote")
|
| 9 |
+
|
| 10 |
|
| 11 |
# Holds latest results
|
| 12 |
last_face_result = None
|
| 13 |
last_voice_result = None
|
| 14 |
last_video_results = None
|
| 15 |
|
| 16 |
+
# @app.local_entrypoint()
|
| 17 |
+
def compare_faces(img1_path: str, img2_path: str) -> str:
|
| 18 |
+
"""Use this tool to compare to faces for a match
|
| 19 |
+
Args:
|
| 20 |
+
img1_path: The path to the first image
|
| 21 |
+
img2_path: The path to the second image
|
| 22 |
+
"""
|
| 23 |
global last_face_result
|
| 24 |
+
|
| 25 |
+
# Read image files as bytes
|
| 26 |
+
with open(img1_path, "rb") as f1, open(img2_path, "rb") as f2:
|
| 27 |
+
img1_bytes = f1.read()
|
| 28 |
+
img2_bytes = f2.read()
|
| 29 |
+
|
| 30 |
+
result = verify_faces_remote.remote(img1_bytes, img2_bytes)
|
| 31 |
result_text = ""
|
| 32 |
|
| 33 |
if "error" in result:
|
|
|
|
| 37 |
result_text = f"β
Match! Distance: {result['distance']:.4f} (Threshold: {result['threshold']})"
|
| 38 |
last_face_result = result_text
|
| 39 |
return result_text
|
| 40 |
+
|
| 41 |
else:
|
| 42 |
result_text = f"β No Match. Distance: {result['distance']:.4f} (Threshold: {result['threshold']})"
|
| 43 |
last_face_result = result_text
|
| 44 |
return result_text
|
| 45 |
|
| 46 |
|
| 47 |
+
def compare_voices(audio1: str, audio2: str) -> str:
|
| 48 |
+
"""Use this tool to compare two voices for a match
|
| 49 |
+
Args:
|
| 50 |
+
audio1: The path to the first audio file
|
| 51 |
+
audio2: The path to the second audio file
|
| 52 |
+
"""
|
| 53 |
global last_voice_result
|
| 54 |
result = verify_voices(audio1, audio2)
|
| 55 |
result_text = ""
|
|
|
|
| 67 |
return result_text
|
| 68 |
|
| 69 |
|
| 70 |
+
def scan_video(video_file: str, ref_img: str, youtube_url="") -> str:
|
| 71 |
+
"""Use this tool to scan a video for deepfake face swaps
|
| 72 |
+
Args:
|
| 73 |
+
video_file: The path to the video file
|
| 74 |
+
ref_img: The path to the reference image
|
| 75 |
+
youtube_url: The YouTube URL (optional)
|
| 76 |
+
"""
|
| 77 |
global last_video_results
|
| 78 |
+
|
| 79 |
if youtube_url:
|
| 80 |
try:
|
| 81 |
video_file = download_youtube_video(youtube_url)
|
| 82 |
except Exception as e:
|
| 83 |
return f"β Error downloading YouTube video: {str(e)}"
|
| 84 |
+
|
| 85 |
results = verify_faces_in_video(video_file, ref_img)
|
| 86 |
report = ""
|
| 87 |
+
last_video_results = results
|
| 88 |
for r in results:
|
| 89 |
if "error" in r:
|
| 90 |
report += f"\nβ οΈ Frame {r['frame']}: {r['error']}"
|
| 91 |
+
|
| 92 |
else:
|
| 93 |
status = "β
Match" if r["verified"] else "β Mismatch"
|
| 94 |
report += f"\nπΌ Frame {r['frame']}: {status} (Distance: {r['distance']})"
|
| 95 |
|
|
|
|
| 96 |
return report
|
| 97 |
|
| 98 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 99 |
def generate_report():
|
| 100 |
return generate_pdf_report(last_face_result, last_voice_result, last_video_results)
|
| 101 |
|
|
|
|
| 116 |
output_text = gr.Textbox(label="Result")
|
| 117 |
# output_gallery = gr.Gallery(label="Matched Results")
|
| 118 |
|
| 119 |
+
run_button.click(compare_faces, inputs=[image1, image2], outputs=[output_text])
|
|
|
|
|
|
|
| 120 |
|
| 121 |
# Voice Verification
|
| 122 |
with gr.Tab("π€ Voice Verification"):
|
|
|
|
| 129 |
voice_output = gr.Textbox(label="Result")
|
| 130 |
|
| 131 |
voice_btn.click(compare_voices, inputs=[audio1, audio2], outputs=voice_output)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 132 |
|
|
|
|
|
|
|
| 133 |
with gr.Tab("πΉ Video Deepfake Scan"):
|
| 134 |
+
gr.Markdown(
|
| 135 |
+
"π Upload a video or paste a YouTube link and we'll analyze it for deepfake face swaps."
|
| 136 |
+
)
|
| 137 |
+
|
| 138 |
ref_img = gr.Image(type="filepath", label="Reference Face")
|
| 139 |
video_input = gr.Video(label="Video File (optional)")
|
| 140 |
youtube_url = gr.Textbox(label="YouTube URL (optional)")
|
| 141 |
scan_btn = gr.Button("Scan Video")
|
| 142 |
scan_output = gr.Textbox(label="Scan Results", lines=10)
|
| 143 |
|
| 144 |
+
scan_btn.click(
|
| 145 |
+
scan_video, inputs=[video_input, ref_img, youtube_url], outputs=scan_output
|
| 146 |
+
)
|
| 147 |
|
| 148 |
with gr.Tab("π Generate Report"):
|
| 149 |
report_btn = gr.Button("Generate PDF Report")
|
detector/face.py
CHANGED
|
@@ -1,6 +1,19 @@
|
|
| 1 |
from deepface import DeepFace
|
| 2 |
import cv2
|
| 3 |
import os
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
|
| 5 |
def verify_faces(img1_path, img2_path, model_name="Facenet", detector_backend="opencv"):
|
| 6 |
"""
|
|
|
|
| 1 |
from deepface import DeepFace
|
| 2 |
import cv2
|
| 3 |
import os
|
| 4 |
+
from modal_app.modal_app import verify_faces_remote
|
| 5 |
+
from fastapi import UploadFile
|
| 6 |
+
|
| 7 |
+
def face_verify_tool(img1: UploadFile, img2: UploadFile):
|
| 8 |
+
img1_bytes = img1.file.read()
|
| 9 |
+
img2_bytes = img2.file.read()
|
| 10 |
+
|
| 11 |
+
results = verify_faces_remote(img1_bytes, img2_bytes)
|
| 12 |
+
|
| 13 |
+
return results
|
| 14 |
+
|
| 15 |
+
|
| 16 |
+
|
| 17 |
|
| 18 |
def verify_faces(img1_path, img2_path, model_name="Facenet", detector_backend="opencv"):
|
| 19 |
"""
|
pyproject.toml
CHANGED
|
@@ -8,8 +8,10 @@ dependencies = [
|
|
| 8 |
"deepface>=0.0.93",
|
| 9 |
"fpdf>=1.7.2",
|
| 10 |
"gradio[mcp]>=5.32.1",
|
|
|
|
| 11 |
"opencv-python-headless>=4.11.0.86",
|
| 12 |
"pydub>=0.25.1",
|
| 13 |
"resemblyzer>=0.1.4",
|
| 14 |
"tf-keras>=2.19.0",
|
|
|
|
| 15 |
]
|
|
|
|
| 8 |
"deepface>=0.0.93",
|
| 9 |
"fpdf>=1.7.2",
|
| 10 |
"gradio[mcp]>=5.32.1",
|
| 11 |
+
"modal>=1.0.3",
|
| 12 |
"opencv-python-headless>=4.11.0.86",
|
| 13 |
"pydub>=0.25.1",
|
| 14 |
"resemblyzer>=0.1.4",
|
| 15 |
"tf-keras>=2.19.0",
|
| 16 |
+
"yt-dlp>=2025.5.22",
|
| 17 |
]
|
uv.lock
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|