Update app.py

app.py

@@ -1,13 +1,20 @@
 import gradio as gr
 import ffmpeg
-import time
 import os
+import time
 
 def interpolate_video(video, fps):
     input_path = video.name
     output_path = "interpolated_video.mp4"
 
-    # Интерполяция кадров с использованием фильтра minterpolate
+    # Проверяем входное видео
+    try:
+        probe = ffmpeg.probe(input_path)
+        total_duration = float(probe['format']['duration'])
+    except ffmpeg.Error as e:
+        return f"Ошибка при анализе видео: {e}"
+
+    # Начинаем процесс интерполяции
     process = (
         ffmpeg
         .input(input_path)
@@ -17,33 +24,42 @@ def interpolate_video(video, fps):
         .run_async(pipe_stdout=True, pipe_stderr=True)
     )
 
-    # Вставим progress bar
-    total_duration = float(ffmpeg.probe(input_path)['format']['duration'])
     progress = 0
     while process.poll() is None:
         time.sleep(0.5)  # Обновление каждые 0.5 секунды
-        stderr_output = process.stderr.read().decode('utf-8')
-        if "frame=" in stderr_output:
-            current_time = float(stderr_output.split('time=')[-1].split(' ')[0].replace(':',''))  # Извлечение текущего времени кадра
-            progress = (current_time / total_duration) * 100
-        yield progress  # Обновление значения progress bar
+        # Чтение строки stderr
+        stderr_line = process.stderr.readline().decode('utf-8').strip()
+        if "time=" in stderr_line:
+            try:
+                # Парсим текущий прогресс
+                time_match = stderr_line.split("time=")[-1].split(" ")[0]
+                h, m, s = map(float, time_match.split(":"))
+                current_time = h * 3600 + m * 60 + s
+                progress = (current_time / total_duration) * 100
+            except Exception:
+                pass
+        yield f"Прогресс: {progress:.2f}%"
 
     process.wait()
-    
-    return output_path
 
+    if os.path.exists(output_path):
+        return output_path
+    else:
+        return "Ошибка в обработке видео."
+
+# Интерфейс Gradio
 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)")
+        gr.File(label="Загрузить видео"),
+        gr.Slider(minimum=24, maximum=60, step=1, value=60, label="Частота кадров (FPS)")
     ],
     outputs=[
-        gr.Video(label="Interpolated Video"),
-        gr.Progress(label="Processing Progress")
+        gr.Video(label="Интерполированное видео"),
+        gr.Text(label="Прогресс обработки")
     ],
-    title="Frame Interpolation App",
-    description="Upload a video and adjust the slider to set the target frame rate (FPS) for interpolation."
+    title="Приложение для интерполяции видео",
+    description="Загрузите видео и выберите желаемую частоту кадров (FPS)."
 )
 
 if __name__ == "__main__":