Update app.py

app.py

@@ -200,7 +200,30 @@ class ForgeryDetector:
         """
         # Handle file path input (from gr.Image with type="filepath")
         if isinstance(image, str):
-            if image.lower().endswith('.pdf'):
+            if image.lower().endswith(('.doc', '.docx')):
+                # Handle Word documents - convert to PDF then to image
+                try:
+                    from docx2pdf import convert
+                    import tempfile
+                    import os
+                    
+                    temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix='.pdf')
+                    temp_pdf.close()
+                    convert(image, temp_pdf.name)
+                    
+                    import fitz
+                    pdf_document = fitz.open(temp_pdf.name)
+                    page = pdf_document[0]
+                    pix = page.get_pixmap(matrix=fitz.Matrix(2, 2))
+                    image = np.frombuffer(pix.samples, dtype=np.uint8).reshape(pix.height, pix.width, pix.n)
+                    if pix.n == 4:
+                        image = cv2.cvtColor(image, cv2.COLOR_RGBA2RGB)
+                    pdf_document.close()
+                    os.unlink(temp_pdf.name)
+                except Exception as e:
+                    raise ValueError(f"Could not process Word document: {str(e)}")
+                    
+            elif image.lower().endswith('.pdf'):
                 # Handle PDF files
                 import fitz  # PyMuPDF
                 pdf_document = fitz.open(image)
@@ -469,8 +492,8 @@ with gr.Blocks(css=custom_css) as demo:
             with gr.Tabs():
                 with gr.Tab("📤 Upload File"):
                     input_file = gr.File(
-                        label="Upload Image or PDF",
-                        file_types=["image", ".pdf"],
+                        label="Upload Image, PDF, or Document",
+                        file_types=["image", ".pdf", ".doc", ".docx"],
                         type="filepath"
                     )