Update app.py

app.py

@@ -97,57 +97,17 @@ 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
     except:
-        try:
-            playground = str(os.getenv("playground"))
-            with closing(create_connection("wss://ashrafb-arpr.hf.space/queue/join", timeout=60)) as conn:
-                conn.send('{"fn_index":0,"session_hash":""}')
-                conn.send(f'{{"fn_index":0,"data":["{prompt}"],"session_hash":""}}')
-                conn.recv()
-                conn.recv()
-                conn.recv()
-                conn.recv()
-                a = conn.recv()
-                print(">> A:", a)
-                photo = json.loads(a)['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
-        except:
-            artigen = str(os.getenv("artigen"))
-            with closing(create_connection("wss://ashrafb-arv3s.hf.space/queue/join", timeout=60)) as conn:
-                conn.send('{"fn_index":0,"session_hash":""}')
-                conn.send(f'{{"fn_index":0,"data":["{prompt}", 0, "No style"],"session_hash":""}}')
-                conn.recv()
-                conn.recv()
-                conn.recv()
-                conn.recv()
-                a = conn.recv()
-                print(">> A:", a)
-                photo = json.loads(a)['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
-            #except Exception as e:
-            #    print("e: --> ",e)
-            #    with closing(create_connection(f"{url_sd4}")) 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
+        with closing(create_connection(f"{url_sd4}")) 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
             
 #except:
 #    try: