Update app.py

app.py

@@ -6,26 +6,30 @@ from transformers import MarianMTModel, MarianTokenizer
 # Fetch and parse language options from the provided URL
 url = "https://huggingface.co/Lenylvt/LanguageISO/resolve/main/iso.md"
 response = requests.get(url)
-# Assuming the response content is a Markdown table, convert it to a DataFrame
-df = pd.read_csv(url, delimiter="|", skiprows=2, header=None).dropna(axis=1, how='all')
+df = pd.read_csv(response.url, delimiter="|", skiprows=2, header=None).dropna(axis=1, how='all')
 df.columns = ['ISO 639-1', 'ISO 639-2', 'Language Name', 'Native Name']
 df['ISO 639-1'] = df['ISO 639-1'].str.strip()
 
 # Prepare language options for the dropdown
-language_options = [(row['ISO 639-1'], f"{row['Language Name']} ({row['ISO 639-1']})") for index, row in df.iterrows()]
+language_options = [(row['ISO 639-1'], f"{row['ISO 639-1']} - {row['Language Name']}") for index, row in df.iterrows()]
 
-# Your translate_text function and Gradio interface setup goes here
-# Replace the previous static language_options list with the dynamic one created above
+def translate_text(text, source_language_code, target_language_code):
+    # Construct model name using ISO 639-1 codes
+    model_name = f"Helsinki-NLP/opus-mt-{source_language_code}-{target_language_code}"
+    if source_language_code == target_language_code:
+        return "Translation between the same languages is not supported."
+    try:
+        tokenizer = MarianTokenizer.from_pretrained(model_name)
+        model = MarianMTModel.from_pretrained(model_name)
+    except Exception as e:
+        return f"Failed to load model for {source_language_code} to {target_language_code}: {str(e)}"
+    translated = model.generate(**tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512))
+    translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
+    return translated_text
 
-# Example translate_text function placeholder
-def translate_text(text, source_language, target_language):
-    return "Translation function implementation"
-
-# Create dropdowns for source and target languages, using only the codes for value
 source_language_dropdown = gr.Dropdown(choices=language_options, label="Source Language")
 target_language_dropdown = gr.Dropdown(choices=language_options, label="Target Language")
 
-# Define the interface
 iface = gr.Interface(
     fn=translate_text,
     inputs=[gr.Textbox(lines=2, placeholder="Enter text to translate..."), source_language_dropdown, target_language_dropdown],
@@ -34,5 +38,4 @@ iface = gr.Interface(
     description="Select source and target languages to translate text."
 )
 
-# Launch the app
-iface.launch()
+iface.launch()