Update app

app.py

@@ -56,6 +56,7 @@ print("Device: ", device)
 holistic = mp_holistic.Holistic(min_detection_confidence=0.5, min_tracking_confidence=0.5)
 
 
+
 def bb_intersection_over_union(boxA, boxB):
 	xA = max(boxA[0], boxB[0])
 	yA = max(boxA[1], boxB[1])
@@ -316,6 +317,7 @@ def process_video_asd(file, sd_root, work_root, data_root, avi_dir, tmp_dir, wor
 	except:
 		msg = "Error in pre-processing the video, please check the input video and try again"
 		return msg
+	print("YOLO done")
 
 	print("Detecting scenes in the video...")
 	try:
@@ -323,29 +325,22 @@ def process_video_asd(file, sd_root, work_root, data_root, avi_dir, tmp_dir, wor
 	except:
 		msg = "Error in detecting the scenes in the video, please check the input video and try again"
 		return msg
-	
+	print("Scene detect done")
+
 	print("Tracking video...")
 	allscenes = []
-	try:
-		for shot in scene:
-			if shot[1].frame_num - shot[0].frame_num >= min_track:
-				allscenes.append(track_shot(faces[shot[0].frame_num:shot[1].frame_num]))
-	except:
-		msg = "Error in tracking people in the video, please check the input video and try again"
-		return msg
+	for shot in scene:
+		if shot[1].frame_num - shot[0].frame_num >= min_track:
+			allscenes.append(track_shot(faces[shot[0].frame_num:shot[1].frame_num]))
 
 	print("Cropping video...")
 	alltracks = []
-	try:
-		for sc_num in range(len(allscenes)):
-			vidtracks = []
-			for ii, track in enumerate(allscenes[sc_num]):
-				os.makedirs(os.path.join(crop_dir, 'scene_'+str(sc_num)), exist_ok=True)
-				vidtracks.append(crop_video(avi_dir, tmp_dir, track, os.path.join(crop_dir, 'scene_'+str(sc_num), '%05d' % ii)))
-			alltracks.append(vidtracks)
-	except:
-		msg = "Error in cropping the video, please check the input video and try again"
-		return msg
+	for sc_num in range(len(allscenes)):
+		vidtracks = []
+		for ii, track in enumerate(allscenes[sc_num]):
+			os.makedirs(os.path.join(crop_dir, 'scene_'+str(sc_num)), exist_ok=True)
+			vidtracks.append(crop_video(avi_dir, tmp_dir, track, os.path.join(crop_dir, 'scene_'+str(sc_num), '%05d' % ii)))
+		alltracks.append(vidtracks)
 
 	savepath = os.path.join(work_dir, 'tracks.pckl')
 
@@ -361,6 +356,7 @@ def process_video_asd(file, sd_root, work_root, data_root, avi_dir, tmp_dir, wor
 	return "success"
 
 
+
 @spaces.GPU(duration=140)
 def preprocess_video(path, result_folder, apply_preprocess, padding=20):
 
@@ -1222,13 +1218,11 @@ def preprocess_asd(video_path, result_folder_input):
 	crop_dir = os.path.join(data_dir, 'pycrop')
 	frames_dir = os.path.join(data_dir, 'pyframes')
 
-	print("Pre-processing the input video...")
 	status = process_video_asd(file, sd_root, work_root, data_root, avi_dir, tmp_dir, work_dir, crop_dir, frames_dir)
-
+	print("Status ", status)
 	if status != "success":
 		return status
-	print("Successfully pre-processed the video")
-
+		
 	return "success"
 
 def process_video_syncoffset(video_path, num_avg_frames, apply_preprocess):