import gradio as gr
from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
import uvicorn


app = FastAPI()

@app.on_event("startup")
def print_foo():
    print("FOO!!!")

def dream(prompt):
    return ["images/lion.jpg"] * 2, {}

block = gr.Blocks().queue()
with block:
    prompt = gr.Text()
    gallery = gr.Gallery().style(grid=[2], height="auto")
    contains_nfsw = gr.JSON(visible=False)
    btn = gr.Button("Generate")
    btn.click(dream, inputs=prompt, outputs=[gallery, contains_nfsw])

gradio_app = gr.routes.App.create_app(block)

app = gr.mount_gradio_app(app, gradio_app, "localhost", "7869", "/gradio")
app.mount("/img", StaticFiles(directory="images", html=True), name="images")
uvicorn.run(app, port=7860)