Update module_translation.py

module_translation.py

@@ -40,11 +40,16 @@ def translate_text(text, tgt_lang):
     ]
 
     # Get the translation
-    translated_text = vlm.get_response(messages)
+    yield from vlm.stream_response(messages)
+
+def translate_text_with_google(text, tgt_lang):
+    """Translate the given text into the given target language using Google Translate.
+    """
     translated_text_google = GoogleTranslator(
         source='auto', target=tgt_lang).translate(text=text)
     
-    return translated_text, translated_text_google
+    return translated_text_google
+
 
 #
 # User interface
@@ -77,7 +82,8 @@ with gr.Blocks() as demo:
             label="Target language",
             render=True
         )
-        translate_btn = gr.Button(value="Translate", variant="primary")
+        translate_btn = gr.Button(value="Translate (Mistral)", variant="primary")
+        translate_google_btn = gr.Button(value="Translate (Google)", variant="primary")
         clear_btn = gr.Button("Clear", variant="secondary")
     
     # Examples
@@ -92,7 +98,7 @@ with gr.Blocks() as demo:
                 ["ברוכה הבאה, קיטי: בית הקפה החדש בלוס אנג'לס החתולה האהובה והחברים שלה מקבלים בית קפה משלהם בשדרות יוניברסל סיטי, שם תוכלו למצוא מגוון של פינוקים מתוקים – החל ממשקאות ועד עוגות", "en"],
             ],
             inputs=[input_text, tgt_lang],
-            outputs=[output_text, output_text_google],
+            outputs=[output_text,],
             fn=translate_text,
             cache_examples=False,
             label="Examples"
@@ -102,7 +108,12 @@ with gr.Blocks() as demo:
     translate_btn.click(
         fn=translate_text,
         inputs=[input_text, tgt_lang],
-        outputs=[output_text, output_text_google]
+        outputs=[output_text,]
+    )
+    translate_google_btn.click(
+        fn=translate_text_with_google,
+        inputs=[input_text, tgt_lang],
+        outputs=[output_text_google,]
     )
     clear_btn.click(
         fn=lambda : ('', '', ''), # input_text, output_text, output_text_google