Update app.py

app.py

@@ -40,13 +40,26 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
     url_sd5 = os.getenv("url_sd5")
     url_sd6 = os.getenv("url_sd6")
     try:
-        data = {"inputs":f"{prompt}, 4k photo","options":{"negative_prompt":"[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","width":1024,"height":1024,"guidance_scale":7,"num_inference_steps":35}}
-        response = requests.post(f'{url_sd5}', json=data)
-        print(response.text)
-        print(response.json()['image']['file_name'])
-        file_name = response.json()['image']['file_name']
-        photo = f"{url_sd6}{file_name}.png"
-        return photo
+        with closing(create_connection(f"{url_sd3}", timeout=60)) as conn:
+            conn.send('{"fn_index":3,"session_hash":""}')
+            conn.send(f'{{"data":["{prompt}, 4k photo","[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry",7.5,"(No style)"],"event_data":null,"fn_index":3,"session_hash":""}}')
+            while True:
+                status = json.loads(conn.recv())['msg']
+                if status == 'estimation':
+                    continue
+                if status == 'process_starts':
+                    break
+            photo = json.loads(conn.recv())['output']['data'][0][0]
+            photo = photo.replace('data:image/jpeg;base64,', '').replace('data:image/png;base64,', '')
+            photo = Image.open(io.BytesIO(base64.decodebytes(bytes(photo, "utf-8"))))
+            return photo
+            #data = {"inputs":f"{prompt}, 4k photo","options":{"negative_prompt":"[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","width":1024,"height":1024,"guidance_scale":7,"num_inference_steps":35}}
+            #response = requests.post(f'{url_sd5}', json=data)
+            #print(response.text)
+            #print(response.json()['image']['file_name'])
+            #file_name = response.json()['image']['file_name']
+            #photo = f"{url_sd6}{file_name}.png"
+            #return photo
     except:
        with closing(create_connection(f"{url_sd4}", timeout=60)) as conn:
             conn.send('{"fn_index":0,"session_hash":""}')
@@ -60,19 +73,6 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
             photo = Image.open(io.BytesIO(base64.decodebytes(bytes(photo, "utf-8"))))
             return photo
     
-#with closing(create_connection(f"{url_sd3}", timeout=60)) as conn:
-#    conn.send('{"fn_index":3,"session_hash":""}')
-#    conn.send(f'{{"data":["{prompt}, 4k photo","[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry",7.5,"(No style)"],"event_data":null,"fn_index":3,"session_hash":""}}')
-#    while True:
-#        status = json.loads(conn.recv())['msg']
-#        if status == 'estimation':
-#            continue
-#        if status == 'process_starts':
-#            break
-#    photo = json.loads(conn.recv())['output']['data'][0][0]
-#    photo = photo.replace('data:image/jpeg;base64,', '').replace('data:image/png;base64,', '')
-#    photo = Image.open(io.BytesIO(base64.decodebytes(bytes(photo, "utf-8"))))
-#    return photo
 #except:
 #    try:
 #        client = Client("https://prodia-sdxl-stable-diffusion-xl.hf.space")