Update app.py

app.py

@@ -15,15 +15,13 @@ login(token=hf_token, add_to_git_credential=True)
 
 # Define available languages with their respective Helsinki model names
 language_models = {
-    "Abkhaz": "Helsinki-NLP/opus-mt-abk-en",
     "Arabic": "Helsinki-NLP/opus-mt-ar-en",
-    "Azerbaijani": "Helsinki-NLP/opus-mt-az-en",
     "Bengali": "Helsinki-NLP/opus-mt-bn-en",
-    "Chinese": "Helsinki-NLP/opus-mt-zh-en",
-    "Danish": "Helsinki-NLP/opus-mt-da-en",
-    "Finnish": "Helsinki-NLP/opus-mt-fi-en",
     "French": "Helsinki-NLP/opus-mt-fr-en",
     "Hindi": "Helsinki-NLP/opus-mt-hi-en",
+    "Russian": "Helsinki-NLP/opus-mt-ru-en",
+    "German": "Helsinki-NLP/opus-mt-de-en",
+    "Spanish": "Helsinki-NLP/opus-mt-es-en",
     "Tamil": "Helsinki-NLP/opus-mt-mul-en"  # Using multilingual model for Tamil
 }
 
@@ -39,10 +37,10 @@ flux_API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FL
 flux_headers = {"Authorization": f"Bearer {hf_token}"}
 
 # Function for translation
-def translate_text(tamil_text, language):
+def translate_text(text, language):
     translator = load_translation_pipeline(language)
     try:
-        translation = translator(tamil_text, max_length=40)
+        translation = translator(text, max_length=40)
         translated_text = translation[0]['translation_text']
         return translated_text
     except Exception as e:
@@ -78,8 +76,8 @@ def generate_creative_text(translated_text):
         return None
 
 # Function to handle the full workflow
-def translate_generate_image_and_text(tamil_text, language):
-    translated_text = translate_text(tamil_text, language)
+def translate_generate_image_and_text(input_text, language):
+    translated_text = translate_text(input_text, language)
     image = generate_image(translated_text)
     creative_text = generate_creative_text(translated_text)
     return translated_text, creative_text, image