feat: any lang to sd test

app.py

@@ -4,6 +4,10 @@ from PIL import Image
 import flag
 import os
 
+from libretranslatepy import LibreTranslateAPI
+lt = LibreTranslateAPI("https://translate.argosopentech.com/")
+
+
 stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
 ### ————————————————————————————————————————
 
@@ -13,6 +17,13 @@ def get_images(prompt):
     gallery_dir = stable_diffusion(prompt, fn_index=2)
     return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
 
+def get_translation(text):
+  lang_detected = lt.detect(text)[0]['language']
+  print(lang_detected)
+  english_translated = lt.translate(text, lang_detected, "en")
+  print(english_translated)
+  return english_translated
+
 
 css = """
         .container {
@@ -151,6 +162,15 @@ css = """
             --tw-gradient-to: #4caf50;
             border-color: #8bc34a;
         }
+        #translate_btn {
+            color: #fff;
+            font-size: 1em;
+            margin-bottom: 20px;
+            --tw-gradient-from: #4caf50;
+            --tw-gradient-stops: var(--tw-gradient-from), var(--tw-gradient-to);
+            --tw-gradient-to: #4caf50;
+            border-color: #8bc34a;
+        }
         #notice {
             padding: 20px 14px 10px;
             display: flex;
@@ -187,16 +207,8 @@ with gr.Blocks(css=css) as demo:
             </p>
     
         ''')
-#        with gr.Row(elem_id="w2sd_container"):
-#            with gr.Column():
-            
-        gr.Markdown(
-            """
-            ## 1. Stable Diffusion Config
-            """
-        )
-        
-        
+
+                
      
         with gr.Accordion(label="Stable Diffusion Settings", elem_id="sd_settings", visible=False):
             with gr.Row():
@@ -212,7 +224,7 @@ with gr.Blocks(css=css) as demo:
         
         with gr.Row():
             
-            transcripted_output = gr.Textbox(
+            enter_prompt = gr.Textbox(
                                     label="Enter prompt", 
                                     lines=3,
                                     elem_id="transcript"
@@ -221,18 +233,17 @@ with gr.Blocks(css=css) as demo:
         with gr.Column():
             translated_output = gr.Textbox(
                                     label="in English", 
-                                    lines=4,
+                                    lines=3,
                                     elem_id="translated"
                                 )
             with gr.Row():
                 clear_btn = gr.Button(value="Clear")
+                translate_btn = gr.Button(value="Translate", elem_id="translate_btn")
                 diffuse_btn = gr.Button(value="YES", elem_id="diffuse_btn")
                 
                 clear_btn.click(fn=lambda value: gr.update(value=""), inputs=clear_btn, outputs=translated_output)
     
                 
-                
-                
                     
 #            with gr.Column():
                 
@@ -288,6 +299,13 @@ with gr.Blocks(css=css) as demo:
                               outputs = sd_output
                           )
         
+        translate_btn.click(get_translation, 
+                              inputs = [
+                                  enter_prompt
+                                  ], 
+                              outputs = translated_output
+                          )
+
     
 if __name__ == "__main__":
     demo.launch()

requirements.txt

@@ -1,7 +1,4 @@
-#torch
 scipy
 ftfy
-#transformers
 emoji-country-flag
-#git+https://github.com/LambdaLabsML/lambda-diffusers.git#egg=lambda-diffusers
-git+https://github.com/openai/whisper.git
+libretranslatepy