Fast API update 2

Dockerfile

@@ -0,0 +1,26 @@
+FROM python:3.9
+
+WORKDIR /code
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    curl \
+    software-properties-common \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements first to leverage Docker cache
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the application
+COPY . /code/
+
+# Make sure the inbox and output directories exist
+RUN mkdir -p /code/inbox /code/output
+
+# Expose the port
+EXPOSE 7860
+
+# Command to run the application
+CMD ["python", "app.py"] 

app.py

@@ -3,8 +3,8 @@ import os
 import shutil
 import logging
 import uvicorn
-from fastapi import FastAPI, File, UploadFile, BackgroundTasks
-from typing import List, Optional
+from fastapi import FastAPI, File, UploadFile
+from fastapi.middleware.cors import CORSMiddleware
 
 # This ensures our models are downloaded and config is set before anything
 # Alternatively you can do this in a "startup" event handler
@@ -16,11 +16,25 @@ from mineru_single import to_markdown
 app = FastAPI()
 logging.basicConfig(level=logging.INFO)
 
+# Add CORS middleware to allow requests from any origin
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allows all origins
+    allow_credentials=True,
+    allow_methods=["*"],  # Allows all methods
+    allow_headers=["*"],  # Allows all headers
+)
+
 INBOX_DIR = "./inbox"
 OUTPUT_DIR = "./output"
 os.makedirs(INBOX_DIR, exist_ok=True)
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
+@app.get("/")
+async def root():
+    """Health check endpoint"""
+    return {"status": "ok", "message": "API is running"}
+
 @app.post("/process")
 async def process_pdf(file: UploadFile = File(...)):
 

requirements.txt

@@ -23,4 +23,7 @@ rapid-table>=1.0.3,<2.0.0
 rapidocr-paddle
 rapidocr-onnxruntime
 gradio-pdf>=0.0.21
-openai
+openai
+fastapi
+uvicorn
+python-multipart