Update app.py

app.py

@@ -31,15 +31,28 @@ PENALTY = 1.2
 TOP_K = 64
 TOP_P = 0.95
 
+LANGUAGES = [
+    "🇬🇧 English",
+    "🇳🇴 Norwegian (Bokmål)",
+    "🇳🇴 Norwegian (Nynorsk)"
+]
+
+
+def set_default_target():
+    return "*Translating...*"
+
 
 def translate(source, source_language, target_language):
     return "This is a fake translation"
 
 
+def switch_inputs(source, target, source_language, target_language):
+    return target, source, target_language, source_language
+
+
 import gradio as gr
 
-#with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
-with gr.Blocks() as demo:
+with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
 
     gr.Markdown("# Norwegian-English translation")
     # gr.HTML('<img src="https://huggingface.co/ltg/norbert3-base/resolve/main/norbert.png" width=6.75%>')
@@ -48,18 +61,21 @@ with gr.Blocks() as demo:
     # chatbot = gr.Chatbot(value=[[None, "Hei, hva kan jeg gjøre for deg? 😊"]])
 
     with gr.Row():
-        with gr.Column(scale=1, variant="panel"):
+        with gr.Column(scale=7, variant="panel"):
             source_language = gr.Dropdown(
-                ["English", "Norwegian (Bokmål)", "Norwegian (Nynorsk)"], value="English", show_label=False
+                LANGUAGES, value=LANGUAGES[0], show_label=False
             )
             source = gr.Textbox(
-                label="Source text", placeholder="What do you want to translate?", show_label=False, lines=7, max_lines=100
+                label="Source text", placeholder="What do you want to translate?", show_label=False, lines=7, max_lines=100, autofocus=True
             )  # .style(container=False)
             submit = gr.Button("Submit", variant="primary")  # .style(full_width=True)
 
-        with gr.Column(scale=1, variant="panel"):
+        with gr.Column(scale=1, variant=None):
+            switch = gr.Button("🔄")
+
+        with gr.Column(scale=7, variant="panel"):
             target_language = gr.Dropdown(
-                ["English", "Norwegian (Bokmål)", "Norwegian (Nynorsk)"], value="Norwegian (Bokmål)", show_label=False
+                LANGUAGES, value=LANGUAGES[1], show_label=False
             )
             target = gr.Textbox(
                 label="Translation", show_label=False, interactive=False, lines=7, max_lines=100
@@ -84,19 +100,35 @@ with gr.Blocks() as demo:
 
 
     submit_event = source.submit(
-        fn=update_state_after_user, inputs=None, outputs=[source, target, source_language, target_language], queue=False
+        fn=update_state_after_user, inputs=None, outputs=[source, submit, source_language, target_language], queue=False
+    ).then(
+        fn=set_default_target, inputs=[], outputs=[target], queue=False
     ).then(
         fn=translate, inputs=[source, source_language, target_language], outputs=[target], queue=True
     ).then(
-        fn=update_state_after_return, inputs=None, outputs=[source, target, source_language, target_language], queue=False
+        fn=update_state_after_return, inputs=None, outputs=[source, submit, source_language, target_language], queue=False
     )
     
     submit_click_event = submit.click(
-        fn=update_state_after_user, inputs=None, outputs=[source, target, source_language, target_language], queue=False
+        fn=update_state_after_user, inputs=None, outputs=[source, submit, source_language, target_language], queue=False
+    ).then(
+        fn=set_default_target, inputs=[], outputs=[target], queue=False
+    ).then(
+        fn=translate, inputs=[source, source_language, target_language], outputs=[target], queue=True
+    ).then(
+        fn=update_state_after_return, inputs=None, outputs=[source, submit, source_language, target_language], queue=False
+    )
+
+    switch_event = switch.click(
+        fn=switch_inputs, inputs=[source, target, source_language, target_language], outputs=[target, source, target_language, source_language], queue=False
+    ).then(
+        fn=update_state_after_user, inputs=None, outputs=[source, submit, source_language, target_language], queue=False
+    ).then(
+        fn=set_default_target, inputs=[], outputs=[target], queue=False
     ).then(
         fn=translate, inputs=[source, source_language, target_language], outputs=[target], queue=True
     ).then(
-        fn=update_state_after_return, inputs=None, outputs=[source, target, source_language, target_language], queue=False
+        fn=update_state_after_return, inputs=None, outputs=[source, submit, source_language, target_language], queue=False
     )
 
 demo.queue(max_size=32, concurrency_count=2)