utils.py

import time
import cv2
import os
def record_response(question, response):
    with open("./interaction.txt", "a") as file:
        timestamp = time.strftime('%Y-%m-%d %H:%M:%S')
        file.write(f"Question: {question}\n")
        file.write(f"Response: {response}\n")
        file.write(f"Time: {timestamp}\n\n")

def save_video(frames, filename):
    if len(frames) == 0:
        print("No frames to save.")
        return
    frame_h,frame_w,_=frames[0].shape
    folder_path = "video_logs"
    paths=os.path.join(folder_path,filename)
    output=cv2.VideoWriter(paths, cv2.VideoWriter_fourcc(*'DIVX'), 30, (frame_w, frame_h))
    for frame in frames:
        output.write(frame)

    output.release()
    print(f"Video saved as {filename}")


def record_video(cap, stframe, duration=3):
    frames = []
    start_time = time.time()

    while time.time() - start_time < duration:
        ret, frame = cap.read()
        #I flipped it since, my camera is default flipped
        stframe.image(cv2.cvtColor(cv2.flip(frame, 1), cv2.COLOR_BGR2RGB))
        if ret:
            frames.append(frame)

    return frames