Update app.py

app.py

@@ -29,6 +29,8 @@ from torchao.quantization import Int8WeightOnlyConfig
 
 import aoti
 
+from moviepy.editor import VideoFileClip, AudioFileClip
+
 
 MODEL_ID = "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
 
@@ -246,7 +248,44 @@ def generate_video(
 
     export_to_video(output_frames_list, video_path, fps=FIXED_FPS)
 
-    return video_path, current_seed
+    with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as final_tmpfile:
+        video_with_audio_path = final_tmpfile.name
+
+    try:
+        # Загружаем видео
+        video_clip = VideoFileClip(video_no_audio_path)
+        
+        # Создаем тихий аудиоклип той же длительности
+        from moviepy.audio.AudioClip import AudioClip
+        silent_audio = AudioClip(lambda t: [0, 0], duration=video_clip.duration, fps=22050)
+        
+        # Устанавливаем аудио в видео
+        video_with_audio = video_clip.set_audio(silent_audio)
+        
+        # Сохраняем видео с аудио
+        video_with_audio.write_videofile(
+            video_with_audio_path,
+            codec='libx264',
+            audio_codec='aac',
+            verbose=False,
+            logger=None
+        )
+        
+        # Закрываем клипы для освобождения ресурсов
+        video_clip.close()
+        video_with_audio.close()
+        
+        # Удаляем временный файл без звука
+        import os
+        os.unlink(video_no_audio_path)
+        
+        return video_with_audio_path, current_seed
+        
+    except Exception as e:
+        print(f"Error adding audio: {e}")
+        # В случае ошибки возвращаем видео без звука
+        return video_no_audio_path, current_seed
+
 
 with gr.Blocks() as demo:
     gr.Markdown("# Fast 4 steps Wan 2.2 I2V (14B) with Lightning LoRA")