import os
import argparse
def main():
# Set up argument parsing
parser = argparse.ArgumentParser(description="Generate an HTML gallery from images in a folder.")
parser.add_argument("image_folder", help="Path to the folder containing images")
parser.add_argument("-o", "--output", default="gallery.html", help="Output HTML file (default: gallery.html)")
args = parser.parse_args()
image_folder = args.image_folder
output_html = args.output
# Start HTML
html_content = """
Image Gallery
"""
# Iterate through the images and create thumbnails
for filename in os.listdir(image_folder):
if filename.lower().endswith(('jpg', 'jpeg', 'png', 'gif')):
full_image_path = os.path.join(image_folder, filename)
thumbnail_url = filename # relative link
html_content += f"""
"""
# End HTML
html_content += """
"""
# Write the HTML to a file
with open(output_html, "w", encoding="utf-8") as file:
file.write(html_content)
print(f"✅ Gallery HTML file has been created: {output_html}")
if __name__ == "__main__":
main()