GiorgioV commited on
Commit
760bf10
·
verified ·
1 Parent(s): b34832d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -1
app.py CHANGED
@@ -288,11 +288,32 @@ def generate_video(
288
 
289
  # Запускаем ffmpeg
290
  subprocess.run(cmd, capture_output=True, check=True)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
291
 
292
  # Удаляем исходный файл без звука
293
  os.unlink(video_path)
 
 
 
 
294
 
295
- return video_with_audio_path, current_seed
296
 
297
  except Exception as e:
298
  print(f"Error adding audio: {e}")
 
288
 
289
  # Запускаем ffmpeg
290
  subprocess.run(cmd, capture_output=True, check=True)
291
+
292
+ # Создаем временный файл для заблюренного видео
293
+ with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as blur_tmpfile:
294
+ blurred_video_path = blur_tmpfile.name
295
+
296
+ # Команда ffmpeg для создания гауссова размытия
297
+ cmd_blur = [
298
+ 'ffmpeg',
299
+ '-i', video_with_audio_path, # Используем видео с аудио как источник
300
+ '-vf', 'gblur=sigma=5', # Гауссово размытие с sigma=5
301
+ '-c:a', 'copy', # Копируем аудио без изменений
302
+ '-y',
303
+ blurred_video_path
304
+ ]
305
+
306
+ # Запускаем ffmpeg для создания блюра
307
+ subprocess.run(cmd_blur, capture_output=True, check=True)
308
 
309
  # Удаляем исходный файл без звука
310
  os.unlink(video_path)
311
+ video_paths = {
312
+ 1: video_with_audio_path,
313
+ 2: blurred_video_path
314
+ }
315
 
316
+ return video_paths, current_seed
317
 
318
  except Exception as e:
319
  print(f"Error adding audio: {e}")