Update app.py

app.py

@@ -37,42 +37,49 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
     url_sd3 = os.getenv("url_sd3")
     url_sd4 = os.getenv("url_sd4")
     
-    print(task)
+    url_sd5 = os.getenv("url_sd5")
+    url_sd6 = os.getenv("url_sd6")
     
-    try:
-        print('n_1')
-        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")
-            result = client.predict(prompt,"[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","sd_xl_base_1.0.safetensors [be9edd61]",25,"DPM++ 2M Karras",7,1024,1024,-1,fn_index=0)
-            return result
-        except:
-            print("n_2")
-            print(url_sd4)
-            with closing(create_connection(f"{url_sd4}", timeout=60)) as conn:
-                conn.send('{"fn_index":0,"session_hash":""}')
-                conn.send(f'{{"data":["{prompt}","[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","dreamshaperXL10_alpha2.safetensors [c8afe2ef]",30,"DPM++ 2M Karras",7,1024,1024,-1],"event_data":null,"fn_index":0,"session_hash":""}}')
-                conn.recv()
-                conn.recv()
-                conn.recv()
-                conn.recv()
-                photo = json.loads(conn.recv())['output']['data'][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
+    
+#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")
+#        result = client.predict(prompt,"[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","sd_xl_base_1.0.safetensors [be9edd61]",25,"DPM++ 2M Karras",7,1024,1024,-1,fn_index=0)
+#        return result
+#    except:
+#        print("n_2")
+#        print(url_sd4)
+#        with closing(create_connection(f"{url_sd4}", timeout=60)) as conn:
+#            conn.send('{"fn_index":0,"session_hash":""}')
+#            conn.send(f'{{"data":["{prompt}","[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","dreamshaperXL10_alpha2.safetensors [c8afe2ef]",30,"DPM++ 2M Karras",7,1024,1024,-1],"event_data":null,"fn_index":0,"session_hash":""}}')
+#            conn.recv()
+#            conn.recv()
+#            conn.recv()
+#            conn.recv()
+#            photo = json.loads(conn.recv())['output']['data'][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
 
 
 def flipp():