Upload 4 files

app.py

@@ -762,15 +762,14 @@ def gradio_process(pdf_file, mode_str):
         include_images = (mode_str != "markdown")
         include_markdown = (mode_str != "images")
         
-        # Process (sync for Gradio simplicity, or use utils)
-        # We need to load model if not loaded
-        load_model_once()
-        extractor.USE_MULTIPROCESSING = False # Gradio usually runs in thread
+        # Process using the multiprocessing pool for speed
+        # The global pool is already initialized in lifespan
+        extractor.USE_MULTIPROCESSING = True
         
         extractor.process_pdf_with_pool(
             dest_path,
             output_dir,
-            pool=None, 
+            pool=_pool,  # Use the global pool instead of None
             extract_images=include_images,
             extract_markdown=include_markdown
         )
@@ -828,40 +827,18 @@ with gr.Blocks(title="PDF Layout Extractor") as demo:
 
 
 # --------------------------------------------------------------------------------
-# Integrate FastAPI routes into Gradio's app
+# Integrate Gradio with FastAPI
 # --------------------------------------------------------------------------------
-# Gradio creates a FastAPI app internally. We add our API routes to it.
-# This approach works better on HF Spaces than mounting Gradio on our FastAPI app.
-
-# Access Gradio's internal FastAPI app
-fastapi_app = demo.fastapi_app
-
-# Add our existing routes to Gradio's FastAPI instance
-# We need to re-register all our routes on this app
-fastapi_app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
+# Mount Gradio at /gradio path (this ensures static files work correctly)
+app = gr.mount_gradio_app(app, demo, path="/gradio")
 
-# Mount static output directory
-fastapi_app.mount("/output", StaticFiles(directory="output"), name="output")
-
-# Re-register all API routes
-fastapi_app.get("/api/docs", response_class=HTMLResponse, tags=["UI"], include_in_schema=False)(api_docs_redirect)
-fastapi_app.get("/api/device-info", response_model=DeviceInfo, tags=["System"])(device_info_endpoint)
-fastapi_app.post("/api/upload", response_model=TaskStartResponse, tags=["Processing"])(upload_files)
-fastapi_app.get("/api/progress/{task_id}", response_model=ProgressResponse, tags=["Processing"])(get_progress)
-fastapi_app.get("/api/pdf-list", response_model=PDFListResponse, tags=["Retrieval"])(pdf_list)
-fastapi_app.get("/api/pdf-details/{pdf_stem}", tags=["Retrieval"])(pdf_details)
-fastapi_app.post("/api/predict", tags=["Legacy"])(predict)
-fastapi_app.post("/api/delete", tags=["Processing"])(delete_pdf)
-
-# For HF Spaces: expose the Gradio demo as 'app'
-app = demo
+# Redirect root to Gradio interface
+@app.get("/", response_class=HTMLResponse, include_in_schema=False)
+async def root_redirect():
+    """Redirect to Gradio interface."""
+    return HTMLResponse('<meta http-equiv="refresh" content="0; url=/gradio" />')
 
 if __name__ == "__main__":
-    # For local development
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    import uvicorn
+    uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=False)
+