Update app.py

app.py

@@ -1,31 +1,42 @@
 import subprocess
+import gradio as gr
+import torch
+from TTS.api import TTS
+
+# Setup detailed logging
+import logging
+logging.basicConfig(level=logging.DEBUG)
 
 # Run the setup.py install command
 try:
     subprocess.run(['python', 'setup.py', 'install', '--user'], check=True)
-    print("Installation successful.")
+    logging.info("Installation successful.")
 except subprocess.CalledProcessError as e:
-    print(f"Installation failed with error: {e}")
-
-import gradio as gr
-import torch
-from TTS.api import TTS
+    logging.error(f"Installation failed with error: {e}")
 
 # Get device
 device = "cuda" if torch.cuda.is_available() else "cpu"
+logging.debug(f"Using device: {device}")
 
 # Init TTS
-tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
+try:
+    tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
+except Exception as e:
+    logging.error(f"Failed to initialize TTS: {e}")
 
 def voice_clone(text: str, speaker_wav: str, language: str):
-    # Run TTS
-    print("Speaker wav:", speaker_wav)
-    tts.tts_to_file(text=text, speaker_wav=speaker_wav, language=language, file_path="output.wav")
-    return "output.wav"
+    try:
+        # Run TTS
+        logging.info(f"Processing with speaker wav: {speaker_wav}")
+        tts.tts_to_file(text=text, speaker_wav=speaker_wav, language=language, file_path="output.wav")
+        return "output.wav"
+    except Exception as e:
+        logging.error(f"Failed to process TTS: {e}")
+        return None
 
 iface = gr.Interface(fn=voice_clone, 
                      inputs=[gr.Textbox(label="Text", info="One or two sentences at a time is better", max_lines=3), gr.Audio(type="filepath", label="Voice spectrogram"), gr.Radio(label="language", info="Select an output language for the synthesised speech", choices=["en", "zh-cn", "ja", "de", "fr", "it", "pt", "pl", "tr", "ko", "nl", "cs", "ar", "es", "hu", "ru"], value="en")], 
                      outputs=gr.Audio(type="filepath", label="Synthesised spectrogram"), 
                      title="Voice Cloning")
 
-iface.launch ()
+iface.launch()