MYGen / app.py
ShashwatStable's picture
app.py
407df6a verified
raw
history blame
1.85 kB
import gradio as gr
import requests
import concurrent.futures
def prompt(pos,hei,wid,no):
hed={
"Content-Type": "application/json",
"Authorization": "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoidXNlcl8yYkpvRzhxd1BYV3lreXpVNmhCQWNZZVlNRk4iLCJpYXQiOjE3MDYyMDM2NjV9.ujXZh1la_WZqwLtPSySw772W1oC34CPab4GKVMipwHA",
}
j={
"deployment_id": "2ce5158e-7fa0-4dd6-b533-550c1e8f8c3b",
"inputs": {
"input_text": f"{pos} (clear image: 1.1), (clean image: 1.1), (high contrast: 1.2), (colorful: 1.2), (vibrant: 1.1), ((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 ": wid,
"Height ": 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","text","text",gr.Slider(1,10)],outputs="gallery")
if __name__ == "__main__":
iface.launch()