Adding option for FPS/Min Tarck Length

.gitignore

@@ -1,5 +1,5 @@
 *.ipynb
-ipynb_checkpoints/
+*ipynb_checkpoints/
 *.mp4
 *.avi
 flagged/
@@ -10,4 +10,5 @@ data/
 raw_data/
 DataFeatures/
 app_trial.py
-models*
+models/
+*__pycache__/

app.py

@@ -81,7 +81,7 @@ def download_and_unzip_google_drive_file(file_id, output_path, unzip_path):
     
 
 # @spaces.GPU()
-def doo(video, progress=gr.Progress()):
+def doo(video, fps, min_track_length, progress=gr.Progress()):
     # download and unzip models
     # file_id = '1agsLD5HV_VmDNpDhjHXTCAVmGUm2IQ6p'
     # output_path = 'models.zip'
@@ -98,6 +98,7 @@ def doo(video, progress=gr.Progress()):
     test_video_list = ["video1"] # list of videos to generate features for
     exp_name = "collagen_motility_inference" # name of experiment
     feature_data_path = os.path.join(feature_dir, exp_name)
+    min_track_length = min_track_length
     
     
     #path to saved models
@@ -158,9 +159,9 @@ def doo(video, progress=gr.Progress()):
         progress(1 / 3, desc=f"Tracking {1}/{3}")
         track_bacteria(video_num=video_num, max_age=35, max_interpolation=35, data_path=feature_data_path)
         progress(2 / 3, desc=f"Tracking {2}/{3}")
-        folder_path = analyse_tracking(video_num=video_num, data_feature_path=feature_data_path, data_root_path=final_data_dir, plot=True)
+        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}")
-        output_video = gen_tracking_video(video_num=video_num, fps=60, data_path=feature_data_path)
+        output_video = gen_tracking_video(video_num=video_num, fps=fps, data_path=feature_data_path)
         final_video = os.path.basename(output_video)
         shutil.copy(output_video, final_video)
         print(output_video)
@@ -178,13 +179,23 @@ examples = [['RBS_2_4_h264.mp4'], ['RBS_4_4_h264.mp4']]# ['RBS_7_6_h264.mp4']]
 title = "🎞️ MEMTrack Bacteria Tracking Video Tool"
 description = "Upload a video or selct from example to track. <br><br> If the input video does not play on browser, ensure its in a browser accetable format. Output will be generated iirespective of playback on browser. Refer: https://colab.research.google.com/drive/1U5pX_9iaR_T8knVV7o4ftKdDoGndCdEM?usp=sharing"
 
+
+inputs = [
+    gr.Video(label="Input Video"),
+    gr.Slider(minimum=1, maximum=100, step=1, default=60, label="FPS", choices=[10, 20, 30, 60]),
+    gr.Slider(minimum=1, maximum=1000, step=1, default=60, label="Minimum Track Length Threshold", choices=[5, 10, 20, 50, 100])
+]
+
+outputs = [
+    gr.Video(label="Tracked Video"),
+    gr.Files(label="CSV Data")
+]
+
+
 iface = gr.Interface(
     fn=doo,
-    inputs=gr.Video(label="Input Video"),
-    outputs=[
-        gr.Video(label="Tracked Video"),
-        gr.Files(label="CSV Data")
-    ],
+    inputs=inputs,
+    outputs=outputs,
     examples=examples,
     title=title,
     description=description,
@@ -193,4 +204,4 @@ iface = gr.Interface(
 
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(share=True)