Update app.py

app.py

@@ -1,7 +1,6 @@
 import os
 import cv2 as cv
 
-
 import gradio as gr
 
 
@@ -21,31 +20,48 @@ def get_nth_frame(v, n):
     img = cv.cvtColor(img, cv.COLOR_BGR2RGB)
     vc.release()
     return img
-    
+
+
+
+
 with gr.Blocks() as demo2:
-    with gr.Tab("get nth frame"):
+    with gr.Tab("Create Image Classify Dataset"):
         with gr.Row():
             with gr.Column():
                 input_video = gr.Video()
-                with gr.Accordion('video info'):
+                with gr.Accordion('Video Info'):
                     output_fps = gr.Textbox(lines=1, label="fps")
                     output_fms = gr.Textbox(lines=1, label="frame count")
                     output_w = gr.Textbox(lines=1, label="width")
                     output_h = gr.Textbox(lines=1, label="height")
             with gr.Column():
                 output_img = gr.Image()
-                input_n = gr.Slider(0, 9999, value=0, step=1, label="n")
-                btn = gr.Button(value="Submit")
+                with gr.Accordion('Show Frame'):
+                    input_n = gr.Slider(0, 9999, value=0, step=1, label="n")
+                    btn = gr.Button(value="Submit")
             input_video.change(
                 video_change, 
                 inputs=[input_video], 
                 outputs=[output_fps, output_fms, output_w, output_h, input_n]
             )
-            
             btn.click(get_nth_frame, inputs=[input_video, input_n], outputs=output_img)
+        
+        gr.Markdown("*****")
+
+        df = gr.Dataframe(
+            headers=["start index", "end index", "label"],
+            datatype=["number", "number", "number"],
+            row_count=(0, 'dynamic'),
+            col_count=(3, "fixed"),
+        )
             
-        gr.Examples([os.path.join(os.path.dirname(__file__), "test.mp4")], inputs=input_video)
+        btn_add_segm = gr.Button(value="create dataset")
+        #btn.click(add_segment, inputs=segms, outputs=segms)
         
+        gr.Markdown("*****")
+        
+        gr.Examples([os.path.join(os.path.dirname(__file__), "test.mp4")], inputs=input_video)
+