Update app.py

app.py

@@ -29,30 +29,22 @@ def transcribe_audio_from_file(file_path):
 # Initialize the translation pipeline
 translation_pipeline = pipeline("translation", model="Helsinki-NLP/opus-mt-en-{target_language}")
 
-def translate_text(text, target_language):
-    """
-    Translates the given text into the specified target language.
-
-    Args:
-        text (str): The text to translate.
-        target_language (str): The target language code (e.g., 'es' for Spanish, 'fr' for French).
+# Function to get the appropriate translation model based on target language
+def get_translation_model(target_language):
+    # Map of target languages to their corresponding model names
+    model_map = {
+        "es": "Helsinki-NLP/opus-mt-en-es",  # English to Spanish
+        "fr": "Helsinki-NLP/opus-mt-en-fr",  # English to French
+        "zh": "Helsinki-NLP/opus-mt-en-zh",  # English to Chinese
+        # Add more languages as needed
+    }
+    return model_map.get(target_language, "Helsinki-NLP/opus-mt-en-fr")  # Default to French if not found
 
-    Returns:
-        str: Translated text if successful, otherwise None.
-    """
-    try:
-        # Dynamically construct the model ID
-        model_id = f"Helsinki-NLP/opus-mt-en-{target_language}"
-
-        # Initialize the translation pipeline for the target language
-        translator = pipeline("translation", model=model_id)
-        translation = translator(text, max_length=1000)
-        translated_text = translation[0]["translation_text"]
-        logger.debug(f"Translation to {target_language}: {translated_text}")
-        return translated_text
-    except Exception as e:
-        logger.error(f"An error occurred during translation: {e}")
-        return None
+# Example usage in your application
+def translate_text(text, target_language):
+    translation_model_id = get_translation_model(target_language)
+    translator = pipeline("translation", model=translation_model_id)
+    return translator(text)[0]['translation_text']
 
 # Mock functions for platform actions and analytics
 def mock_post_to_platform(platform, content_title):