Update app.py

app.py

@@ -6,7 +6,7 @@ import rembg
 import cv2
 import uuid
 uid=uuid.uuid4()
-
+if not os.path.exists(uid): os.makedirs(uid)
 load_js = """
 function(text_input, url_params) {
     console.log(text_input, url_params);
@@ -50,14 +50,14 @@ def capture_function(vid=None,frame_count=None):
         capture.set(cv2.CAP_PROP_POS_FRAMES, i)
         ret, frame_f = capture.read(i)
         frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
-        cv2.imwrite(f'{uid}-{i}.png',frame_ff)
-        fbox.append(f'{uid}-{i}.png')
+        cv2.imwrite(f'{uid}/{i}.png',frame_ff)
+        fbox.append(f'{uid}/{i}.png')
         
     capture.set(cv2.CAP_PROP_POS_FRAMES, int(frame_count)-1)
     ret, frame_f = capture.read()
     frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
-    cv2.imwrite(f'{uid}-{int(frame_count)-1}.png',frame_ff)
-    fbox.append(f'{uid}-{int(frame_count)-1}.png')
+    cv2.imwrite(f'{uid}/{int(frame_count)-1}.png',frame_ff)
+    fbox.append(f'{uid}/{int(frame_count)-1}.png')
     capture.release()
     return fbox,fbox
 
@@ -118,9 +118,12 @@ with gr.Blocks() as app:
                 frame_count=gr.Textbox()
                 outp_file=gr.Files()
     with gr.Tab("Frames"):
-        frame_btn = gr.Button("Get Frames")                
-        frame_gal = gr.Gallery(columns=6)
-        frame_file = gr.Files()
+        frame_btn = gr.Button("Get Frames")
+        with gr.Row():
+            with gr.Column():
+                frame_gal = gr.Gallery(columns=6,show_label=True)
+            with gr.Column():
+                frame_file = gr.Files()
         
     with gr.Tab("Rem BG"):
         with gr.Row():