global variables are set to gradio state

app.py

@@ -41,19 +41,6 @@ siammask = Custom(anchors=cfg['anchors'])
 siammask = load_pretrain(siammask, pretrained_path1)
 siammask = siammask.eval().to(device)
 
-# datetime object containing current date and time
-now = datetime.now()
- 
-dt_string = now.strftime("%d_%m_%Y_%H_%M_%S")
-print("date and time =", dt_string)
-# constants
-object_x = 0
-object_y = 0
-object_width = 0
-object_height = 0
-in_fps = 24
-
-
 parser = argparse.ArgumentParser()
 # parser.add_argument('--opt', default='configs/object_removal.yaml',
 #                     help='Please select your config file for inference')
@@ -69,7 +56,7 @@ parser.add_argument(
 parser.add_argument(
     '--outroot', default=os.path.abspath(project_name), help="output directory")
 parser.add_argument(
-    '--outfilename', default=dt_string+"_result.mp4", help="output filename")
+    '--outfilename', default="result.mp4", help="output filename")
 parser.add_argument('--consistencyThres', dest='consistencyThres', default=5, type=float,
                     help='flow consistency error threshold')
 parser.add_argument('--alpha', dest='alpha', default=0.1, type=float)
@@ -144,7 +131,7 @@ def getBoundaries(mask):
     return x1, y1, (x2-x1), (y2-y1)
 
 
-def track_and_mask(vid, masked_frame, original_list, mask_list):
+def track_and_mask(vid, masked_frame, original_list, mask_list, in_fps, dt_string):
     x, y, w, h = getBoundaries(masked_frame)
     f = 0
 
@@ -197,19 +184,23 @@ def track_and_mask(vid, masked_frame, original_list, mask_list):
     else:
         print("can't open the given input video file!")
 
+    outname = (dt_string+"_output.avi")
     print('Original Frame Count: ',len(original_list))
     print('Mask Frame Count: ',len(mask_list))
-    return {out_video_inpaint:(dt_string+"_output.avi"),
+    return {out_video_inpaint:outname,
             original_frame_list: original_list,
-            mask_list: mask_list}
+            mask_list: mask_list,
+            in_fps: in_fps}
 
 
 def inpaint_video(original_frame_list, mask_list):
+    outname = (dt_string+"_result.mp4")
     args.out_fps = in_fps
+    args.outfilename = outname
     video_inpainting(args, original_frame_list, mask_list)
     original_frame_list = []
     mask_list = []
-    return {out_video_inpaint:(dt_string+"_result.mp4"),
+    return {out_video_inpaint:outname,
             original_frame_list: original_frame_list,
             mask_list: mask_list}
 
@@ -250,6 +241,9 @@ def getStartEndPoints(mask):
 with gr.Blocks() as demo:
     original_frame_list = gr.State([])
     mask_list = gr.State([])
+    # constants
+    in_fps = gr.State(24)
+    dt_string = gr.State(datetime.now().strftime("%d_%m_%Y_%H_%M_%S"))
     with gr.Row():
         with gr.Column(scale=2):
             with gr.Row():
@@ -274,7 +268,7 @@ with gr.Blocks() as demo:
     approve_mask.click(lambda x: [x['image'], x['mask']], first_frame, [
                        original_image, masked_image])
     track_mask.click(fn=track_and_mask, inputs=[
-                     in_video, masked_image, original_frame_list, mask_list], outputs=[out_video, original_frame_list, mask_list])
+                     in_video, masked_image, original_frame_list, mask_list, in_fps, dt_string], outputs=[out_video, original_frame_list, mask_list, in_fps])
     inpaint.click(fn=inpaint_video, outputs=[out_video_inpaint, original_frame_list, mask_list])