Update app.py

app.py

@@ -1,10 +1,11 @@
 import gradio as gr
 import cv2
 import numpy as np
+import ffmpeg
 
-def resize_and_stretch(video_path, direction):
+def resize_and_stretch_ffmpeg(video_path, direction):
     """
-    Уменьшает видео по горизонтали или вертикали на 25%, а затем растягивает обратно с помощью интерполяции Ланцоша.
+    Уменьшает видео по горизонтали или вертикали на 25%, а затем растягивает обратно с помощью интерполяции Ланцоша, используя ffmpeg.
 
     Args:
         video_path: Путь к видеофайлу.
@@ -14,57 +15,52 @@ def resize_and_stretch(video_path, direction):
         Путь к обработанному видеофайлу.
     """
 
-    cap = cv2.VideoCapture(video_path)
-    if not cap.isOpened():
-        raise ValueError("Не удалось открыть видеофайл.")
+    output_path = "output_video_ffmpeg.mp4"
 
-    # Получаем свойства видео
-    fps = cap.get(cv2.CAP_PROP_FPS)
-    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
-    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+    # Получаем свойства видео с помощью ffprobe
+    probe = ffmpeg.probe(video_path)
+    video_stream = next((stream for stream in probe['streams'] if stream['codec_type'] == 'video'), None)
+    width = int(video_stream['width'])
+    height = int(video_stream['height'])
 
     # Вычисляем новые размеры
     if direction == "horizontal":
         new_width = int(width * 0.75)
         new_height = height
+        scale_down_filter = f"scale={new_width}:{new_height}:flags=lanczos"
+        scale_up_filter = f"scale={width}:{height}:flags=lanczos"
     else:  # vertical
         new_width = width
         new_height = int(height * 0.75)
+        scale_down_filter = f"scale={new_width}:{new_height}:flags=lanczos"
+        scale_up_filter = f"scale={width}:{height}:flags=lanczos"
 
-    # Создаем объект VideoWriter для записи выходного видео
-    fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Или используйте другой кодек, если нужно
-    output_path = "output_video.mp4"
-    out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
-
-    while True:
-        ret, frame = cap.read()
-        if not ret:
-            break
-
-        # Уменьшаем размер кадра
-        resized_frame = cv2.resize(frame, (new_width, new_height), interpolation=cv2.INTER_LANCZOS4)
-
-        # Растягиваем обратно с помощью интерполяции Ланцоша
-        stretched_frame = cv2.resize(resized_frame, (width, height), interpolation=cv2.INTER_LANCZOS4)
-
-        # Записываем кадр в выходное видео
-        out.write(stretched_frame)
-
-    cap.release()
-    out.release()
+    try:
+        (
+            ffmpeg
+            .input(video_path)
+            .filter('scale', new_width, new_height, flags='lanczos')
+            .filter('scale', width, height, flags='lanczos')
+            .output(output_path, vcodec='libx264', crf=28) #Измените кодек и параметр crf при необходимости
+            .run(capture_stdout=True, capture_stderr=True)
+        )
+    except ffmpeg.Error as e:
+        print('stdout:', e.stdout.decode('utf8'))
+        print('stderr:', e.stderr.decode('utf8'))
+        raise e
 
     return output_path
 
-# Создаем интерфейс Gradio
+# Создаем интерфейс Gradio (остается таким же, как и в первом примере)
 iface = gr.Interface(
-    fn=resize_and_stretch,
+    fn=resize_and_stretch_ffmpeg,
     inputs=[
         gr.Video(label="Загрузите видео"),
         gr.Radio(["horizontal", "vertical"], label="Направление сжатия")
     ],
     outputs=gr.Video(label="Обработанное видео"),
     title="Сжатие и растягивание видео",
-    description="Загрузите видео, выберите направление сжатия (горизонтальное или вертикальное), и программа уменьшит его размер на 25% в этом направлении, а затем растянет обратно с помощью интерполяции Ланцоша."
+    description="Загрузите видео, выберите направление сжатия (горизонтальное или вертикальное), и программа уменьшит его размер на 25% в этом направлении, а затем растянет обратно с помощью интерполяции Ланцоша (с использованием ffmpeg)."
 )
 
 iface.launch()