Update app.py

app.py

@@ -12,6 +12,9 @@ import io
 import base64
 import re
 from gradio_client import Client
+from fake_useragent import UserAgent
+import random
+
 
 def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
     result = {"prompt": prompt,"negative_prompt": negative_prompt,"task": task,"steps": steps,"sampler": sampler,"cfg_scale": cfg_scale,"seed": seed}
@@ -93,6 +96,64 @@ 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
+    except:
+        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']
+        except:
+            try:
+                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:
+                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
+
             #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)
@@ -100,20 +161,20 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
             #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
-    
+            #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
+            
 #except:
 #    try:
 #        client = Client("https://prodia-sdxl-stable-diffusion-xl.hf.space")