File size: 642 Bytes
ab2012f
57117a0
ab2012f
 
57117a0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ab2012f
57117a0
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
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)