Update app.py

app.py

@@ -248,36 +248,40 @@ def generate_video(
 
     export_to_video(output_frames_list, video_path, fps=FIXED_FPS)
 
-    try:
-        # Команда ffmpeg для добавления тихого аудио
-        cmd = [
-            'ffmpeg',
-            '-f', 'lavfi', 
-            '-i', 'anullsrc=channel_layout=stereo:sample_rate=44100',
-            '-i', video_no_audio_path,
-            '-c:v', 'copy',          # Копируем видео без перекодирования
-            '-c:a', 'aac',           # Кодируем аудио в AAC
-            '-shortest',             # Обрезаем по короткому потоку
-            '-y',                    # Перезаписываем выходной файл
-            video_with_audio_path
-        ]
-        
-        # Запускаем ffmpeg
-        result = subprocess.run(cmd, capture_output=True, text=True, check=True)
-        
-        # Удаляем временный файл без звука
-        os.unlink(video_no_audio_path)
-        
-        return video_with_audio_path, current_seed
-        
-    except subprocess.CalledProcessError as e:
-        print(f"FFmpeg error: {e}")
-        print(f"FFmpeg stderr: {e.stderr}")
-        # В случае ошибки возвращаем видео без звука
-        return video_no_audio_path, current_seed
-    except Exception as e:
-        print(f"Unexpected error: {e}")
-        return video_no_audio_path, current_seed
+    if check_ffmpeg():
+        try:
+            # Создаем временный файл для видео с звуком
+            with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as audio_tmpfile:
+                video_with_audio_path = audio_tmpfile.name
+            
+            # Команда ffmpeg для добавления тихого аудио
+            cmd = [
+                'ffmpeg',
+                '-f', 'lavfi', 
+                '-i', 'anullsrc=channel_layout=stereo:sample_rate=44100',
+                '-i', video_path,
+                '-c:v', 'copy',
+                '-c:a', 'aac',
+                '-shortest',
+                '-y',
+                video_with_audio_path
+            ]
+            
+            # Запускаем ffmpeg
+            subprocess.run(cmd, capture_output=True, check=True)
+            
+            # Удаляем исходный файл без звука
+            os.unlink(video_path)
+            
+            return video_with_audio_path, current_seed
+            
+        except Exception as e:
+            print(f"Error adding audio: {e}")
+            # В случае ошибки возвращаем видео без звука
+            return video_path, current_seed
+    else:
+        print("FFmpeg not available, returning video without audio")
+        return video_path, current_seed
 
 
 with gr.Blocks() as demo:
@@ -329,5 +333,12 @@ with gr.Blocks() as demo:
         inputs=[input_image_component, prompt_input, steps_slider], outputs=[video_output, seed_input], fn=generate_video, cache_examples="lazy"
     )
 
+def check_ffmpeg():
+    try:
+        subprocess.run(['ffmpeg', '-version'], capture_output=True, check=True)
+        return True
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        return False
+
 if __name__ == "__main__":
     demo.queue().launch(mcp_server=True)