Update app.py

app.py

@@ -35,6 +35,10 @@ import torch
 from TTS.tts.configs.xtts_config import XttsConfig
 from pydub import AudioSegment
 from pyannote.audio import Pipeline
+import traceback
+import wave
+
+logger = logging.getLogger(__name__)
 
 # Accept license terms for Coqui XTTS
 os.environ["COQUI_TOS_AGREED"] = "1"
@@ -157,7 +161,7 @@ def segment_background_audio(audio_path, output_path="background_segments.wav"):
     result_audio.export(output_path, format="wav")
     print(f"Saved non-speech (background) audio to: {output_path}")
 
-    return output_path
+    return True
 
 def transcribe_video_with_speakers(video_path):
     # Extract audio from video
@@ -166,7 +170,7 @@ def transcribe_video_with_speakers(video_path):
     video.audio.write_audiofile(audio_path)
     logger.info(f"Audio extracted from video: {audio_path}")
 
-    segment_background_audio(audio_path)
+    segment_result = segment_background_audio(audio_path)
     print(f"Saved non-speech (background) audio to local")
     
     # Set up device
@@ -343,7 +347,6 @@ def update_translations(file, edited_table, mode):
     except Exception as e:
         raise ValueError(f"Error updating translations: {e}")
 
-
 def create_subtitle_clip_pil(text, start_time, end_time, video_width, video_height, font_path):
     try:
         subtitle_width = int(video_width * 0.8)
@@ -484,14 +487,6 @@ def add_transcript_voiceover(video_path, translated_json, output_path, add_voice
 
     return error_messages
 
-import os
-import traceback
-from TTS.api import TTS
-import wave
-import logging
-
-logger = logging.getLogger(__name__)
-
 # Initialize TTS model only once (outside the function)
 tts = TTS(model_name="tts_models/multilingual/multi-dataset/xtts_v2")