Update app.py

app.py

@@ -1,29 +1,87 @@
+import cv2
 import os
 import gradio as gr
-from rife_ncnn_vulkan import RIFE
-
-# Инициализация модели RIFE
-model = RIFE()
 
+# Функция для увеличения 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
 
-    # Используем RIFE для интерполяции кадров
-    model.process_video(input_video_path, output_path, target_fps)
+    cap.release()
+    output.release()
 
     return output_path
 
+# Gradio интерфейс
 def process_video(video_path):
-    return increase_fps(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 с использованием интерполяции RIFE. Загрузите видео с низким FPS, чтобы получить плавный результат.",
+    description=(
+        "Этот инструмент увеличивает частоту кадров видео до 60 FPS с использованием "
+        "оптического потока. Загрузите видео с низким FPS, чтобы получить плавный результат."
+    ),
     examples=[]
 )
 
+# Запуск интерфейса
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch(share=True)