Update app.py

app.py

@@ -35,16 +35,23 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
 options = ["German", "Romanian", "English", "French", "Spanish"]
 models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large"]
 
+# Define a function to swap dropdown values
+def swap_languages(src_lang, tgt_lang):
+    return tgt_lang, src_lang 
+
 def create_interface():
     with gr.Blocks() as interface:
         gr.Markdown("## Text Machine Translation")
 
         with gr.Row():
             input_text = gr.Textbox(label="Enter text to translate:", placeholder="Type your text here...")
-
+        # Create the interface with gr.Blocks() as demo: with gr.Row(): sselected_language = gr.Dropdown(choices=options, value="German", label="Source language") tselected_language = gr.Dropdown(choices=options, value="Romanian", label="Target language") swap_button = gr.Button("Swap Languages") # Set up the button to call the swap function
+        
         with gr.Row():
             sselected_language = gr.Dropdown(choices=options, value="German", label="Source language")
             tselected_language = gr.Dropdown(choices=options, value="Romanian", label="Target language")
+            swap_button = gr.Button("Swap Languages")
+            swap_button.click(fn=swap_languages, inputs=[sselected_language, tselected_language], outputs=[sselected_language, tselected_language]
 
         model_name = gr.Dropdown(choices=models, value="Helsinki-NLP", label="Select a model")
         translate_button = gr.Button("Translate")