|
|
|
|
|
from fastapi import FastAPI |
|
from fastapi.middleware.cors import CORSMiddleware |
|
from fastapi.responses import HTMLResponse |
|
from fastapi.staticfiles import StaticFiles |
|
from huggingface_hub import login |
|
from config import settings |
|
from routers import tool_bpy_doc, tool_gpu_checker, tool_find_related, tool_wiki_search, tool_calls |
|
|
|
login(settings.huggingface_key) |
|
|
|
app = FastAPI(openapi_url="/api/v1/openapi.json", |
|
docs_url="/api/v1/docs") |
|
|
|
app.add_middleware( |
|
CORSMiddleware, |
|
allow_origins=["https://projects.blender.org"], |
|
allow_methods=["GET", "POST"], |
|
allow_headers=["Authorization", "Content-Type"], |
|
allow_credentials=True, |
|
) |
|
|
|
app.include_router( |
|
tool_bpy_doc.router, prefix="/api/v1", tags=["Tools"]) |
|
|
|
app.include_router( |
|
tool_gpu_checker.router, prefix="/api/v1", tags=["Tools"]) |
|
|
|
app.include_router( |
|
tool_find_related.router, prefix="/api/v1", tags=["Tools"]) |
|
|
|
app.include_router( |
|
tool_wiki_search.router, prefix="/api/v1", tags=["Tools"]) |
|
|
|
app.include_router( |
|
tool_calls.router, prefix="/api/v1", tags=["Function Calls"]) |
|
|
|
|
|
@app.get("/", response_class=HTMLResponse) |
|
async def root(): |
|
return """ |
|
<!DOCTYPE html> |
|
<html> |
|
<head> |
|
<title>My Endpoints</title> |
|
</head> |
|
<body> |
|
<h1>Welcome to @mano-wii API</h1> |
|
<p>Click the button below to access the documentation:</p> |
|
<a href="/api/v1/docs" style="text-decoration: none;"> |
|
<button style="padding: 10px 20px; background-color: #007BFF; color: #fff; border: none; cursor: pointer;">Go to Documentation</button> |
|
</a> |
|
</body> |
|
</html> |
|
""" |
|
|
|
app.mount("/api/v1/static", StaticFiles(directory="static"), name="static") |
|
|
