logic change

app.py

@@ -77,10 +77,6 @@ class myTheme(Base):
         )
 
 
-def on_select(evt: gr.SelectData):  # SelectData is a subclass of EventData
-    return evt.value
-
-
 def detect_lang(article):
     """
     Language Detection using library langid
@@ -109,6 +105,7 @@ def opus_trans(article, target_language):
     Returns:
         string: translated piece of article based off target_language
     """
+
     result_lang = detect_lang(article)
 
     if target_language == "English":
@@ -135,24 +132,25 @@ def nllb_trans(article, target_language):
     pass
 
 
+def translate(article, toolkit, target_language):
+    if toolkit == "OPUS":
+        translated = opus_trans(article, target_language)
+    elif toolkit == "NLLB":
+        pass
+
+    return translated
+
+
 myTheme = myTheme()
 
 with gr.Blocks(theme=myTheme) as demo:
     article = gr.Textbox(label="Article")
-    toolkit_select = gr.Dropdown(["OPUS", "NLLB"], label="Select Translation Model")
+    toolkit_select = gr.Radio(["OPUS", "NLLB"], label="Select Translation Model")
     lang_select = gr.Radio(["English", "Chinese"], label="Select Desired Language")
     result = gr.Textbox(label="Translated Result")
     trans_btn = gr.Button("Translate")
-    selected = toolkit_select.select(on_select)
-    print(selected)
-    # fntouse = f"{selected.lower()}_trans"
-    trans_btn.click(fn=opus_trans, inputs=[article, lang_select], outputs=result)
-    # if selected == "OPUS":
-    #     print(" chosen OPUS")
-
-    #     trans_btn.click(fn=opus_trans, inputs=[article, lang_select], outputs=result)
-    # elif selected == "NLLB":
-    #     print(" chosen NLLB")
-    #     trans_btn = gr.Button("Translate")
-    #     trans_btn.click(fn=nllb_trans, inputs=[article, lang_select], outputs=result)
+    trans_btn.click(
+        fn=translate, inputs=[article, toolkit_select, lang_select], outputs=result
+    )
+
 demo.launch()