pvanand's picture
Update main.py
52b8c04 verified
raw
history blame
1.2 kB
# main.py
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from dotenv import load_dotenv
from routers.llm_chat import router as llm_chat_router
from routers.scraping_router import router as scraping_router
from routers.electronics_router import router as electronics_router
from fastapi.staticfiles import StaticFiles
app = FastAPI(
title="LLM Chat API",
description="A FastAPI application to interact with an external LLM API.",
version="1.0.0",
)
# Enable CORS for all origins
app.add_middleware(
CORSMiddleware,
allow_origins=["*"], # Allows all origins
allow_credentials=True,
allow_methods=["*"], # Allows all methods
allow_headers=["*"], # Allows all headers
)
# Add UI
app.mount("/", StaticFiles(directory="static/ui", html=True), name="index")
# Include the routers
app.include_router(llm_chat_router)
app.include_router(scraping_router)
app.include_router(electronics_router)
# Add to main.py
@app.get("/routes")
async def get_routes():
return {"routes": [{"path": route.path, "name": route.name} for route in app.routes]}
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=8000)