generate_gallery.py

import os

template = """
        <div class="col-md-6 captioned_videos" id="res_{vid_id}" style="margin-top: 0px; margin-bottom: 0px ;">
            <video class="video lazy" loop playsinline autoplay muted style="width: 100%">
                <source data-src="./static/DMTet-Dreamfusion/{keyword}.mp4" type="video/mp4"></source>
            </video>
            <p class="compare-caption">{description}</p>
        </div>"""


video_tokens = []
keys  = []
with open("./static/dmtet-dreamfusion.txt", "r") as f:
    for line in f:
        video_names = line.strip()
        video_tokens.append(video_names)
        keys.append(os.path.splitext(os.path.basename(video_names))[0])




with open("gallery_template.html", "r") as f:
    html_template = f.read()

content = ""
page_idx = 0
prev_i = 0
page_size = 10

for i,k in enumerate(keys):
    description = keys[i]
    current_video = template.format(vid_id=i, keyword=k, description=description)
    content += current_video

    
    first_page = page_idx - 1 < 0
    last_page = i + 1 >= len(keys)
    if i - prev_i + 1 >= page_size or last_page:
        link1 = f"dmtet_dreamfusion_gallery_{page_idx - 1}.html"
        link2 = f"dmtet_dreamfusion_gallery_{page_idx + 1}.html"
        active1 = "disabled" if first_page else ""
        active2 = "disabled" if last_page else ""
        html = html_template.replace("{{{content}}}", content).replace("{{{link1}}}", link1).replace("{{{link2}}}", link2)
        html = html.replace("{{{active1}}}", active1).replace("{{{active2}}}", active2)
        html = html.replace("{{{start_idx}}}", str(prev_i+1)).replace("{{{end_idx}}}", str(i+1)).replace("{{{total}}}", str(len(keys)))
        with open(f"dmtet_dreamfusion_gallery_{page_idx}.html", "w") as f:
            f.write(html)
        content = ""
        page_idx += 1
        prev_i = i + 1