Update app.py

app.py

@@ -25,34 +25,46 @@ def get_flores_200_code(language):
             return code['FLORES-200 code']
     return None
 
-def translate_text(text, destination_language, src_lang=None):
+def detect_language(text):
     """
-    Translates text from the source language to the selected destination language using the T5 model.
+    Detects the language of the given text.
+
+    Args:
+        text (str): The text to detect language from.
+
+    Returns:
+        str: The detected language code.
+    """
+    try:
+        lang_code = detect(text)
+        return lang_code
+    except Exception as e:
+        return str(e)
+
+def translate_text(text, source_language, destination_language):
+    """
+    Translates text from the source language to the destination language using the T5 model.
 
     Args:
         text (str): The text to translate.
-        destination_language (str): The target language for translation.
-        src_lang (str, optional): The source language of the text. If not provided, it will be detected automatically.
+        source_language (str): The source language code.
+        destination_language (str): The target language code.
 
     Returns:
         str: The translated text.
     """
-    # Detect source language if not provided
-    if src_lang is None:
-        try:
-            src_lang = detect(text)
-        except Exception as e:
-            return f"Language detection error: {str(e)}"
-
+    src_code = get_flores_200_code(source_language)
     dest_code = get_flores_200_code(destination_language)
 
+    if not src_code:
+        return "Unsupported source language selected. Please choose a valid language."
     if not dest_code:
-        return "Unsupported language selected. Please choose a valid language."
+        return "Unsupported destination language selected. Please choose a valid language."
 
     try:
         # Perform translation using T5 model pipeline
         translation = text_translator(text,
-                                      src_lang=src_lang,
+                                      src_lang=src_code,
                                       tgt_lang=dest_code)
         translated_text = translation[0]["translation_text"]
         return translated_text
@@ -65,12 +77,12 @@ def main():
         fn=translate_text,
         inputs=[
             gr.Textbox(label="Input text to translate", lines=6, placeholder="Enter text..."),
-            gr.Dropdown([code['Language'] for code in language_data], label="Select input language", default="English"),
+            gr.Dropdown([code['Language'] for code in language_data], label="Select source language"),
             gr.Dropdown([code['Language'] for code in language_data], label="Select destination language")
         ],
         outputs=[gr.Textbox(label="Translated text", lines=4)],
         title="Multi-Language Translator",
-        description="Translate text between multiple languages using the T5 model. Select input and destination languages from the dropdown menus and enter the text to translate.",
+        description="Translate text between multiple languages using the T5 model. Select the source and destination languages from the dropdown menus.",
         theme=gr.themes.Soft(),
         live=True  # Enable live updates
     )