pngwn's picture
pngwn HF staff
Update app.py
a6bd9d9 verified
import gradio as gr
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
class Data:
def __init__(self, data):
self.data = [[data]]
def update(self):
new_data = f"New data - item: {len(self.data) + 1}"
self.data.append([new_data])
return new_data
def get_data(self):
return self.data
app = FastAPI()
data_for_app = Data("initial data")
# Add CORS middleware
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.post("/endpoint")
async def custom_endpoint(data: dict):
new = data_for_app.update();
return {"message": "Received data", "data": new}
with gr.Blocks() as blocks:
gr.Dataframe(value=data_for_app.get_data, label="My Data")
# Mount Gradio app to root
app = gr.mount_gradio_app(app, blocks, path="/", ssr_mode=False)
# Custom POST endpoint
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=7860)