Update app.py

app.py

@@ -41,6 +41,26 @@ client = openai.OpenAI(api_key=api_key)
 stop_capture = False
 alerts_mode = True
 
+def encode_to_video_fast(frames, fps):
+    
+    os.makedirs('videos', exist_ok=True)
+    video_clip_path = f"videos/{uuid.uuid4()}.mp4"
+
+    # Get frame size
+    height, width, layers = frames[0].shape
+    size = (width, height)
+
+    # Define the codec and create VideoWriter object
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # You can also try 'XVID', 'MJPG', etc.
+    out = cv2.VideoWriter(video_clip_path, fourcc, fps, size)
+
+    for frame in frames:
+        out.write(frame)
+
+    out.release()
+
+    return video_clip_path
+
 
 def encode_to_video(frames, fps):
     os.makedirs('videos', exist_ok=True)
@@ -115,7 +135,7 @@ def process_clip(prompt, frames, chatbot):
     
     if api_response["condition_met"] == True:
         finish_time = datetime.now(israel_tz).strftime('%H:%M:%S')
-        video_clip_path = encode_to_video(frames, fps)
+        video_clip_path = encode_to_video_fast(frames, fps)
         chatbot.append(((video_clip_path,), None))
         result = f"Time: {start_time}\n"
         chatbot.append((result, None))
@@ -142,7 +162,7 @@ def process_clip_from_file(prompt, frames, chatbot, fps):
         
         result = None
         if api_response and api_response.get("condition_met", False):
-            video_clip_path = encode_to_video(frames, fps)
+            video_clip_path = encode_to_video_fast(frames, fps)
             chatbot.append(((video_clip_path,), None))
             chatbot.append((f"Time: {start_time}\nDetails: {api_response.get('details', '')}", None))