| |
| import os |
| import json |
| import uuid |
| import uvicorn |
| import pika |
| import logging |
| from fastapi import FastAPI, Body, Header, HTTPException |
| from fastapi.middleware.cors import CORSMiddleware |
| from worker import main as rabbit_worker |
|
|
| logging.basicConfig( |
| level=logging.INFO, |
| format="%(asctime)s [%(levelname)s] %(name)s - %(message)s" |
| ) |
| logger = logging.getLogger(__name__) |
|
|
| app = FastAPI() |
|
|
| API_KEY = os.getenv("SECRET_KEY") |
|
|
| app.add_middleware( |
| CORSMiddleware, |
| allow_origins=["*"], |
| allow_credentials=True, |
| allow_methods=["*"], |
| allow_headers=["*"], |
| ) |
|
|
| @app.get("/") |
| async def root(): |
| logger.info("Root endpoint called.") |
| return {"status": "ok", "message": "API is running"} |
|
|
|
|
| if __name__ == "__main__": |
| os.system('python download_models_hf.py') |
| rabbit_worker() |
| uvicorn.run(app, host="0.0.0.0", port=8000) |
