Update app.py

app.py

@@ -13,6 +13,7 @@ import os
 from moviepy.editor import ImageSequenceClip
 from gradio_client import Client, file
 import subprocess
+import ffmpeg
 
 api_key = os.getenv("OPEN_AI_KEY")
 user_name = os.getenv("USER_NAME")
@@ -38,6 +39,24 @@ client = openai.OpenAI(api_key=api_key)
 stop_capture = False
 alerts_mode = True
 
+def clip_video_segment_2(input_video_path, start_time, duration):
+    os.makedirs('videos', exist_ok=True)
+    output_video_path = f"videos/{uuid.uuid4()}.mp4"
+    
+    # Use ffmpeg-python to clip the video
+    try:
+        (
+            ffmpeg
+            .input(input_video_path, ss=start_time)  # Seek to start_time
+            .output(output_video_path, t=duration, c='copy')  # Set the duration
+            .run(overwrite_output=True)
+        )
+        print('input_video_path', input_video_path, output_video_path)
+        return output_video_path
+    except ffmpeg.Error as e:
+        print(f"Error clipping video: {e}")
+        return None
+
 def clip_video_segment(input_video_path, start_time, duration):
     os.makedirs('videos', exist_ok=True)
     output_video_path = f"videos/{uuid.uuid4()}.mp4"
@@ -171,7 +190,7 @@ def process_clip_from_file(prompt, frames, chatbot, fps, video_path, id):
         result = None
         if api_response and api_response.get("condition_met", False):
             # video_clip_path = encode_to_video_fast(frames, fps)
-            video_clip_path = clip_video_segment(video_path, id*LENGTH, LENGTH)
+            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))