| from fastapi import FastAPI |
| from fastapi.middleware.cors import CORSMiddleware |
| from rag import initialize_rag |
| from routes import setup_routes |
|
|
| app = FastAPI(title="Style GPT API", version="1.0.0") |
|
|
| app.add_middleware( |
| CORSMiddleware, |
| allow_origins=["*"], |
| allow_credentials=True, |
| allow_methods=["*"], |
| allow_headers=["*"], |
| ) |
|
|
| @app.on_event("startup") |
| async def startup_event(): |
| print("Initializing RAG system...") |
| initialize_rag() |
| print("RAG ready for requests! Model will load on first request.") |
|
|
| setup_routes(app) |
|
|
| if __name__ == "__main__": |
| import uvicorn |
| uvicorn.run(app, host="0.0.0.0", port=7860) |
|
|
