Update app.py

app.py

@@ -2,14 +2,8 @@ import sys
 import traceback
 import logging
 import warnings
-import gradio as gr
-import torch
-import torchaudio
 import os
-import g4f
-from transformers import pipeline
-from TTS.api import TTS
-from gtts import gTTS
+import gradio as gr
 
 # Придушення попереджень
 warnings.filterwarnings('ignore')
@@ -25,13 +19,26 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+# Безпечний імпорт модулів
+try:
+    import torch
+    import torchaudio
+    import g4f
+    from transformers import pipeline
+    from TTS.api import TTS
+    from gtts import gTTS
+    logger.info("All modules imported successfully")
+except Exception as e:
+    logger.error(f"Error importing modules: {e}")
+    logger.error(traceback.format_exc())
+
 # Функція для перекладу через g4f
 def neural_translate(text, target_language, source_language):
     try:
         prompt = f"Translate the following text from {source_language} to {target_language} keeping the original meaning and emotional tone: '{text}'"
 
         response = g4f.ChatCompletion.create(
-            model="gpt-4",
+            model="gpt-3.5-turbo",  # Using 3.5-turbo instead of gpt-4 for better compatibility
             messages=[{"role": "user", "content": prompt}],
             stream=False
         )
@@ -56,7 +63,7 @@ def gtts_synthesize(text, target_language, output_filename='output_gtts.wav'):
 # Безпечна транскрипція
 def safe_transcribe_audio(audio_path):
     try:
-        asr = pipeline("automatic-speech-recognition", model="openai/whisper-base")
+        asr = pipeline("automatic-speech-recognition", model="openai/whisper-tiny")  # Using tiny model for faster loading
         result = asr(audio_path)
         return result['text']
     except Exception as e:
@@ -89,8 +96,10 @@ def safe_clone_and_synthesize(audio_path, source_language, target_language):
         mapped_language = LANGUAGE_MAP.get(target_language, target_language)
 
         # Спроба використати клонування голосу для підтримуваних мов
+        output_path = None
         try:
-            tts = TTS(model_name="tts_models/multilingual/multi-dataset/xtts_v2")
+            logger.info("Attempting to use TTS for voice cloning")
+            tts = TTS(model_name="tts_models/multilingual/multi-dataset/your_tts")
             output_path = "output_cloned.wav"
             tts.tts_to_file(
                 text=translated_text,
@@ -98,19 +107,35 @@ def safe_clone_and_synthesize(audio_path, source_language, target_language):
                 speaker_wav=audio_path,
                 language=mapped_language
             )
+            logger.info("Voice cloning successful")
         except Exception as e:
             logger.warning(f"Помилка клонування голосу: {e}. Використання gTTS.")
             output_path = gtts_synthesize(translated_text, mapped_language)
+            logger.info("gTTS synthesis successful")
 
         return output_path, original_text, translated_text
 
     except Exception as e:
         logger.error(f"Загальна помилка: {e}")
-        return None, original_text, translated_text
+        logger.error(traceback.format_exc())
+        return None, f"Помилка: {str(e)}", ""
+
+# Інформація про можливі помилки
+def show_error_info():
+    return """
+    # Важлива інформація!
+    
+    Якщо ви отримали помилку, це може бути через:
+    1. Обмеження пам'яті в Hugging Face Spaces
+    2. Проблеми з моделлю TTS
+    
+    Спробуйте використати менші аудіофайли або змінити налаштування.
+    """
 
 # Створення Gradio інтерфейсу
-with gr.Blocks(title="Голосовий Клон + Перекладач") as demo:
-    gr.Markdown("# 🎙️ Голосовий Клон + Перекладач")
+with gr.Blocks(title="VoiceClone Translator Pro") as demo:
+    gr.Markdown("# 🎙️ VoiceClone Translator Pro")
+    gr.Markdown(show_error_info())
 
     with gr.Row():
         input_audio = gr.Audio(label="Вхідне Аудіо", type="filepath")