marker-io / marker /output.py
Ritvik19's picture
Add all files and directories
c8a32e7
raw
history blame
No virus
1.27 kB
import os
import json
def get_subfolder_path(out_folder, fname):
subfolder_name = fname.split(".")[0]
subfolder_path = os.path.join(out_folder, subfolder_name)
return subfolder_path
def get_markdown_filepath(out_folder, fname):
subfolder_path = get_subfolder_path(out_folder, fname)
out_filename = fname.rsplit(".", 1)[0] + ".md"
out_filename = os.path.join(subfolder_path, out_filename)
return out_filename
def markdown_exists(out_folder, fname):
out_filename = get_markdown_filepath(out_folder, fname)
return os.path.exists(out_filename)
def save_markdown(out_folder, fname, full_text, images, out_metadata):
subfolder_path = get_subfolder_path(out_folder, fname)
os.makedirs(subfolder_path, exist_ok=True)
markdown_filepath = get_markdown_filepath(out_folder, fname)
out_meta_filepath = markdown_filepath.rsplit(".", 1)[0] + "_meta.json"
with open(markdown_filepath, "w+", encoding='utf-8') as f:
f.write(full_text)
with open(out_meta_filepath, "w+") as f:
f.write(json.dumps(out_metadata, indent=4))
for filename, image in images.items():
image_filepath = os.path.join(subfolder_path, filename)
image.save(image_filepath, "PNG")
return subfolder_path