Update app.py

app.py

@@ -3,13 +3,13 @@ import zipfile
 import subprocess
 import gradio as gr
 from tortoise.api import TextToSpeech
-from tortoise.utils.audio import save_wav
+import torchaudio  # Added import for torchaudio
 
 # Paths
-ZIP_FILE = "Trump-LipSync.zip"  # Name of the uploaded zip file
-EXTRACTED_FOLDER = "Trump-LipSync"  # Folder where the zip file will be extracted
-TRUMP_VIDEO_PATH = os.path.join(EXTRACTED_FOLDER, "videos", "trump.mp4")  # Path to Trump's video
-CHECKPOINT_PATH = os.path.join(EXTRACTED_FOLDER, "Wav2Lip", "checkpoints", "wav2lip_gan.pth")  # Path to Wav2Lip checkpoint
+ZIP_FILE = "Trump-LipSync.zip"  
+EXTRACTED_FOLDER = "Trump-LipSync"  
+TRUMP_VIDEO_PATH = os.path.join(EXTRACTED_FOLDER, "videos", "trump.mp4")  
+CHECKPOINT_PATH = os.path.join(EXTRACTED_FOLDER, "Wav2Lip", "checkpoints", "wav2lip_gan.pth")  
 
 # Function to extract the zip file
 def extract_zip():
@@ -34,9 +34,16 @@ def extract_zip():
 def generate_speech(text, output_wav):
     try:
         tts = TextToSpeech()
-        # Use Trump's voice model
+
         speech = tts.tts(text, "trump")
-        save_wav(speech, output_wav)
+        
+        speech = speech.squeeze()  
+        if speech.dim() == 1:
+            speech = speech.unsqueeze(0)  
+        
+        sample_rate = 22050  
+        torchaudio.save(output_wav, speech.cpu(), sample_rate)
+        
         return True
     except Exception as e:
         print(f"Error generating speech: {str(e)}")
@@ -58,7 +65,6 @@ def run_wav2lip(video_path, audio_path, output_video):
         print(f"Error running Wav2Lip: {str(e)}")
         return False
 
-# Function to handle the lip-sync pipeline
 def process_lipsync(text):
     try:
         # Check if the zip file is extracted