some simplifications;

app.py

@@ -69,27 +69,24 @@ def draw_cockpit(frame, top_pred,cnt):
     # Bullet points:
     high_danger_color = (255,0,0)
     low_danger_color = yellowgreen = (154,205,50)
+
     shark_sighted = 'Shark Detected: ' + str(top_pred['shark_sighted'])
     human_sighted = 'Number of Humans: ' + str(top_pred['human_n'])
     shark_size_estimate = 'Biggest shark size: ' + str(top_pred['biggest_shark_size'])
     shark_weight_estimate = 'Biggest shark weight: ' + str(top_pred['biggest_shark_weight'])
     danger_level = 'Danger Level: ' 
     danger_level += 'High' if top_pred['dangerous_dist'] else 'Low'
-    danger_color = 'orangered' if top_pred['dangerous_dist'] else 'yellowgreen'
-    # Create a list of strings to plot
+    danger_color = high_danger_color if top_pred['dangerous_dist'] else low_danger_color # 'orangered' for high danger , 'yellowgreen' for low danger'
+   
+   # Create a list of strings to plot
     strings = [shark_sighted, human_sighted, shark_size_estimate, shark_weight_estimate, danger_level]
     relative = max(frame.shape[0],frame.shape[1])
-    if top_pred['shark_sighted'] and top_pred['dangerous_dist'] and cnt%2 == 0:
+    if top_pred['shark_sighted'] and (cnt % 2) == 0:
         relative = max(frame.shape[0],frame.shape[1])
-        frame  = add_border(frame, color=high_danger_color, thickness=int(relative*0.025))
-    elif top_pred['shark_sighted'] and not top_pred['dangerous_dist'] and cnt%2 == 0:
-         relative = max(frame.shape[0],frame.shape[1])
-         frame  = add_border(frame, color=low_danger_color, thickness=int(relative*0.025))
-    overlay_text_on_image(frame, strings, font=cv2.FONT_HERSHEY_SIMPLEX, font_size=1.5, font_thickness=3, margin=int(relative*0.05), color=(255, 255, 255))
+        frame  = add_border(frame, color=danger_color, thickness=int(relative*0.025))
+    frame = overlay_text_on_image(frame, strings, font=cv2.FONT_HERSHEY_SIMPLEX, font_size=1.5, font_thickness=3, margin=int(relative*0.05), color=(255, 255, 255))
     return frame
     
-    
-
 def process_video(input_video, out_fps = 'auto', skip_frames = 7):
     cap = cv2.VideoCapture(input_video)
 
@@ -110,6 +107,7 @@ def process_video(input_video, out_fps = 'auto', skip_frames = 7):
     cnt = 0
     
     while iterating:
+        print(cnt)
         if (cnt % skip_frames) == 0:
             # flip frame vertically
             display_frame, result = inference_frame_serial(frame)
@@ -121,12 +119,11 @@ def process_video(input_video, out_fps = 'auto', skip_frames = 7):
             frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
             prediction_frame = cv2.cvtColor(display_frame, cv2.COLOR_BGR2RGB)
             
-            
             frame = cv2.resize(frame, (int(width*4), int(height*4)))
             
-            if cnt*skip_frames %2==0 and top_pred['shark_sighted']: 
+            if ((cnt*skip_frames) % 2==0): # and top_pred['shark_sighted']: 
                 prediction_frame = cv2.resize(prediction_frame, (int(width*4), int(height*4)))
-                frame =prediction_frame
+                frame = prediction_frame
              
             if  top_pred['shark_sighted']: 
                 frame = draw_cockpit(frame, top_pred,cnt*skip_frames)
@@ -137,6 +134,7 @@ def process_video(input_video, out_fps = 'auto', skip_frames = 7):
             yield prediction_frame,frame , None, pred_dashbord
         cnt += 1
         iterating, frame = cap.read()
+        print('interating: ', iterating)
     
     video.release()
     yield None, None, output_path, None