Update app.py

app.py

@@ -1,73 +1,31 @@
-import cv2
-import subprocess
 import gradio as gr
+import ffmpeg
 
-def interpolate_video(input_file, output_file):
-    # Проверка пути к входному видео
-    cap = cv2.VideoCapture(input_file)
-    if not cap.isOpened():
-        return False, "Ошибка: не удалось открыть входное видео."
-    
-    # Получение информации о входном видео
-    fps = cap.get(cv2.CAP_PROP_FPS)
-    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-    
-    # Закрытие объекта VideoCapture
-    cap.release()
-    
-    # Проверка на корректность значений FPS
-    if fps <= 0 or total_frames <= 0:
-        return False, "Ошибка: некорректное видео или не удалось определить FPS/количество кадров."
-    
-    # Новый FPS
-    new_fps = 60
-    
-    # Команда для FFmpeg
-    command = [
-        'ffmpeg',
-        '-i', input_file,
-        '-filter:v', f'fps={new_fps}',
-        output_file
-    ]
-    
-    try:
-        # Запуск FFmpeg через subprocess
-        result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
-        
-        # Проверка результата
-        if result.returncode != 0:
-            return False, f"Ошибка FFmpeg: {result.stderr}"
-        
-        return True, f"Интерполяция завершена успешно! Видео сохранено в {output_file}"
-    except Exception as e:
-        return False, f"Произошла ошибка: {str(e)}"
+def interpolate_video(video, fps):
+    input_path = video.name
+    output_path = "interpolated_video.mp4"
 
-# Интерфейс Gradio
-with gr.Blocks() as iface:
-    gr.Markdown("# Интерполяция видео до 60 FPS")
-    
-    with gr.Row():
-        input_file = gr.File(label="Выберите входное видео")
-        output_file = gr.Textbox(label="Введите имя выходного файла (с .mp4)")
-    
-    with gr.Row():
-        submit_button = gr.Button("Отправить")
-        clear_button = gr.Button("Очистить")
-    
-    success_output = gr.Checkbox(label="Успешно")
-    message_output = gr.Textbox(label="Сообщение")
-    
-    submit_button.click(
-        fn=interpolate_video,
-        inputs=[input_file, output_file],
-        outputs=[success_output, message_output]
-    )
-    
-    clear_button.click(
-        fn=lambda: (False, ""),
-        inputs=[],
-        outputs=[success_output, message_output]
+    # Команда ffmpeg для интерполяции кадров
+    (
+        ffmpeg
+        .input(input_path)
+        .filter('fps', fps=fps, round='up')
+        .output(output_path)
+        .run(overwrite_output=True)
     )
 
+    return output_path
+
+iface = gr.Interface(
+    fn=interpolate_video,
+    inputs=[
+        gr.File(label="Upload Video"),
+        gr.Slider(minimum=24, maximum=60, step=1, value=60, label="Frame Rate (FPS)")
+    ],
+    outputs=gr.Video(label="Interpolated Video"),
+    title="Frame Interpolation App",
+    description="Upload a video and adjust the slider to set the target frame rate (FPS) for interpolation."
+)
+
 if __name__ == "__main__":
     iface.launch()