Update app.py

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 import torch
 import numpy as np
 from simple_salesforce import Salesforce
+import time
 
 # Salesforce connection details (replace these with your actual credentials)
 sf = Salesforce(username='prashanth@safetyanaluzer.com', password='SaiPrash461', security_token='MOA3BXBfGyqnjBneog8a9IcGw')
@@ -39,7 +40,7 @@ except Exception as e:
     device = torch.device("cpu")
 
 # ==========================
-# Load Model
+# Load Model (Use YOLOv8n for Faster Inference)
 # ==========================
 if os.path.isfile(MODEL_PATH):
     selected_model = MODEL_PATH
@@ -58,7 +59,7 @@ except Exception as e:
 # ==========================
 # Video Processing with Optimizations
 # ==========================
-def process_video(video_path, frame_skip=5):
+def process_video(video_path, frame_skip=5, max_frames=100):
     try:
         video = cv2.VideoCapture(video_path)
         if not video.isOpened():
@@ -67,17 +68,19 @@ def process_video(video_path, frame_skip=5):
         frame_count = 0
         violations = []
         processed_frame_count = 0
+        start_time = time.time()
 
         while True:
             ret, frame = video.read()
             if not ret:
                 break
 
-            # Skip frames to reduce processing time
+            # Skip frames to reduce processing time (process only every 'frame_skip' frame)
             if frame_count % frame_skip != 0:
                 frame_count += 1
                 continue
 
+            # Model inference for detecting violations
             results = model(frame, device=device)
 
             for result in results:
@@ -96,7 +99,15 @@ def process_video(video_path, frame_skip=5):
 
             frame_count += 1
             processed_frame_count += 1
-            if processed_frame_count >= 100:  # Limit processing to 100 frames to reduce time
+
+            # Stop processing after a fixed number of frames to save time
+            if processed_frame_count >= max_frames:
+                break
+
+            # Check elapsed time, stop if we exceed 30 seconds
+            elapsed_time = time.time() - start_time
+            if elapsed_time > 30:
+                print("⏰ Exceeded 30 seconds of processing time.")
                 break
 
         video.release()
@@ -184,6 +195,7 @@ def upload_video_to_salesforce(video_path):
     except Exception as e:
         print(f"❌ Error uploading video to Salesforce: {e}")
         return None
+
 # ==========================
 # Gradio Interface
 # ==========================