generate random prompts via restapi

2023-06-25 22:56:13 +03:00
parent 2dc4265c64
commit da0489081b
1 changed files with 44 additions and 0 deletions
@@ -0,0 +1,44 @@
+from fastapi import FastAPI, Body
+import gradio as gr
+from build_dynamic_prompt import *
+
+def one_button_prompt_api(_: gr.Blocks, app: FastAPI):
+    @app.get("/one_button_prompt/version")
+    async def version():
+        return {"version": 1.0}
+
+    @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')):
+
+            
+            keys = ['insanitylevel', 'forcesubject', 'artists', 'imagetype', 'onlyartists', 'antivalues', 'prefixprompt',
+            'suffixprompt', 'promptcompounderlevel', 'seperator', 'givensubject', 'smartsubject', 'giventypeofimage',
+            'imagemodechance']
+            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