From da0489081bf9a9b9e442883c9fb9f8891e65add5 Mon Sep 17 00:00:00 2001
From: Abdullah Alfaraj <abdullah.a.f2014@gmail.com>
Date: Sun, 25 Jun 2023 22:56:13 +0300
Subject: [PATCH] generate random prompts via restapi

---
 scripts/api.py | 44 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 scripts/api.py

diff --git a/scripts/api.py b/scripts/api.py
new file mode 100644
index 0000000..f438c99
--- /dev/null
+++ b/scripts/api.py
@@ -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