Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 import json
+from langdetect import detect
 
 # Initialize the translation pipeline with the specific model
 text_translator = pipeline("translation", model="facebook/nllb-200-distilled-600M")
@@ -24,17 +25,25 @@ def get_flores_200_code(language):
             return code['FLORES-200 code']
     return None
 
-def translate_text(text, destination_language):
+def translate_text(text, destination_language, src_lang=None):
     """
-    Translates text from English to the selected destination language using the T5 model.
+    Translates text from the source language to the selected 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.
 
     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)}"
+
     dest_code = get_flores_200_code(destination_language)
 
     if not dest_code:
@@ -43,7 +52,7 @@ def translate_text(text, destination_language):
     try:
         # Perform translation using T5 model pipeline
         translation = text_translator(text,
-                                      src_lang="eng_Latn",
+                                      src_lang=src_lang,
                                       tgt_lang=dest_code)
         translated_text = translation[0]["translation_text"]
         return translated_text
@@ -55,12 +64,13 @@ def main():
     interface = gr.Interface(
         fn=translate_text,
         inputs=[
-            gr.Textbox(label="Input text to translate", lines=6, placeholder="Enter text in English..."),
+            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 destination language")
         ],
         outputs=[gr.Textbox(label="Translated text", lines=4)],
         title="Multi-Language Translator",
-        description="Translate English text to multiple languages using the T5 model. Select the target language from the dropdown menu and enter the text to translate.",
+        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.",
         theme=gr.themes.Soft(),
         live=True  # Enable live updates
     )