Update app.py

app.py

@@ -120,6 +120,14 @@ def create_video_from_audio(audio_path, images, output_path):
     video_clip.close()
     final_clip.close()
 
+def truncate_prompt(prompt, tokenizer, max_length=512):
+    """Truncate prompt to fit within the maximum token length."""
+    tokens = tokenizer.tokenize(prompt)
+    if len(tokens) > max_length:
+        tokens = tokens[:max_length]
+        prompt = tokenizer.convert_tokens_to_string(tokens)
+    return prompt
+
 @spaces.GPU
 def predict(
     prompt,
@@ -170,6 +178,10 @@ def predict(
         prompt = re.sub("([^\x00-\x7F]|\w)(\.|\。|\?)", r"\1 \2\2", prompt)
         if normalize_text and language == "vi":
             prompt = normalize_vietnamese_text(prompt)
+        
+        # Truncate prompt to fit within the maximum token length
+        prompt = truncate_prompt(prompt, tokenizer)
+        
         print("I: Generating new audio...")
         t0 = time.time()
         out = MODEL.inference(
@@ -197,7 +209,14 @@ def predict(
         
         # Tạo video từ file audio
         print("I: Generating video from audio...")
-        folder_name, sentences = pexels.generate_videos(prompt, pexels_api_key, "landscape", 1080, 1920, llm_chain, sum_llm_chain)
+        # Sử dụng UUID để tạo tên thư mục ngắn gọn
+        folder_name = f"video_{uuid.uuid4().hex}"
+        os.makedirs(folder_name, exist_ok=True)
+        folder_path = os.path.join(folder_name, "images")
+        os.makedirs(folder_path, exist_ok=True)
+        
+        # Tạo video từ file audio và các hình ảnh
+        folder_name, sentences = pexels.generate_videos(prompt, pexels_api_key, "landscape", 1080, 1920, llm_chain, sum_llm_chain, folder_path)
         utils.combine_videos(folder_name)
         video_path = os.path.join(folder_name, "Final_Ad_Video.mp4")