Update app.py

app.py

@@ -4,15 +4,7 @@ import cv2 as cv
 import gradio as gr
 
 
-def video_change(v):
-    vc = cv.VideoCapture(v)
-    fps = vc.get(cv.CAP_PROP_FPS)
-    fms = vc.get(cv.CAP_PROP_FRAME_COUNT)
-    w = vc.get(cv.CAP_PROP_FRAME_WIDTH)
-    h = vc.get(cv.CAP_PROP_FRAME_HEIGHT)
-    vc.release()
-    return fps, fms, w, h, gr.update(maximum=int(fms)-1)
-def video_change2(v, vc):
+def video_change(v, vc):
     if vc:
         vc.release()
     vc = cv.VideoCapture(v)
@@ -22,15 +14,7 @@ def video_change2(v, vc):
     h = vc.get(cv.CAP_PROP_FRAME_HEIGHT)
     return fps, fms, w, h, gr.update(maximum=int(fms)-1), vc
 
-
-def get_nth_frame(v, n):
-    vc = cv.VideoCapture(v)
-    vc.set(cv.CAP_PROP_POS_FRAMES, n)
-    ok, img = vc.read()
-    img = cv.cvtColor(img, cv.COLOR_BGR2RGB)
-    vc.release()
-    return img
-def get_nth_frame2(vc, n):
+def get_nth_frame(vc, n):
     vc.set(cv.CAP_PROP_POS_FRAMES, n)
     ok, img = vc.read()
     img = cv.cvtColor(img, cv.COLOR_BGR2RGB)
@@ -56,11 +40,11 @@ with gr.Blocks() as demo:
                     btn = gr.Button(value="Submit")
                     
             input_video.change(
-                video_change2, 
+                video_change, 
                 inputs=[input_video, state_vc], 
                 outputs=[output_fps, output_fms, output_w, output_h, input_n, state_vc]
             )
-            btn.click(get_nth_frame2, inputs=[state_vc, input_n], outputs=output_img)
+            btn.click(get_nth_frame, inputs=[state_vc, input_n], outputs=output_img)
         
         gr.Markdown("*****")
 
@@ -71,11 +55,11 @@ with gr.Blocks() as demo:
             return [ path+"/test.mp4", path+"/label.csv"]
         with gr.Row():
             df = gr.Dataframe(
+                value=[[1,2,3], [4,5,6]]
                 headers=["start_index", "end_index", "label"],
                 datatype=["number", "number", "number"],
-                row_count=0,
-                col_count=(3, "fixed"),
                 height=500,
+                interactive=False,
             )
             file = gr.File(file_count='multiple')
         btn_make_dataset = gr.Button(value="make dataset")