File size: 979 Bytes
7161ebe 9179cf3 4e4e893 3dfd27d dc8a913 9179cf3 7161ebe ede4b3b 7161ebe 4e4e893 ede4b3b 4e4e893 7161ebe 4e4e893 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
from fastapi import FastAPI
from pydantic import BaseModel
import gradio as gr
import uvicorn
from fastapi.middleware.cors import CORSMiddleware
app = FastAPI()
# Allow all origins
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
class Item(BaseModel):
prompt: str
zeitstempel: int
@app.post("/items/")
async def create_item(item: Item):
global prompt
prompt = item.prompt
zeitstempel = item.zeitstempel
return {"prompt": prompt, "zeitstempel": zeitstempel}
# Initial prompt value
prompt = ""
def get_prompt():
return prompt
gr_interface = gr.Interface(fn=get_prompt, inputs=[], outputs="text", live=True)
# Mount the Gradio app under "/gradio"
app.mount("/gradio", gr_interface.app)
@app.get("/")
def read_root():
return {"message": "Welcome to the root endpoint."}
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=8000) |