retrieve the subjects, artists, image types lists through the api

This commit is contained in:
Abdullah Alfaraj
2023-06-29 21:45:18 +03:00
parent aecabb8d82
commit 6874d22e05
+12
View File
@@ -1,12 +1,24 @@
from fastapi import FastAPI, Body
import gradio as gr
from build_dynamic_prompt import *
from scripts.onebuttonprompt import subjects, artists, imagetypes
def one_button_prompt_api(_: gr.Blocks, app: FastAPI):
@app.get("/one_button_prompt/version")
async def version():
return {"version": 1.0}
@app.get("/one_button_prompt/config")
async def get_config():
config ={
"subjects": subjects,
"artists": artists,
"imagetypes": imagetypes,
}
return config
@app.post("/one_button_prompt/prompt/random")
async def random_prompts(numberofprompts:int = Body(1,title="number of prompts"),
insanitylevel: int= Body(5,title="insanity level"),