Update app.py

app.py

@@ -6,50 +6,54 @@ import os
 import ffmpeg
 
 def finalise_video(start_video):
-    try:
-        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as video_tmpfile:
-            if isinstance(start_video, str):
-                shutil.copy(start_video, video_tmpfile.name)
-            else:
-                start_video.save(video_tmpfile.name, format="MP4")
-            video_path = video_tmpfile.name
+    if check_ffmpeg():
+        try:
+            with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as video_tmpfile:
+                if isinstance(start_video, str):
+                    shutil.copy(start_video, video_tmpfile.name)
+                else:
+                    start_video.save(video_tmpfile.name, format="MP4")
+                video_path = video_tmpfile.name
+            
+            with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as audio_tmpfile:
+                video_with_audio_path = audio_tmpfile.name
+            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
+            ]
         
-        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as audio_tmpfile:
-            video_with_audio_path = audio_tmpfile.name
-        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
-        ]
-    
-        subprocess.run(cmd, capture_output=True, check=True)
-    
-        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as blur_tmpfile:
-            blurred_video_path = blur_tmpfile.name
-    
-        cmd_blur = [
-            'ffmpeg',
-            '-i', video_with_audio_path,  # Используем видео с аудио как источник
-            '-vf', 'gblur=sigma=25',       # Гауссово размытие с sigma=5
-            '-c:a', 'copy',               # Копируем аудио без изменений
-            '-y',
-            blurred_video_path
-        ]
-    
-        subprocess.run(cmd_blur, capture_output=True, check=True)
-
-        os.unlink(video_path)
-
-        return video_with_audio_path, blurred_video_path
-    except Exception as e:
-        print(f"Error adding audio: {e}")
-        return
+            subprocess.run(cmd, capture_output=True, check=True)
+        
+            with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as blur_tmpfile:
+                blurred_video_path = blur_tmpfile.name
+        
+            cmd_blur = [
+                'ffmpeg',
+                '-i', video_with_audio_path,  # Используем видео с аудио как источник
+                '-vf', 'gblur=sigma=25',       # Гауссово размытие с sigma=5
+                '-c:a', 'copy',               # Копируем аудио без изменений
+                '-y',
+                blurred_video_path
+            ]
+        
+            subprocess.run(cmd_blur, capture_output=True, check=True)
+        
+            os.unlink(video_path)
+        
+            return video_with_audio_path, blurred_video_path
+        except Exception as e:
+            print(f"Error finaliser: {e}")
+            return
+    else:
+        print("FFmpeg not available, returning video without audio")
+        return video_path, video_path, current_seed
     
 with gr.Blocks() as demo:
     gr.Markdown("Finaliser")
@@ -63,5 +67,14 @@ with gr.Blocks() as demo:
     
     ui_inputs = [input_video_component]
     generate_button.click(fn=finalise_video, inputs=ui_inputs, outputs=[video_output_1, video_output_2])
+
+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)