Update app.py

app.py

@@ -3,6 +3,7 @@ import random
 import re
 import struct
 import gradio as gr
+import subprocess
 
 # Основная функция для обработки видео
 def glitch_video(file, mode, countframes, positframes, audio, firstframe, kill):
@@ -76,15 +77,15 @@ def glitch_video(file, mode, countframes, positframes, audio, firstframe, kill):
     elif mode == "reverse":
         final = clean[::-1]
     elif mode == "invert":
-        final = list(chain(*zip(clean[0::2], clean[1::2])))
+        final = sum(zip(clean[1::2], clean[::2]), ())
     elif mode == "bloom":
         final = clean[:int(positframes)] + [clean[int(positframes)]] * int(countframes) + clean[int(positframes):]
     elif mode == "pulse":
-        final = [frame for i, frame in enumerate(clean) for _ in range(int(countframes)) if i % int(positframes) == 0] + clean
+        final = [val for i, val in enumerate(clean) for _ in range(int(countframes)) if i % int(positframes) == 0] + clean
     elif mode == "jiggle":
         final = [clean[min(max(0, x + int(random.gauss(0, int(positframes)))), len(clean) - 1)] for x in range(len(clean))]
     elif mode == "overlap":
-        final = [frame for sublist in [clean[i:i+int(countframes)] for i in range(0, len(clean), int(positframes))] for frame in sublist]
+        final = [item for sublist in [clean[i:i+int(countframes)] for i in range(0, len(clean), int(positframes)) ] for item in sublist]
 
     # Сборка видео обратно
     fileout = f"{os.path.splitext(filein)[0]}-{mode}.avi"
@@ -100,13 +101,19 @@ def glitch_video(file, mode, countframes, positframes, audio, firstframe, kill):
                 wr.write(rd.read(size))
         bstream_until_marker(temp_idx1, fileout)
 
+    # Конвертация в .mp4 для обеспечения совместимости
+    fileout_mp4 = f"{os.path.splitext(filein)[0]}-{mode}.mp4"
+    cmd = f"ffmpeg -i {fileout} -c:v libx264 -crf 18 -preset veryfast {fileout_mp4}"
+    subprocess.run(cmd, shell=True, check=True)
+
     # Очистка временных файлов
     os.remove(temp_hdrl)
     os.remove(temp_movi)
     os.remove(temp_idx1)
+    os.remove(fileout)
     os.rmdir(temp_dir)
 
-    return fileout
+    return fileout_mp4
 
 # Интерфейс Gradio
 with gr.Blocks() as demo: