Update app.py

app.py

@@ -1,87 +1,42 @@
-import cv2
-import os
 import gradio as gr
-
-# Функция для увеличения FPS
-def increase_fps(input_video_path, target_fps=60):
-    output_path = "output.mp4"
-    cap = cv2.VideoCapture(input_video_path)
-    input_fps = cap.get(cv2.CAP_PROP_FPS)
-    frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
-    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
-    codec = cv2.VideoWriter_fourcc(*"mp4v")
-
-    # Проверка: если входное FPS >= 60, просто копируем файл
-    if input_fps >= target_fps:
-        cap.release()
-        return input_video_path  # Возвращаем оригинальное видео
-
-    # Расчет коэффициента интерполяции
-    ratio = target_fps / input_fps
-    output = cv2.VideoWriter(output_path, codec, target_fps, (width, height))
-    ret, prev_frame = cap.read()
-
-    # Проверка первого кадра
-    if not ret:
-        cap.release()
-        output.release()
-        raise ValueError("Не удалось прочитать кадры из видео.")
-
-    prev_frame_gray = cv2.cvtColor(prev_frame, cv2.COLOR_BGR2GRAY)
-
-    for _ in range(frame_count - 1):
-        ret, next_frame = cap.read()
-        if not ret:
-            break
-
-        # Преобразуем текущий кадр в черно-белый
-        next_frame_gray = cv2.cvtColor(next_frame, cv2.COLOR_BGR2GRAY)
-
-        # Убедимся, что размеры кадров совпадают
-        if prev_frame_gray.shape != next_frame_gray.shape:
-            next_frame_gray = cv2.resize(next_frame_gray, (prev_frame_gray.shape[1], prev_frame_gray.shape[0]))
-
-        # Оптический поток
-        flow = cv2.calcOpticalFlowFarneback(
-            prev_frame_gray, next_frame_gray, None, 0.5, 3, 15, 3, 5, 1.2, 0
-        )
-
-        # Генерация промежуточных кадров
-        for i in range(1, int(ratio)):
-            alpha = i / ratio
-            intermediate_frame = cv2.addWeighted(prev_frame_gray, 1 - alpha, next_frame_gray, alpha, 0)
-            intermediate_bgr = cv2.cvtColor(intermediate_frame, cv2.COLOR_GRAY2BGR)
-            output.write(intermediate_bgr)
-
-        output.write(next_frame)
-        prev_frame_gray = next_frame_gray
-
-    cap.release()
-    output.release()
-
-    return output_path
-
-# Gradio интерфейс
-def process_video(video_path):
-    try:
-        result_path = increase_fps(video_path)
-        return result_path
-    except Exception as e:
-        return f"Ошибка обработки: {str(e)}"
-
-interface = gr.Interface(
-    fn=process_video,
-    inputs=gr.Video(label="Загрузите видео для обработки"),
-    outputs=gr.Video(label="Видео с увеличенным FPS"),
-    title="Увеличение FPS до 60",
-    description=(
-        "Этот инструмент увеличивает частоту кадров видео до 60 FPS с использованием "
-        "оптического потока. Загрузите видео с низким FPS, чтобы получить плавный результат."
-    ),
-    examples=[]
+import numpy as np
+from PIL import Image
+
+def interpolate_frames(frame1, frame2, num_frames):
+    # Преобразовать изображения в массивы NumPy
+    frame1 = np.array(frame1)
+    frame2 = np.array(frame2)
+
+    # Вычислить разницу между двумя изображениями
+    diff = frame2 - frame1
+
+    # Создать список для хранения интерполированных кадров
+    interpolated_frames = []
+
+    # Интерполировать кадры
+    for i in range(num_frames):
+        t = i / (num_frames - 1)
+        interpolated_frame = frame1 + t * diff
+        interpolated_frame = interpolated_frame.astype(np.uint8)
+        interpolated_frames.append(Image.fromarray(interpolated_frame))
+
+    return interpolated_frames
+
+# Создать интерфейс Gradio
+demo = gr.Interface(
+    fn=interpolate_frames,
+    inputs=[
+        gr.Image(type="pil", label="Кадр 1"),
+        gr.Image(type="pil", label="Кадр 2"),
+        gr.Slider(label="Количество кадров", minimum=2, maximum=100, value=10)
+    ],
+    outputs=[
+        gr.Gallery(label="Интерполированные кадры")
+    ],
+    title="Интерполяция кадров",
+    description="Интерполируйте кадры между двумя изображениями"
 )
 
-# Запуск интерфейса
+# Запустить приложение
 if __name__ == "__main__":
-    interface.launch(share=True)
+    demo.launch()