Spaces:
Runtime error
Runtime error
File size: 3,367 Bytes
04c4194 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 | """
DocuLens — Standalone FastAPI entry point for Hugging Face Spaces.
Pure FastAPI server served by uvicorn on port 7860 (HF Spaces default).
No Gradio dependency — just the REST API.
"""
import os
import logging
from fastapi import FastAPI
from fastapi.responses import JSONResponse
# ---------------------------------------------------------------------------
# Structured logging — configure once, before anything else logs
# ---------------------------------------------------------------------------
LOG_LEVEL = os.environ.get("LOG_LEVEL", "INFO").upper()
LOG_FORMAT = os.environ.get(
"LOG_FORMAT",
"%(asctime)s [%(levelname)s] %(name)s: %(message)s",
)
logging.basicConfig(
level=getattr(logging, LOG_LEVEL, logging.INFO),
format=LOG_FORMAT,
datefmt="%Y-%m-%d %H:%M:%S",
)
# Quieten noisy third-party loggers
for _noisy in ("httpx", "httpcore", "urllib3", "uvicorn.access"):
logging.getLogger(_noisy).setLevel(logging.WARNING)
logger = logging.getLogger(__name__)
# ---------------------------------------------------------------------------
# FastAPI application
# ---------------------------------------------------------------------------
app = FastAPI(
title="DocuLens",
description="AI-Powered Invoice & Document Extraction API",
version="1.2.0",
)
# ---------------------------------------------------------------------------
# CORS — applied BEFORE startup (unlike the Gradio version, this works)
# ---------------------------------------------------------------------------
from api import add_cors_middleware
add_cors_middleware(app)
# ---------------------------------------------------------------------------
# Mount all API routers
# ---------------------------------------------------------------------------
from api import api_router
from api_runs import runs_router
from api_prompts import prompts_router
from api_chat import chat_router
from api_webhooks import webhooks_router
from api_export import export_router
from api_batch import batch_router
from api_tiers import tiers_router
from api_golden import golden_router
app.include_router(api_router)
app.include_router(runs_router)
app.include_router(prompts_router)
app.include_router(chat_router)
app.include_router(webhooks_router)
app.include_router(export_router)
app.include_router(batch_router)
app.include_router(tiers_router)
app.include_router(golden_router)
# ---------------------------------------------------------------------------
# Root route — useful for health probes and quick verification
# ---------------------------------------------------------------------------
@app.get("/")
async def root():
return JSONResponse(
{
"service": "DocuLens",
"version": "1.2.0",
"status": "running",
"docs": "/docs",
"health": "/api/v1/health",
}
)
# ---------------------------------------------------------------------------
# Entry point — HF Spaces uses port 7860 by default
# ---------------------------------------------------------------------------
if __name__ == "__main__":
import uvicorn
port = int(os.environ.get("PORT", "7860"))
logger.info("Starting DocuLens on port %d", port)
uvicorn.run(
"main:app",
host="0.0.0.0",
port=port,
log_level=LOG_LEVEL.lower(),
)
|