feat: updated code

api/server.py

@@ -33,14 +33,16 @@ from typing import Annotated, Any, Dict, List, Optional
 from urllib.parse import urlparse
 
 import httpx
-from fastapi import FastAPI, File, Form, HTTPException, UploadFile, status
+import uvicorn
+from fastapi import FastAPI, File, Form, HTTPException, UploadFile
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.middleware.gzip import GZipMiddleware
-from fastapi.responses import PlainTextResponse
-from pydantic import BaseModel, field_validator
+from pydantic import BaseModel, Field
 
+from banner import print_banner
 from core import ConversionError, ConversionResult, DocumentConverter, SUPPORTED_EXTENSIONS
 from extraction.generic_json_extractor import extract
+from extraction.label_mapper import validate_mappings
 from logger import get_logger
 
 logger = get_logger(__name__)
@@ -98,6 +100,7 @@ def _start_ping_scheduler() -> None:
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     """Application lifespan handler — runs startup and shutdown logic."""
+    print_banner()
     logger.info(
         "MarkItDown API starting | version=2.1.0 | host=0.0.0.0:7860 | started_at=%s",
         datetime.now(timezone.utc).strftime("%Y-%m-%d %H:%M:%S UTC"),

main.py

@@ -4,26 +4,31 @@ MarkItDown API — entry point.
 Run directly::
     python main.py
 
-Or via start.sh for production (environment variable overrides supported).
+Environment variables:
+    WORKERS     uvicorn worker count (default: 1)
+    PORT        listen port (default: 7860)
 """
 from __future__ import annotations
 
+import os
+
 import uvicorn
 
-from banner import print_banner
 from logger import get_logger
 
 logger = get_logger(__name__)
 
 if __name__ == "__main__":
-    print_banner()
-    logger.info("Starting MarkItDown API server")
+    workers = int(os.getenv("WORKERS", os.cpu_count() or 1))
+    port = int(os.getenv("PORT", "7860"))
+
+    logger.info("Starting server | workers=%d | port=%d", workers, port)
     uvicorn.run(
         "api.server:app",
         host="0.0.0.0",
-        port=7860,
+        port=port,
         reload=False,
-        workers=1,
+        workers=workers,
         log_level="info",
         access_log=True,
     )