utils

utils.py

@@ -0,0 +1,54 @@
+import base64
+import cv2
+from typing import List
+from io import BytesIO
+import numpy as np
+from PIL import Image
+
+def encode_frame_to_base64(frame: cv2.Mat) -> str:
+    """
+    Convert a cv2.Mat frame to a base64-encoded string.
+    """
+    _, buffer = cv2.imencode('.jpg', frame)
+    encoded_string = base64.b64encode(buffer).decode('utf-8')
+    return encoded_string
+
+def decode_base64_to_frame(encoded_string: str) -> cv2.Mat:
+    """
+    Convert a base64-encoded string back to a cv2.Mat frame.
+    """
+    decoded_data = base64.b64decode(encoded_string)
+    np_array = np.frombuffer(decoded_data, dtype=np.uint8)
+    return cv2.imdecode(np_array, cv2.IMREAD_COLOR)
+
+def get_bytes(image_path: str) -> bytes:
+    with open(image_path, "rb") as f:
+        return f.read()
+
+def get_text_file_contents(file_path):
+    """
+    Reads a text file from the specified path and returns its content as a string.
+
+    Args:
+        file_path (str): The path to the text file.
+
+    Returns:
+        str: The content of the file, or None if an error occurred.
+    """
+    try:
+        with open(file_path, 'r', encoding='utf-8') as file:
+            content = file.read()
+        return content
+    except FileNotFoundError:
+        print(f"Error: File not found at path: {file_path}")
+        return None
+    except Exception as e:
+        print(f"An error occurred while reading the file: {e}")
+        return None
+    
+def clean_text(text: str) -> str:
+    return text.encode("ascii", errors="ignore").decode()
+
+def get_base64(path: str) -> str:
+    with open(path, "rb") as f:
+        return base64.b64encode(f.read()).decode("utf-8")