MYGen / app.py
ShashwatStable's picture
Update app.py
126ea31 verified
raw
history blame
1.91 kB
import gradio as gr
import requests
import concurrent.futures
def prompt(pos,hei,wid,no):
hed={
"Content-Type": "application/json",
"Authorization": "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoidXNlcl8yYkpZMWpSZVFzWmRnWHhQZWxoazF0SnpYZVciLCJpYXQiOjE3MDYyODUyMjJ9.yX1xGr3SjZcxuUzOyfJuLtNvx-HpV7VcKu49BsUOoj8",
}
j={
"deployment_id": "2ce5158e-7fa0-4dd6-b533-550c1e8f8c3b",
"inputs": {
"input_text": f"{pos} (clear image: 1.1), (clean image: 1.1),masterpiece , best quality , hires , 8k, ((midlength)) ",
"Negative": "worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, artist name,text, watermark,(Low quality , blurred , bad hands ,diffused body, ugly face , worst quality, low quality, normal quality, painting, drawing, monochrome, grayscale, bad anatomy, bad proportions, signature, watermark)",
"Width ": f"{wid}",
"Height ": f"{hei}",
"Batch": "",
}
}
id = []
urls=[]
p=0
no=no
def gen():
r = requests.post("https://www.comfydeploy.com/api/run",headers=hed,json=j)
a=r.json()
a=a['run_id']
id.append(a)
futures = []
with concurrent.futures.ThreadPoolExecutor() as executor:
for url in range(no):
futures.append(executor.submit(gen))
for future in concurrent.futures.as_completed(futures):
print(future)
for i in id :
li = requests.get(f"https://www.comfydeploy.com/api/run?run_id={i}",headers=hed)
lo=li.json()
url=lo['outputs'][0]['data']['images'][0]['url']
urls.append(url)
return urls
with gr.Blocks() as iface:
gr.Interface(fn=prompt, inputs=["text",gr.Slider(8,1024,step=8.0 ,value=512),gr.Slider(8,1024,step=8.0, value=512),gr.Slider(1,10,step=1.0)],outputs="gallery")
if __name__ == "__main__":
iface.launch()