Make document processor ZeroGPU compatible

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import json
+import spaces
 
 from document_processor import DocumentProcessor
 
@@ -7,6 +7,7 @@ from document_processor import DocumentProcessor
 processor = DocumentProcessor()
 
 
+@spaces.GPU
 def process_document(file):
 
     if file is None:
@@ -15,19 +16,16 @@ def process_document(file):
         }
 
     try:
-
-        result = processor.process(file)
-
-        return result
+        return processor.process(file)
 
     except Exception as e:
-
         return {
             "error": str(e)
         }
 
 
 demo = gr.Interface(
+
     fn=process_document,
 
     inputs=gr.File(
@@ -43,7 +41,7 @@ demo = gr.Interface(
     title="AI Tutor Document Processor",
 
     description=(
-        "Upload a PDF and extract its "
+        "Upload a PDF and extract "
         "text and page structure."
     )
 )

how HEADdocument_processor.py

@@ -0,0 +1,38 @@
+from pathlib import Path
+import fitz
+
+
+class DocumentProcessor:
+
+    def process(self, file_path):
+
+        path = Path(file_path)
+
+        if path.suffix.lower() != ".pdf":
+            raise ValueError(
+                f"Unsupported file type: {path.suffix}"
+            )
+
+        return self.process_pdf(path)
+
+    def process_pdf(self, path):
+
+        document = fitz.open(path)
+
+        pages = []
+
+        for page_number, page in enumerate(document):
+
+            pages.append({
+                "page_number": page_number + 1,
+                "text": page.get_text()
+            })
+
+        return {
+            "document": {
+                "filename": path.name,
+                "format": "pdf",
+                "page_count": len(document)
+            },
+            "pages": pages
+        }

requirements.txt

@@ -1,2 +1,3 @@
 gradio
 pymupdf
+spaces