PDF report generator

app.py

@@ -2,12 +2,17 @@ import gradio as gr
 from detector.face import verify_faces, analyze_face
 from detector.voice import verify_voices
 from detector.video import verify_faces_in_video
+from report.report import generate_pdf_report
+
+# Holds latest results
+last_face_result = None
+last_voice_result = None
+last_video_results = []
 
 
 def start_scan(image, audio):
     return "Scanning in progress...", None
 
-
 def compare_faces(img1_path, img2_path):
     result = verify_faces(img1_path, img2_path)
 
@@ -45,6 +50,8 @@ def scan_video(video_path, ref_img):
 
     return report
 
+def generate_report():
+    return generate_pdf_report(last_face_result, last_voice_result, last_video_results)
 
 
 with gr.Blocks(title="Deepfake Watchdog") as demo:
@@ -91,5 +98,12 @@ with gr.Blocks(title="Deepfake Watchdog") as demo:
         scan_btn.click(scan_video, inputs=[video_input, ref_img], outputs=scan_output)
 
 
+    gr.Markdown("### 📄 Generate Report")
+    with gr.Tab("📄 Generate Report"):
+        report_btn = gr.Button("Generate PDF Report")
+        report_output = gr.File(label="Download Report")
+
+        report_btn.click(generate_report, outputs=report_output)
+
 
 demo.launch(mcp_server=True)

report/report.py

@@ -0,0 +1,57 @@
+from fpdf import FPDF
+import os
+import datetime
+
+class ReportGenerator(FPDF):
+    def header(self):
+        self.set_font("Arial", "B", 16)
+        self.cell(0, 10, "Deepfake Watchdog Report", ln=1, align="C")
+        self.ln(5)
+        
+    def add_section_title(self, title):
+        self.set_font("Arial", "B", 12)
+        self.cell(0, 10, title, ln=1, align="L")
+        self.ln(2)
+    
+    def add_text(self, text):
+        self.set_font("Arial", "", 12)
+        self.multi_cell(0, 8, text)
+        self.ln(2)
+        
+    def add_image(self, image_path, w=60):
+        if os.path.exists(image_path):
+            self.image(image_path, w=w)
+            self.ln(5)
+        else:
+            print(f"Image not found: {image_path}")
+            
+# Generate PDF Report            
+def generate_pdf_report(face_result, voice_result, video_results, output_path=f"report{datetime.datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.pdf"):
+    pdf = ReportGenerator()
+    pdf.add_page()
+
+    # Section 1: Face Verification
+    pdf.add_section_title("Face Verification Results")
+    pdf.add_text(face_result)
+    
+    # Section 2: Voice Verification
+    pdf.add_section_title("Voice Verification Results")
+    pdf.add_text(voice_result)
+    
+    # Section 3: Video Verification
+    pdf.add_section_title("Video Verification Results")
+    
+    for frame in video_results:
+        if "error" in frame:
+            pdf.add_text(f"{frame['frame']}: ERROR - {frame['error']}")
+        else:
+            status = "✅ Match" if frame["verified"] else "❌ Mismatch"
+            pdf.add_text(f"{frame['frame']} - {status} (Distance: {frame['distance']})")
+            
+            if not frame["verified"]:
+                pdf.add_image(frame["frame"], w=80)
+                
+    pdf.output(output_path)
+    print(f"PDF report generated at: {output_path}")
+    return output_path
+