Update app.py

app.py

@@ -435,15 +435,21 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
 def swap_languages(src_lang, tgt_lang):
     return tgt_lang, src_lang 
 
-def get_info(model_name):
+def get_info(model_name, sl, tl):
+    if model_name == "Helsinki-NLP":
+        return requests.get(f'https://huggingface.co/{model_name}/opus-mt-{sl}-{tl}/raw/main/README.md').text
+    if model_name == "Argos":
+        return requests.get(f'https://huggingface.co/TiberiuCristianLeon/Argostranslate/raw/main/README.md').text
+    if model_name == "Google":
+        return '### Google Translate'
     return requests.get(f'https://huggingface.co/{model_name}/raw/main/README.md').text
 
 def create_interface():
     with gr.Blocks() as interface:
         gr.Markdown("### Machine Text Translation with Gradio API and MCP Server")
 
-        with gr.Row():
-            input_text = gr.Textbox(label="Enter text to translate:", placeholder="Type your text here, maximum 512 tokens")
+        # with gr.Row():
+        input_text = gr.Textbox(label="Enter text to translate:", placeholder="Type your text here, maximum 512 tokens")
         
         with gr.Row():
             s_language = gr.Dropdown(choices=options, value = options[0], label="Source language", interactive=True)
@@ -452,16 +458,16 @@ def create_interface():
             swap_btn.click(fn=swap_languages, inputs=[s_language, t_language], outputs=[s_language, t_language], api_name=False, show_api=False)
 
         with gr.Row():
-            model_name = gr.Dropdown(choices=models, label=f"Select a model. Default is {models[1]}.", value = models[1], interactive=True, scale=2)
-            model_infobtn = gr.Button("Model Info", scale=1)
-        translate_btn = gr.Button("Translate")
+            model_name = gr.Dropdown(choices=models, label=f"Select a model. Default is {models[0]}.", value = models[0], interactive=True, scale=3)
+            # model_infobtn = gr.Button("Model Info", scale=1)
+            translate_btn = gr.Button("Translate", scale=1)
 
         translated_text = gr.Textbox(label="Translated text:", placeholder="Display field for translation", interactive=False, show_copy_button=True)
         message_text = gr.Textbox(label="Messages:", placeholder="Display field for status and error messages", interactive=False,
                                   value=f'Default translation settings: from {s_language.value} to {t_language.value} with {model_name.value}.')
         allmodels = gr.HTML(label="Model links:", value=', '.join([f'<a href="https://huggingface.co/{model}">{model}</a>' for model in models]))
-        model_info = gr.Markdown(label="Model info:", value=requests.get(f'https://huggingface.co/{model_name}/raw/main/README.md').text, show_copy_button=True)
-        model_name.change(get_info, model_name, model_info)
+        model_info = gr.Markdown(label="Model info:", value=get_info(model_name, s_language, t_language), show_copy_button=True)
+        model_name.change(get_info, inputs=[model_name, s_language, t_language], outputs=model_info)
         
         translate_btn.click(
             fn=translate_text,