Update app.py

app.py

@@ -7,12 +7,17 @@ import string
 import time
 from queue import Queue
 from threading import Thread
+from transformers import pipeline
 import emoji
 
+ru2en = pipeline("translation", model="Helsinki-NLP/opus-mt-ru-en")
+def translate(text):
+  return ru2en(text)[0]["translation_text"]
 
 text_gen=gr.Interface.load("spaces/phenomenon1981/MagicPrompt-Stable-Diffusion")
 def get_prompts(prompt_text):
     return text_gen("photo, " + prompt_text)
+
 proc1=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0")
 
 def restart_script_periodically():
@@ -54,28 +59,41 @@ def send_it(inputs, noise_level, proc1=proc1):
 
 with gr.Blocks(css='style.css') as demo:
     with gr.Column(elem_id="col-container"):
+        
+        with gr.Row(variant="compact"):
+            input_text_ru = gr.Textbox(
+                label="Short Ru Prompt",
+                show_label=False,
+                max_lines=2,
+                placeholder="Enter a basic idea on Russian",
+            ).style(
+                container=False,
+            )
+            Translate = gr.Button("Ru ▶ En").style(full_width=False)
+        
         with gr.Row(variant="compact"):
             input_text = gr.Textbox(
-                label="Short Prompt",
+                label="Short En Prompt",
                 show_label=False,
                 max_lines=2,
-                placeholder="Enter a basic idea and click 'Magic Prompt'",
+                placeholder="Enter a basic idea",
             ).style(
                 container=False,
             )
-            see_prompts = gr.Button("✨ Magic Prompt ✨").style(full_width=False)
-
+            Translate = gr.Button("Ru ▶ En").style(full_width=False)
         
         with gr.Row(variant="compact"):
-            with gr.Column():
-                prompt = gr.Textbox(
-                    label="Enter your prompt",
-                    show_label=False,
-                    max_lines=2,
-                    placeholder="Full Prompt",
-                ).style(
-                    container=False,
-                )
+            prompt = gr.Textbox(
+                label="Enter your prompt",
+                show_label=False,
+                max_lines=2,
+                placeholder="Full Prompt",
+            ).style(
+                container=False,
+            )
+            see_prompts = gr.Button("✨ Magic Prompt ✨").style(full_width=False)
+            
+        with gr.Row(variant="compact"):
             with gr.Column():
                 noise_level = gr.Slider(minimum=0.0, maximum=3, step=0.1, label="Noise Level")
         
@@ -93,11 +111,12 @@ with gr.Blocks(css='style.css') as demo:
                 run4 = gr.Button("Generate").style(full_width=False)
                 output4=gr.Image()
         
+        Translate.click(translate, inputs=[input_text_ru], outputs=[input_text], queue=False)
         see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt], queue=False)
         run1.click(send_it, inputs=[prompt, noise_level], outputs=[output1])
         run2.click(send_it, inputs=[prompt, noise_level], outputs=[output2])
-        run3.click(send_it, inputs=[prompt, noise_level], outputs=[output3])
-        run4.click(send_it, inputs=[prompt, noise_level], outputs=[output4])
+        run3.click(send_it, inputs=[prompt, noise_level], outputs=[output3], queue=False)
+        run4.click(send_it, inputs=[prompt, noise_level], outputs=[output4], queue=False)
         
 
     demo.launch(enable_queue=True, inline=True)