Update app.py

app.py

@@ -65,10 +65,14 @@ def text_to_speech(text):
             audio_file.write(response.content)
         return audio_path
     else:
-        return f"Error: {response.status_code} - {response.text}"
+        print(f"Error: {response.status_code} - {response.text}")
+        return None  # Return None if there's an error
 
 # Function to create video from image and audio
 def generate_video(image_path, audio_path):
+    if audio_path is None:
+        return None  # Skip video generation if there's no valid audio file
+
     image_clip = ImageClip(image_path).set_duration(AudioFileClip(audio_path).duration)
     audio_clip = AudioFileClip(audio_path)
     video_clip = image_clip.set_audio(audio_clip)
@@ -92,6 +96,10 @@ with gr.Blocks(theme=theme) as demo:
     def process_image(image):
         roast_text = generate_roast(image)
         audio_path = text_to_speech(roast_text)
+        
+        if audio_path is None:
+            return "Error generating audio. Please try again.", None, None
+        
         video_path = generate_video(image, audio_path)
         return roast_text, audio_path, video_path