Update app.py

app.py

@@ -15,36 +15,65 @@ from gradio_client import Client
 
 
 def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
-    headers = {
-        'authority': 'multimodalart-stable-cascade.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://multimodalart-stable-cascade.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': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 YaBrowser/24.1.0.0 Safari/537.36',
-    }
-    client = Client("multimodalart/stable-cascade", headers=headers)
-    result = client.predict(
-    		"cat!!",	# str  in 'Prompt' Textbox component
-    		"",	# str  in 'Negative prompt' Textbox component
-    		0,	# float (numeric value between 0 and 2147483647) in 'Seed' Slider component
-    		1024,	# float (numeric value between 1024 and 1536) in 'Width' Slider component
-    		1024,	# float (numeric value between 1024 and 1536) in 'Height' Slider component
-    		20,	# float (numeric value between 10 and 30) in 'Prior Inference Steps' Slider component
-    		4,	# float (numeric value between 0 and 20) in 'Prior Guidance Scale' Slider component
-    		10,	# float (numeric value between 4 and 12) in 'Decoder Inference Steps' Slider component
-    		0,	# float (numeric value between 0 and 0) in 'Decoder Guidance Scale' Slider component
-    		1,	# float (numeric value between 1 and 2) in 'Number of Images' Slider component
-      
-    		api_name="/run"
-    )
-    return result
+    prompt = re.sub(r'[^a-zA-Zа-яА-Я\s]', '', prompt)
+    
+    try:
+        language = detect(prompt)
+        if language == 'ru':
+            prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
+            print(prompt)
+    except:
+        pass
+
+    try:
+        with closing(create_connection(f"wss://artgan-diffusion-api.hf.space/queue/join", timeout=60)) as conn:
+            conn.send('{"fn_index":0,"session_hash":""}')
+            conn.send(f'{{"fn_index":0,"data":["{prompt}",{negative_prompt},1024,1024,4,42,1,20,10],"session_hash":""}}')
+            conn.recv()
+            conn.recv()
+            conn.recv()
+            conn.recv()
+            a = conn.recv()
+            photo = json.loads(a)['output']['data'][0][0]['name']
+            photo = "https://artgan-diffusion-api.hf.space/file=" + photo
+            return photo
+    except:
+        try:
+            headers = {
+                'authority': 'ehristoforu-stable-cascade.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-stable-cascade.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': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 YaBrowser/24.1.0.0 Safari/537.36',
+            }
+            client = Client("https://ehristoforu-stable-cascade.hf.space", headers=headers)
+            result = client.predict(prompt, negative_prompt, 1024, 1024, True)
+            return result[0]['image']
+        except:
+            headers = {
+                'authority': 'multimodalart-stable-cascade.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://multimodalart-stable-cascade.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': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 YaBrowser/24.1.0.0 Safari/537.36',
+            }
+            client = Client("multimodalart/stable-cascade", headers=headers)
+            result = client.predict(prompt, negative_prompt,seed, 1024,	1024, 20, 4, 10, 0, 1, api_name="/run")
+            return result
 
 
 def mirror(image_output, scale_by, method, gfpgan, codeformer):
@@ -96,23 +125,10 @@ with gr.Blocks(css=css) as demo:
     with gr.Tab("Базовые настройки"):
         with gr.Row():
             prompt = gr.Textbox(placeholder="Введите описание изображения...", show_label=True, label='Описание изображения:', lines=3)
-        with gr.Row():
-            task = gr.Radio(interactive=True, value="Stable Diffusion XL 1.0", show_label=True, label="Модель нейросети:", choices=['Stable Diffusion XL 1.0', 'Crystal Clear XL', 
-                                                                                                              'Juggernaut XL', 'DreamShaper XL',
-                                                                                                              'SDXL Niji', 'Cinemax SDXL', 'NightVision XL',
-                                                                                                              'Playground v2', 'OpenDalle v1.1'])
     with gr.Tab("Расширенные настройки"):
         with gr.Row():
             negative_prompt = gr.Textbox(placeholder="Negative Prompt", show_label=True, label='Negative Prompt:', lines=3, value="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry")
-        with gr.Row():
-            sampler = gr.Dropdown(value="DPM++ SDE Karras", show_label=True, label="Sampling Method:", choices=[
-                "Euler", "Euler a", "Heun", "DPM++ 2M", "DPM++ SDE", "DPM++ 2M Karras", "DPM++ SDE Karras", "DDIM"])
-        with gr.Row():
-            steps = gr.Slider(show_label=True, label="Sampling Steps:", minimum=1, maximum=50, value=35, step=1)
-        with gr.Row():
-            cfg_scale = gr.Slider(show_label=True, label="CFG Scale:", minimum=1, maximum=20, value=7, step=1)
-        with gr.Row():
-            seed = gr.Number(show_label=True, label="Seed:", minimum=-1, maximum=1000000, value=-1, step=1)
+            seed = gr.Number(show_label=True, label="Seed:", minimum=1, maximum=1000000, value=1, step=1)
     
     with gr.Tab("Настройки апскейлинга"):
         with gr.Column():
@@ -130,7 +146,7 @@ with gr.Blocks(css=css) as demo:
         text_button = gr.Button("Сгенерировать изображение", variant='primary', elem_id="generate")
     with gr.Column():
         image_output = gr.Image(show_download_button=True, interactive=False, label='Результат:', elem_id='image_output', type='filepath')
-        text_button.click(flip_text, inputs=[prompt, negative_prompt, task, steps, sampler, cfg_scale, seed], outputs=image_output, concurrency_limit=12)
+        text_button.click(flip_text, inputs=[prompt, negative_prompt, seed], outputs=image_output, concurrency_limit=12)
         
         img2img_b = gr.Button("Увеличить изображение", variant='secondary')
         image_i2i = gr.Image(show_label=True, label='Увеличенное изображение:')