fix: one button generate

app.py

@@ -13,9 +13,6 @@ stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
 
 title="Any Text to Stable Diffusion"
 
-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']
@@ -24,6 +21,10 @@ def get_translation(text):
   print(english_translated)
   return english_translated
 
+def get_images(prompt):
+    prompt_t = get_translation(prompt)
+    gallery_dir = stable_diffusion(prompt_t, fn_index=2)
+    return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
 
 css = """
         .container {
@@ -246,7 +247,7 @@ with gr.Blocks(css=css) as demo:
             with gr.Row():
                 clear_btn = gr.Button(value="Clear")
                 translate_btn = gr.Button(value="Translate", elem_id="translate_btn")
-                diffuse_btn = gr.Button(value="Stable Diffuse!", elem_id="diffuse_btn")
+                diffuse_btn = gr.Button(value="Translate >> SD!", elem_id="diffuse_btn")
                 
                 clear_btn.click(fn=lambda value: gr.update(value=""), inputs=clear_btn, outputs=translated_output)
     
@@ -301,7 +302,7 @@ with gr.Blocks(css=css) as demo:
         
         diffuse_btn.click(get_images, 
                               inputs = [
-                                  translated_output
+                                  enter_prompt
                                   ], 
                               outputs = sd_output
                           )
@@ -315,4 +316,4 @@ with gr.Blocks(css=css) as demo:
 
     
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()