Update app.py

app.py

@@ -5,11 +5,13 @@ import cv2
 import numpy as np
 from PIL import Image, ImageDraw, ImageFont
 import os
+from concurrent.futures import ThreadPoolExecutor
+import asyncio
 
 app = FastAPI()
 
 # Function to split the script into smaller chunks
-def split_script(script: str, max_words: int = 50):
+def split_script(script: str, max_words: int = 30):
     words = script.split()
     chunks = [" ".join(words[i:i + max_words]) for i in range(0, len(words), max_words)]
     return chunks
@@ -19,7 +21,7 @@ def create_video_segment(script_chunk: str, background_color: str, text_color: s
     try:
         # Step 1: Convert script chunk to audio using gTTS
         tts = gTTS(script_chunk)
-        audio_file = "output_audio.mp3"
+        audio_file = f"output_audio_{os.urandom(4).hex()}.mp3"
         tts.save(audio_file)
 
         # Step 2: Create a blank image with text
@@ -48,7 +50,7 @@ def create_video_segment(script_chunk: str, background_color: str, text_color: s
         frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
 
         # Step 3: Create a video segment with the image and audio
-        video_segment_file = f"video_segment_{len(os.listdir())}.mp4"
+        video_segment_file = f"video_segment_{os.urandom(4).hex()}.mp4"
         fps = 24
         fourcc = cv2.VideoWriter_fourcc(*"mp4v")
         video_writer = cv2.VideoWriter(video_segment_file, fourcc, fps, (width, height))
@@ -105,11 +107,12 @@ async def generate_video(script: str, background_color: str = "#000000", text_co
         # Step 1: Split the script into smaller chunks
         script_chunks = split_script(script)
 
-        # Step 2: Generate video segments for each chunk
-        video_segment_files = []
-        for chunk in script_chunks:
-            video_segment = create_video_segment(chunk, background_color, text_color, font_size)
-            video_segment_files.append(video_segment)
+        # Step 2: Generate video segments in parallel
+        with ThreadPoolExecutor() as executor:
+            video_segment_files = list(executor.map(
+                lambda chunk: create_video_segment(chunk, background_color, text_color, font_size),
+                script_chunks
+            ))
 
         # Step 3: Combine video segments into a single video
         final_video_file = combine_video_segments(video_segment_files)