import os
import gradio as gr
from rife_ncnn_vulkan import RIFE

# Инициализация модели RIFE
model = RIFE()

def increase_fps(input_video_path, target_fps=60):
    output_path = "output.mp4"

    # Используем RIFE для интерполяции кадров
    model.process_video(input_video_path, output_path, target_fps)

    return output_path

def process_video(video_path):
    return increase_fps(video_path)

interface = gr.Interface(
    fn=process_video,
    inputs=gr.Video(label="Загрузите видео для обработки"),
    outputs=gr.Video(label="Видео с увеличенным FPS"),
    title="Увеличение FPS до 60",
    description="Этот инструмент увеличивает частоту кадров видео до 60 FPS с использованием интерполяции RIFE. Загрузите видео с низким FPS, чтобы получить плавный результат.",
    examples=[]
)

if __name__ == "__main__":
    interface.launch()