import gradio as gr
import fastapi
import uvicorn
from pydantic import BaseModel
import json

app = fastapi.FastAPI()


class FetchBody(BaseModel):
    serialize: bool
    config: str


@app.post("/api")
async def fetch_api_info(body: FetchBody):
    try:
        api = gr.blocks.get_api_info(json.loads(body.config), serialize=body.serialize)
        return {"api": api}
    except Exception as e:
        raise fastapi.HTTPException(status_code=fastapi.status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(e))

uvicorn.run(app, host="0.0.0.0", port=7860)