main.py 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. from fastapi import FastAPI, Depends
  2. from fastapi.routing import APIRoute
  3. from fastapi.middleware.cors import CORSMiddleware
  4. from apps.web.routers import auths, users, chats, modelfiles, configs, utils
  5. from config import WEBUI_VERSION, WEBUI_AUTH
  6. app = FastAPI()
  7. origins = ["*"]
  8. app.state.ENABLE_SIGNUP = True
  9. app.state.DEFAULT_MODELS = "llava:13b"
  10. app.add_middleware(
  11. CORSMiddleware,
  12. allow_origins=origins,
  13. allow_credentials=True,
  14. allow_methods=["*"],
  15. allow_headers=["*"],
  16. )
  17. app.include_router(auths.router, prefix="/auths", tags=["auths"])
  18. app.include_router(users.router, prefix="/users", tags=["users"])
  19. app.include_router(chats.router, prefix="/chats", tags=["chats"])
  20. app.include_router(modelfiles.router, prefix="/modelfiles", tags=["modelfiles"])
  21. app.include_router(configs.router, prefix="/configs", tags=["configs"])
  22. app.include_router(utils.router, prefix="/utils", tags=["utils"])
  23. @app.get("/")
  24. async def get_status():
  25. return {
  26. "status": True,
  27. "version": WEBUI_VERSION,
  28. "auth": WEBUI_AUTH,
  29. "default_models": app.state.DEFAULT_MODELS,
  30. }