Spaces:
Running
Running
| # main.py | |
| 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_calls, tool_find_related, tool_wiki_search | |
| login(settings.huggingface_key) | |
| app = FastAPI(openapi_url="/api/v1/openapi.json", | |
| docs_url="/api/v1/docs") | |
| app.add_middleware( | |
| CORSMiddleware, | |
| allow_origins=["*"], | |
| allow_methods=["*"], | |
| allow_headers=["*"], | |
| 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"]) | |
| 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") | |