Files
OmniParser/computer_use_demo/gradio/agent/llm_utils/utils.py
Thomas Dhome Casanova (from Dev Box) e268184f8d init r1
2025-01-31 18:37:14 -08:00

13 lines
396 B
Python

import base64
def is_image_path(text):
image_extensions = (".jpg", ".jpeg", ".png", ".gif", ".bmp", ".tiff", ".tif")
if text.endswith(image_extensions):
return True
else:
return False
def encode_image(image_path):
"""Encode image file to base64."""
with open(image_path, "rb") as image_file:
return base64.b64encode(image_file.read()).decode("utf-8")