Update app.py

app.py

@@ -134,7 +134,7 @@ def check_condition(prompt, base64Frames):
     
 
 # Function to process video clip and update the chatbot
-def process_clip(prompt, frames, chatbot):
+def process_clip(prompt, frames, chatbot, id):
     # Print current time in Israel
     israel_tz = pytz.timezone('Asia/Jerusalem')
     start_time = datetime.now(israel_tz).strftime('%H:%M:%S')
@@ -147,9 +147,10 @@ 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_fast(frames, fps)
+        # video_clip_path = encode_to_video_fast(frames, fps)
+        video_clip_path = clip_video_segment_2("https://uqnmqpvwlbpmdvutucia.supabase.co/storage/v1/object/public/videos_test/junction.mp4?t=2024-12-26T09%3A58%3A11.931Z", id*LENGTH, LENGTH)
         chatbot.append(((video_clip_path,), None))
-        chatbot.append((f"Time: {start_time}\nDetails: {api_response.get('details', '')}", None))
+        chatbot.append((f"ID: {id}. Time: {start_time}\nDetails: {api_response.get('details', '')}", None))
     
         frame_paths = []
         for i, base64_frame in enumerate(base64Frames):
@@ -176,6 +177,7 @@ def process_clip_from_file(prompt, frames, chatbot, fps, video_path, id):
             video_clip_path = clip_video_segment_2(video_path, id*LENGTH, LENGTH)
             chatbot.append(((video_clip_path,), None))
             chatbot.append((f"Event ID: {id+1}\nDetails: {api_response.get('details', '')}", None))
+            yield chatbot
     
     return chatbot
 
@@ -191,6 +193,7 @@ def analyze_stream(prompt, chatbot):
 
     frames = []
     start_time = time.time()
+    id = 0
     while not stop_capture:
         ret, frame = cap.read()
         if not ret:
@@ -200,9 +203,10 @@ def analyze_stream(prompt, chatbot):
         # Sample the frames every 5 seconds
         if time.time() - start_time >= LENGTH:
             # Start a new thread for processing the video clip
-            Thread(target=process_clip, args=(prompt, frames.copy(), chatbot,)).start()
+            Thread(target=process_clip, args=(prompt, frames.copy(), chatbot, id)).start()
             frames = []
             start_time = time.time()
+        id+=1
         yield chatbot
 
     cap.release()