Spaces:
Running
Running
| import os | |
| import sys | |
| from pathlib import Path | |
| from dotenv import load_dotenv | |
| from fastapi import FastAPI | |
| from fastapi.middleware.cors import CORSMiddleware | |
| from fastapi.staticfiles import StaticFiles | |
| # Ensure the 'app' directory is always discoverable in sys.path | |
| APP_DIR = Path(__file__).resolve().parent | |
| if str(APP_DIR) not in sys.path: | |
| sys.path.insert(0, str(APP_DIR)) | |
| from routers import api_router | |
| from utils.logger import app_logger | |
| load_dotenv() | |
| # App initialization | |
| app = FastAPI( | |
| title="Atlas - Document Intelligence API", | |
| description="Atlas Hierarchical Tree Search Document Analysis API with Multimodal Support", | |
| version="2.0.0", | |
| ) | |
| # Static files for extracted document images | |
| STATIC_DIR = os.path.join(APP_DIR, "static") | |
| os.makedirs(STATIC_DIR, exist_ok=True) | |
| app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static") | |
| # CORS middleware | |
| _raw_cors = os.getenv("CORS_ALLOW_ORIGINS", "http://localhost:5173,http://localhost:3000") | |
| cors_origins = [origin.strip() for origin in _raw_cors.split(",") if origin.strip()] | |
| app.add_middleware( | |
| CORSMiddleware, | |
| allow_origins=cors_origins, | |
| allow_origin_regex=os.getenv("CORS_ALLOW_ORIGIN_REGEX", r"https://.*\.vercel\.app|https://.*\.hf\.space"), | |
| allow_credentials=False, | |
| allow_methods=["GET", "POST", "PATCH", "DELETE", "OPTIONS"], | |
| allow_headers=["*"], | |
| ) | |
| # Register API routers | |
| app.include_router(api_router) | |
| # Mount FastMCP Server (SSE & HTTP protocol for external MCP clients) | |
| try: | |
| from mcp_server.web_search_mcp import mcp as research_mcp | |
| app.mount("/mcp", research_mcp.http_app()) | |
| app_logger.info("FastMCP Research & Study Server mounted successfully at /mcp") | |
| except Exception as mcp_err: | |
| app_logger.warning(f"Could not mount FastMCP at /mcp: {mcp_err}") | |
| async def root(): | |
| """Public root endpoint.""" | |
| return {"message": "Atlas - Document Intelligence API", "docs": "/docs", "health": "/health"} | |
| async def health_check(): | |
| """Health check endpoint to verify service availability and database status.""" | |
| from core.persistent_db import DB_STATUS, persistent_db | |
| try: | |
| docs = persistent_db.list_docs() | |
| count = len(docs) | |
| except Exception: | |
| count = 0 | |
| return { | |
| "status": "ok", | |
| "service": "Vectorless RAG API", | |
| "fastmcp": "active", | |
| "database": DB_STATUS, | |
| "docs_count": count, | |
| } | |
| async def on_startup(): | |
| app_logger.info("Vectorless RAG API service started with FastMCP Web Search enabled.") |