main.py 871 Bytes
Newer Older
1
2
3
from fastapi import FastAPI, Request, Depends, HTTPException
from fastapi.middleware.cors import CORSMiddleware

Timothy J. Baek's avatar
Timothy J. Baek committed
4
from apps.web.routers import auths, users, chats, modelfiles, utils
5
from config import WEBUI_VERSION, WEBUI_AUTH
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

app = FastAPI()

origins = ["*"]

app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)


app.include_router(auths.router, prefix="/auths", tags=["auths"])
Timothy J. Baek's avatar
Timothy J. Baek committed
21
app.include_router(users.router, prefix="/users", tags=["users"])
Timothy J. Baek's avatar
Timothy J. Baek committed
22
app.include_router(chats.router, prefix="/chats", tags=["chats"])
Timothy J. Baek's avatar
Timothy J. Baek committed
23
24
25
26
app.include_router(modelfiles.router, prefix="/modelfiles", tags=["modelfiles"])


app.include_router(utils.router, prefix="/utils", tags=["utils"])
27
28
29
30


@app.get("/")
async def get_status():
31
    return {"status": True, "version": WEBUI_VERSION, "auth": WEBUI_AUTH}