debug output video error with no tracks

MEMTrack/src/GenerateVideo.py

@@ -10,12 +10,16 @@ def create_video(data_dir, image_dir, video_name,fps, all_image_path=None, all_i
     plotted_images = natsorted(os.listdir(image_dir))
     images = plotted_images
     sample_image_dir = image_dir
+    if len(images) == 0:
+        all_images = True
+        print("Turning to all images as no tracked images")
+
     if all_images:
         all_images_list = natsorted(os.listdir(all_image_path))
         images = all_images_list
         sample_image_dir = all_image_path
-        
-        
+           
+
     img_sample = cv2.imread(os.path.join(sample_image_dir,images[0]))
     print(images[0])
     print(img_sample.shape)

MEMTrack/src/TrackingAnalysis.py

@@ -166,6 +166,7 @@ def analyse_tracking(video_num, min_track_length=60, custom_test_dir=None, data_
 
 
     track_id_details = groupby_trackid(tracking_data_filtered)
+    track_id_unfiltered_length = track_id_details
     num_frames, num_fragments, fragment_lengths, skipped_frames, track_id_filtered_length =analyze_track_preds(track_id_details,filter_length=True,min_con_length=min_track_length)
     
     try:

app.py

@@ -79,6 +79,8 @@ def download_and_unzip_google_drive_file(file_id, output_path, unzip_path):
     with zipfile.ZipFile(output_path, 'r') as zip_ref:
         zip_ref.extractall(unzip_path)    
     
+def clear_form():
+    return None, None, None, 60, 60  # Resetting inputs and sliders to default or empty states
 
 # @spaces.GPU()
 def doo(video, tiff_stack, images, fps, min_track_length, progress=gr.Progress()):
@@ -161,11 +163,16 @@ def doo(video, tiff_stack, images, fps, min_track_length, progress=gr.Progress()
         progress(2 / 3, desc=f"Tracking {2}/{3}")
         folder_path = analyse_tracking(video_num=video_num, min_track_length=min_track_length, data_feature_path=feature_data_path, data_root_path=final_data_dir, plot=True)
         progress(3 / 3, desc=f"Tracking {3}/{3}")
+        # Generate video with tracked frames only at the given filter threshold 
         output_video1 = gen_tracking_video(video_num=video_num, fps=fps, data_path=feature_data_path)
+        # Generate video with all frames at the given filter threshold
         output_video2 = gen_tracking_video(video_num=video_num, fps=fps, data_path=feature_data_path, all_images=True)
-        final_videos = [os.path.basename(output_video1), os.path.basename(output_video2)]
+        # Generate video with all frames with all tracks, so no filter threshold
+        output_video3 = gen_tracking_video(video_num=video_num, fps=fps, data_path=feature_data_path, all_images=True)
+        final_videos = [os.path.basename(output_video1), os.path.basename(output_video2), os.path.basename(output_video3)]
         shutil.copy(output_video1, final_videos[0])
         shutil.copy(output_video2, final_videos[1])
+        shutil.copy(output_video3, final_videos[2])
         print(output_video1)
         print(final_videos)
         
@@ -173,7 +180,7 @@ def doo(video, tiff_stack, images, fps, min_track_length, progress=gr.Progress()
         tracking_preds = find_and_return_csv_files(folder_path, search_pattern)
          
  
-    return final_videos[0], final_videos[1], tracking_preds
+    return final_videos[0], final_videos[1], final_videos[2], tracking_preds
 
 
 
@@ -207,15 +214,28 @@ if __name__ == "__main__":
                 # Common sliders for all inputs
                 fps_slider = gr.Slider(minimum=1, maximum=100, step=1, value=60, label="Output Video FPS")
                 track_length_slider = gr.Slider(minimum=0, maximum=1000, step=1, value=60, label="Minimum Track Length Threshold")
+
+                # Submit and Clear buttons
+                submit_button = gr.Button("Submit")
+                clear_button = gr.Button("Clear")
+                
+                
+                clear_button.click(
+                    fn=clear_form,
+                    inputs=[],
+                    outputs=[video_input, tiff_input, image_input, fps_slider, track_length_slider]
+                )
+
             with gr.Column():
                     outputs = [
                         gr.Video(label="Tracked Video (tracked frames)"),
                         gr.Video(label="Tracked Video (all frames)"),
+                        gr.Video(label="Tracked Video (all frames, all tracks)"),
                         gr.Files(label="CSV Data")
                     ]
             
-    
-        submit_button = gr.Button("Process Input")
+        
+        # submit_button = gr.Button("Process Input")
         submit_button.click(
             fn=doo,
             inputs=[video_input, tiff_input, image_input, fps_slider, track_length_slider],