Update app.py

app.py

@@ -1,65 +1,59 @@
-import gradio as gr
-import requests
+# @title Default title text
 import concurrent.futures
+import time
+import gradio as gr
 import os
-SECRET_TOKEN = os.getenv("KEY")
-def prompt(pos,hei,wid,no):
- hed={
-    "Content-Type": "application/json",
-    "Authorization": "Bearer " + SECRET_TOKEN
-      }
- 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 = []
+import requests
+def pr(positive,Neg,width,height,cfg,no):
+ head ={
+         "Content-Type": "application/json",
+         "Authorization": "Bearer " + "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoidXNlcl8yYkpvRzhxd1BYV3lreXpVNmhCQWNZZVlNRk4iLCJpYXQiOjE3MDYyMDM2NjV9.ujXZh1la_WZqwLtPSySw772W1oC34CPab4GKVMipwHA",
+               }
+ j =    {
+     "deployment_id": "62d2228a-38c3-4894-bdc1-6f2eaf1d9ef8",
+     "inputs": {
+       "Cfg": cfg ,
+       "Width ": f"{width}",
+       "Height ": f"{height}",
+       "Ps": f"{positive}",
+       "Neg": f"{Neg}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) ",
+     }
+   }
+ 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']
-  while url ==[]:
-    li = requests.get(f"https://www.comfydeploy.com/api/run?run_id={i}",headers=hed)
-    lo=li.json()
-    url =lo['outputs']
-  while 'url' not in url:
-    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=url['url']
-  urls.append(url)
- return urls
-
+    r = requests.post("https://www.comfydeploy.com/api/run",headers=head,json=j)
+    a=r.json()
+    print(a)
+    a=a['run_id']
+    id.append(a)
+
+    for i in id:
+      li = requests.get(f"https://www.comfydeploy.com/api/run?run_id={i}",headers=head)
+      lo=li.json()
+      url=lo['outputs']
+      while url ==[]:
+        li = requests.get(f"https://www.comfydeploy.com/api/run?run_id={i}",headers=head)
+        lo=li.json()
+        url =lo['outputs']
+      while 'url' not in url:
+        li = requests.get(f"https://www.comfydeploy.com/api/run?run_id={i}",headers=head)
+        lo=li.json()
+        url=lo['outputs'][0]['data']['images'][0]
+      url=url['url']
+      urls.append(url)
+ 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)
+ return list(set(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")
+    gr.Interface(fn=pr, inputs=["text","text",gr.Slider(8,1024,step=8.0 ,value=512),gr.Slider(8,1024,step=8.0, value=512),gr.Slider(1,25,step=1.0,value=7),gr.Slider(1,10,step=1.0)],outputs="gallery")
 
     
 if __name__ == "__main__":
-    iface.launch() 
-
-
+    iface.launch()