Spaces:
Paused
Paused
File size: 1,381 Bytes
73683aa d0abfe5 73683aa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
# utils.py
import os
import zipfile
import base64
import re
import logging
def compress_directory_to_zip(directory_path, output_zip_path):
"""
Compresses the specified directory into a ZIP file.
"""
try:
with zipfile.ZipFile(output_zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
for root, dirs, files in os.walk(directory_path):
for file in files:
file_path = os.path.join(root, file)
arcname = os.path.relpath(file_path, directory_path)
zipf.write(file_path, arcname)
return True
except Exception as e:
logging.exception("Error compressing directory: %s", e)
return False
def image_to_base64(image_path):
with open(image_path, "rb") as image_file:
return base64.b64encode(image_file.read()).decode('utf-8')
def replace_image_with_base64(markdown_text, image_dir_path):
"""
Finds markdown image tags and replaces file paths with embedded base64 data.
"""
pattern = r'\!\[(?:[^\]]*)\]\(([^)]+)\)'
def replace(match):
relative_path = match.group(1)
full_path = os.path.join(image_dir_path, relative_path)
base64_image = image_to_base64(full_path)
return f""
return re.sub(pattern, replace, markdown_text)
|