save to tmp

app.py

@@ -49,9 +49,9 @@ def clean(Seg_Tracker):
 
 def get_meta_from_video(Seg_Tracker, input_video, scale_slider, checkpoint):
 
-    output_dir = 'output_frames'
-    output_masks_dir = 'output_masks'
-    output_combined_dir = 'output_combined'
+    output_dir = '/tmp/output_frames'
+    output_masks_dir = '/tmp/output_masks'
+    output_combined_dir = '/tmp/output_combined'
     clear_folder(output_dir)
     clear_folder(output_masks_dir)
     clear_folder(output_combined_dir)
@@ -116,7 +116,7 @@ def mask2bbox(mask):
 
 def sam_stroke(Seg_Tracker, drawing_board, last_draw, frame_num, ann_obj_id):
     predictor, inference_state, image_predictor = Seg_Tracker
-    image_path = f'output_frames/{frame_num:07d}.jpg'
+    image_path = f'/tmp/output_frames/{frame_num:07d}.jpg'
     image = cv2.imread(image_path)
     image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     display_image = drawing_board["image"]
@@ -180,7 +180,7 @@ def sam_click(Seg_Tracker, frame_num, point_mode, click_stack, ann_obj_id, evt:
         labels=labels_dict[ann_frame_idx][ann_obj_id],
     )
 
-    image_path = f'output_frames/{ann_frame_idx:07d}.jpg'
+    image_path = f'/tmp/output_frames/{ann_frame_idx:07d}.jpg'
     image = cv2.imread(image_path)
     image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
@@ -232,8 +232,8 @@ def show_mask(mask, image=None, obj_id=None):
     return mask_image
 
 def show_res_by_slider(frame_per, click_stack):
-    image_path = 'output_frames'
-    output_combined_dir = 'output_combined'
+    image_path = '/tmp/output_frames'
+    output_combined_dir = '/tmp/output_combined'
     
     combined_frames = sorted([os.path.join(output_combined_dir, img_name) for img_name in os.listdir(output_combined_dir)])
     if combined_frames:
@@ -272,11 +272,11 @@ def zip_folder(folder_path, output_zip_path):
                 zipf.write(file_path, os.path.relpath(file_path, folder_path))
 
 def tracking_objects(Seg_Tracker, frame_num, input_video):
-    output_dir = 'output_frames'
-    output_masks_dir = 'output_masks'
-    output_combined_dir = 'output_combined'
-    output_video_path = 'output_video.mp4'
-    output_zip_path = 'output_masks.zip'
+    output_dir = '/tmp/output_frames'
+    output_masks_dir = '/tmp/output_masks'
+    output_combined_dir = '/tmp/output_combined'
+    output_video_path = '/tmp/output_video.mp4'
+    output_zip_path = '/tmp/output_masks.zip'
     clear_folder(output_masks_dir)
     clear_folder(output_combined_dir)
     if os.path.exists(output_video_path):