File size: 2,710 Bytes
f4a41d8 |
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 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
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"),
forcesubject: str =Body('all',title="force subject"),
artists : str =Body('all',title="artists"),
imagetype : str =Body('all',title="image type"),
onlyartists: bool =Body(False,title="only artists"),
antivalues : str =Body('',title="anti values"),
prefixprompt: str =Body('',title="prefix prompt"),
suffixprompt: str =Body('',title="suffix prompt"),
promptcompounderlevel: str =Body('1',title="prompt compounder level"),
seperator: str =Body('comma',title="seperator"),
givensubject: str =Body('',title="givensubject"),
smartsubject:bool = Body(True,title='smart subject'),
giventypeofimage: str = Body('',title='given type of image'),
imagemodechance: int = Body(20,title='image mode chance'),
gender: str = Body('all',title='gender'),
subtypeobject: str = Body('all',title='subtypeobject'),
subtypehumanoid: str = Body('all', title='subtypehumanoid'),
subtypeconcept: str = Body('all', title='subtypeconcept'),
advancedprompting:bool = Body(True,title='advancedprompting'),
hardturnoffemojis:bool = Body(False,title='hardturnoffemojis')
):
keys = ['insanitylevel', 'forcesubject', 'artists', 'imagetype', 'onlyartists', 'antivalues', 'prefixprompt',
'suffixprompt', 'promptcompounderlevel', 'seperator', 'givensubject', 'smartsubject', 'giventypeofimage',
'imagemodechance', 'gender', 'subtypeobject','subtypehumanoid','subtypeconcept','advancedprompting','hardturnoffemojis']
payload = {}
for key in keys:
payload[key] = locals()[key]
prompts = [build_dynamic_prompt(**payload) for _ in range(numberofprompts)]
return {"prompts": prompts}
try:
import modules.script_callbacks as script_callbacks
script_callbacks.on_app_started(one_button_prompt_api)
except:
pass
|