Update app.py

app.py

@@ -35,15 +35,7 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
 
     width = 1024
     height = 1024
-    #url_sd1 = os.getenv("url_sd1")
-    #url_sd2 = os.getenv("url_sd2")
-    
-    #url_sd3 = os.getenv("url_sd3")
-    #url_sd4 = os.getenv("url_sd4")
-    
-    #url_sd5 = os.getenv("url_sd5")
-    #url_sd6 = os.getenv("url_sd6")
-    #hf_token = os.getenv("hf_token")
+
     if task == "Playground v2":
         playground = str(os.getenv("playground"))
         with closing(create_connection("wss://ashrafb-arpr.hf.space/queue/join", timeout=60)) as conn:
@@ -75,35 +67,32 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
             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
-        
     try:
-        ua = UserAgent()
-        headers = {
-            'authority': 'ehristoforu-dalle-3-xl-lora-v2.hf.space',
-            'accept': 'text/event-stream',
-            'accept-language': 'ru,en;q=0.9,la;q=0.8,ja;q=0.7',
-            'cache-control': 'no-cache',
-            'referer': 'https://ehristoforu-dalle-3-xl-lora-v2.hf.space/?__theme=light',
-            'sec-ch-ua': '"Not_A Brand";v="8", "Chromium";v="120", "YaBrowser";v="24.1", "Yowser";v="2.5"',
-            'sec-ch-ua-mobile': '?0',
-            'sec-ch-ua-platform': '"Windows"',
-            'sec-fetch-dest': 'empty',
-            'sec-fetch-mode': 'cors',
-            'sec-fetch-site': 'same-origin',
-            'user-agent': f'{ua.random}'
-        }
-        client = Client("ehristoforu/dalle-3-xl-lora-v2", headers=headers)
-        result = client.predict(prompt,"(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation",True,0,1024,1024,6,True, api_name='/run')
-        return result[0][0]['image']
+        with closing(create_connection(f"{url_sd3}")) 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":""}}')
+            c = 0
+            while c < 60:
+                status = json.loads(conn.recv())['msg']
+                if status == 'estimation':
+                    c += 1
+                    time.sleep(1)
+                    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:
             ua = UserAgent()
             headers = {
-                'authority': 'nymbo-sd-xl.hf.space',
+                'authority': 'ehristoforu-dalle-3-xl-lora-v2.hf.space',
                 'accept': 'text/event-stream',
                 'accept-language': 'ru,en;q=0.9,la;q=0.8,ja;q=0.7',
                 'cache-control': 'no-cache',
-                'referer': 'https://nymbo-sd-xl.hf.space/?__theme=light',
+                'referer': 'https://ehristoforu-dalle-3-xl-lora-v2.hf.space/?__theme=light',
                 'sec-ch-ua': '"Not_A Brand";v="8", "Chromium";v="120", "YaBrowser";v="24.1", "Yowser";v="2.5"',
                 'sec-ch-ua-mobile': '?0',
                 'sec-ch-ua-platform': '"Windows"',
@@ -112,28 +101,48 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
                 'sec-fetch-site': 'same-origin',
                 'user-agent': f'{ua.random}'
             }
-            client = Client("Nymbo/SD-XL", headers=headers)
-            result = client.predict(prompt,negative_prompt,"","",True,False,False,0,1024,1024,7,1,25,25,False,api_name="/run")
-            return result
+            client = Client("ehristoforu/dalle-3-xl-lora-v2", headers=headers)
+            result = client.predict(prompt,"(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation",True,0,1024,1024,6,True, api_name='/run')
+            return result[0][0]['image']
         except:
-            ua = UserAgent()
-            headers = {
-                'authority': 'radames-real-time-text-to-image-sdxl-lightning.hf.space',
-                'accept': 'text/event-stream',
-                'accept-language': 'ru,en;q=0.9,la;q=0.8,ja;q=0.7',
-                'cache-control': 'no-cache',
-                'referer': 'https://radames-real-time-text-to-image-sdxl-lightning.hf.space/?__theme=light',
-                'sec-ch-ua': '"Not_A Brand";v="8", "Chromium";v="120", "YaBrowser";v="24.1", "Yowser";v="2.5"',
-                'sec-ch-ua-mobile': '?0',
-                'sec-ch-ua-platform': '"Windows"',
-                'sec-fetch-dest': 'empty',
-                'sec-fetch-mode': 'cors',
-                'sec-fetch-site': 'same-origin',
-                'user-agent': f'{ua.random}'
-            }
-            client = Client("radames/Real-Time-Text-to-Image-SDXL-Lightning", headers=headers)
-            result = client.predict(prompt, [], 0, random.randint(1, 999999), fn_index=0)
-            return result
+            try:
+                ua = UserAgent()
+                headers = {
+                    'authority': 'nymbo-sd-xl.hf.space',
+                    'accept': 'text/event-stream',
+                    'accept-language': 'ru,en;q=0.9,la;q=0.8,ja;q=0.7',
+                    'cache-control': 'no-cache',
+                    'referer': 'https://nymbo-sd-xl.hf.space/?__theme=light',
+                    'sec-ch-ua': '"Not_A Brand";v="8", "Chromium";v="120", "YaBrowser";v="24.1", "Yowser";v="2.5"',
+                    'sec-ch-ua-mobile': '?0',
+                    'sec-ch-ua-platform': '"Windows"',
+                    'sec-fetch-dest': 'empty',
+                    'sec-fetch-mode': 'cors',
+                    'sec-fetch-site': 'same-origin',
+                    'user-agent': f'{ua.random}'
+                }
+                client = Client("Nymbo/SD-XL", headers=headers)
+                result = client.predict(prompt,negative_prompt,"","",True,False,False,0,1024,1024,7,1,25,25,False,api_name="/run")
+                return result
+            except:
+                ua = UserAgent()
+                headers = {
+                    'authority': 'radames-real-time-text-to-image-sdxl-lightning.hf.space',
+                    'accept': 'text/event-stream',
+                    'accept-language': 'ru,en;q=0.9,la;q=0.8,ja;q=0.7',
+                    'cache-control': 'no-cache',
+                    'referer': 'https://radames-real-time-text-to-image-sdxl-lightning.hf.space/?__theme=light',
+                    'sec-ch-ua': '"Not_A Brand";v="8", "Chromium";v="120", "YaBrowser";v="24.1", "Yowser";v="2.5"',
+                    'sec-ch-ua-mobile': '?0',
+                    'sec-ch-ua-platform': '"Windows"',
+                    'sec-fetch-dest': 'empty',
+                    'sec-fetch-mode': 'cors',
+                    'sec-fetch-site': 'same-origin',
+                    'user-agent': f'{ua.random}'
+                }
+                client = Client("radames/Real-Time-Text-to-Image-SDXL-Lightning", headers=headers)
+                result = client.predict(prompt, [], 0, random.randint(1, 999999), fn_index=0)
+                return result
 
 
 def mirror(image_output, scale_by, method, gfpgan, codeformer):