Update app.py

app.py

@@ -32,16 +32,20 @@ def check_for_mp4_in_outputs(given_folder):
 
 
 def infer():
+
+    filepath = "./assets/examples/synthetic_1.mp4"
     # Get the current timestamp
     timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
 
     output_folder_name = f"results_{timestamp}"
 
     # Example: Run the inference script (replace with your actual command)
-    run_command(f"{sys.executable} inference_keep.py -i=./assets/examples/synthetic_1.mp4 -o={output_folder_name} --has_aligned --save_video -s=1")
+    run_command(f"{sys.executable} inference_keep.py -i={filepath} -o={output_folder_name} --has_aligned --save_video -s=1")
 
     # Call the function and print the result
-    mp4_file_path = check_for_mp4_in_outputs(output_folder_name)
+    this_infer_folder = os.path.splitext(os.path.basename(filepath))[0]
+    joined_path = os.path.join(output_folder_name, this_infer_folder)
+    mp4_file_path = check_for_mp4_in_outputs(joined_path)
     print(mp4_file_path)
 
     print(f"RESULT: {mp4_file_path}")