|
|
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) |
|
|
|
