Update app.py

app.py

@@ -10,7 +10,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
 from datetime import datetime
 import json
 import os
-# from moviepy.editor import ImageSequenceClip
+from moviepy import ImageSequenceClip
 from gradio_client import Client, file
 import subprocess
 import ffmpeg
@@ -89,19 +89,19 @@ def encode_to_video_fast(frames, fps):
     return video_clip_path
 
 
-# def encode_to_video(frames, fps):
-#     os.makedirs('videos', exist_ok=True)
-#     video_clip_path = f"videos/{uuid.uuid4()}.mp4"
+def encode_to_video(frames, fps):
+    os.makedirs('videos', exist_ok=True)
+    video_clip_path = f"videos/{uuid.uuid4()}.mp4"
     
-#     # Create a video clip from the frames using moviepy
-#     clip = ImageSequenceClip([frame[:, :, ::-1] for frame in frames], fps=fps)  # Convert from BGR to RGB
-#     clip.write_videofile(video_clip_path, codec="libx264")
+    # Create a video clip from the frames using moviepy
+    clip = ImageSequenceClip([frame[:, :, ::-1] for frame in frames], fps=fps)  # Convert from BGR to RGB
+    clip.write_videofile(video_clip_path, codec="libx264")
     
-#     # Convert the video file to base64
-#     with open(video_clip_path, "rb") as video_file:
-#         video_data = base64.b64encode(video_file.read()).decode('utf-8')
+    # Convert the video file to base64
+    with open(video_clip_path, "rb") as video_file:
+        video_data = base64.b64encode(video_file.read()).decode('utf-8')
     
-#     return video_clip_path
+    return video_clip_path
 
 # Function to process video frames using GPT-4 API
 def process_frames(frames, frames_to_skip = 1):
@@ -167,7 +167,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_fast(frames, fps)
+        video_clip_path = encode_to_video(frames, fps)
         chatbot.append(((video_clip_path,), None))
         chatbot.append((f"Time: {start_time}\nDetails: {api_response.get('details', '')}", None))
     
@@ -226,7 +226,7 @@ def process_clip_from_file_sync(prompt, frames, fps, video_path, id):
 def analyze_stream(prompt, chatbot):
     global stop_capture
     stop_capture = False
-    stream = "https://streamapi2.eu.loclx.io/stream/channel_102"
+    stream = "rtsp://FRefgrWt:Fgh342!23$@!h@93.170.28.66:554/Streaming/channels/101"
 
 
     cap = cv2.VideoCapture(stream or WEBCAM)