| 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 |
|
|
