Update render.py

render.py

@@ -30,7 +30,7 @@ def get_history(prompt_id):
     with urllib.request.urlopen(f"http://{server_address}/history/{prompt_id}") as response:
         return json.loads(response.read())
 
-def get_images(ws, prompt):
+def get_images(ws,prompt,progress):
     progress=gr.Progress(track_tqdm=True)
     prompt_id = queue_prompt(prompt)['prompt_id']
     output_images = {}
@@ -46,7 +46,7 @@ def get_images(ws, prompt):
                 percentage = int((current_progress / max_progress) * 100)
                 if percentage >= last_reported_percentage + 10:
                     last_reported_percentage = percentage
-
+                    progress(percentage/100)
             elif message['type'] == 'executing':
                 data = message['data']
                 if data['node'] is None and data['prompt_id'] == prompt_id:
@@ -72,7 +72,7 @@ def pil_to_base64(image):
     base64_string=base64.b64encode(buffer.getvalue()).decode("utf-8")
     return f"data:image/png;base64,{base64_string}" 
 
-def generate_images(positive_prompt, image):
+def generate_images(positive_prompt,image,progress):
     
     ws = websocket.WebSocket()
     ws_url = f"ws://{server_address}/ws?clientId={client_id}"
@@ -87,7 +87,7 @@ def generate_images(positive_prompt, image):
         data["62"]["inputs"]["images"] = ["61",0]
     seed = random.randint(1, 1000000000)
     data["47"]["inputs"]["noise_seed"] = seed
-    images = get_images(ws, data)
+    images = get_images(ws,data,progress)
     ws.close()
     for node_id in images:
         for image_data in images[node_id]: