Update app.py

app.py

@@ -1,41 +1,50 @@
-import gradio as gr
-import numpy as np
 import cv2
+import numpy as np
+from ffmpeg_python import FFmpeg
+import gradio as gr
 
-def interpolate_frames(video_file, num_interpolations):
-    cap = cv2.VideoCapture(video_file)
-    frames = []
-    while cap.isOpened():
-        ret, frame = cap.read()
-        if not ret:
-            break
-        frames.append(frame)
-    cap.release()
-
-    # Convert frames to numpy array
-    frames = np.array(frames)
-
-    # Interpolate frames
-    interpolated_frames = []
-    for i in range(len(frames) - 1):
-        for j in range(num_interpolations + 1):
-            alpha = j / (num_interpolations + 1)
-            interpolated_frame = cv2.addWeighted(frames[i], 1 - alpha, frames[i + 1], alpha, 0)
-            interpolated_frames.append(interpolated_frame)
-
-    return interpolated_frames
-
-# Create a Gradio interface
-with gr.Blocks() as demo:
-    gr.Markdown("## Video Frame Interpolation")
-    video_input = gr.File(label="Input Video")
-    num_interpolations = gr.Slider(1, 10, step=1, label="Number of Interpolations")
-    video_output = gr.Video(label="Interpolated Video")
-    start_button = gr.Button("Start Interpolation")
+def interpolate_video(input_path, output_path):
+    # Загрузка видео
+    cap = cv2.VideoCapture(input_path)
+    
+    # Получение количества кадров и частоты кадров исходного видео
+    fps = int(cap.get(cv2.CAP_PROP_FPS))
+    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    
+    # Настройка нового FPS
+    new_fps = 60
+    
+    # Создание списка команд для ffmpeg
+    command = [
+        'ffmpeg',
+        '-i', input_path,
+        '-filter:v', f'fps={new_fps}',
+        output_path
+    ]
+    
+    # Запуск процесса интерполяции через ffmpeg
+    try:
+        process = FFmpeg(command)
+        out, err = process.run(capture_stdout=True, capture_stderr=True)
+        
+        if err:
+            return False, err.decode('utf-8')
+            
+        return True, "Интерполяция завершена успешно!"
+    except Exception as e:
+        return False, str(e)
 
-    # Define the event listener for the start button
-    start_button.click(interpolate_frames, inputs=[video_input, num_interpolations], outputs=video_output)
+# Интерфейс Gradio
+iface = gr.Interface(
+    fn=interpolate_video,
+    inputs=[
+        gr.inputs.Textbox(label="Введите путь к входному видео"),
+        gr.inputs.Textbox(label="Введите путь к выходному видео")
+    ],
+    outputs=[gr.outputs.Boolean(label="Успешность"), gr.outputs.Textbox(label="Сообщение")],
+    title="Интерполяция видео до 60 FPS",
+    description="Приложение для увеличения частоты кадров видео до 60 FPS."
+)
 
-# Launch the interface
 if __name__ == "__main__":
-    demo.launch(show_error=True)
+    iface.launch()