Yaroslav commited on
Commit
65f9346
·
verified ·
1 Parent(s): 4307a99

fixing buuugs

Browse files
Files changed (1) hide show
  1. app.py +7 -23
app.py CHANGED
@@ -1,18 +1,11 @@
1
  import cv2
2
- import numpy as np
3
  import os
4
  import gradio as gr
5
 
6
  # Функция для увеличения FPS
7
- def increase_fps(input_video, target_fps=60):
8
- input_path = "input.mp4"
9
  output_path = "output.mp4"
10
-
11
- # Сохранение входного видео
12
- with open(input_path, "wb") as f:
13
- f.write(input_video.read())
14
-
15
- cap = cv2.VideoCapture(input_path)
16
  input_fps = cap.get(cv2.CAP_PROP_FPS)
17
  frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
18
  width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
@@ -21,9 +14,8 @@ def increase_fps(input_video, target_fps=60):
21
 
22
  # Проверка: если входное FPS >= 60, просто копируем файл
23
  if input_fps >= target_fps:
24
- os.rename(input_path, output_path)
25
- with open(output_path, "rb") as f:
26
- return f.read()
27
 
28
  # Расчет коэффициента интерполяции
29
  ratio = target_fps / input_fps
@@ -55,19 +47,11 @@ def increase_fps(input_video, target_fps=60):
55
  cap.release()
56
  output.release()
57
 
58
- # Чтение результата
59
- with open(output_path, "rb") as f:
60
- output_video = f.read()
61
-
62
- # Очистка временных файлов
63
- os.remove(input_path)
64
- os.remove(output_path)
65
-
66
- return output_video
67
 
68
  # Gradio интерфейс
69
- def process_video(video):
70
- return increase_fps(video)
71
 
72
  interface = gr.Interface(
73
  fn=process_video,
 
1
  import cv2
 
2
  import os
3
  import gradio as gr
4
 
5
  # Функция для увеличения FPS
6
+ def increase_fps(input_video_path, target_fps=60):
 
7
  output_path = "output.mp4"
8
+ cap = cv2.VideoCapture(input_video_path)
 
 
 
 
 
9
  input_fps = cap.get(cv2.CAP_PROP_FPS)
10
  frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
11
  width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
 
14
 
15
  # Проверка: если входное FPS >= 60, просто копируем файл
16
  if input_fps >= target_fps:
17
+ cap.release()
18
+ return input_video_path # Возвращаем оригинальное видео
 
19
 
20
  # Расчет коэффициента интерполяции
21
  ratio = target_fps / input_fps
 
47
  cap.release()
48
  output.release()
49
 
50
+ return output_path
 
 
 
 
 
 
 
 
51
 
52
  # Gradio интерфейс
53
+ def process_video(video_path):
54
+ return increase_fps(video_path)
55
 
56
  interface = gr.Interface(
57
  fn=process_video,