Image Gallery Viewer

#!/usr/bin/env python3 """ Simple HTTP server for image gallery Usage: python server.py [port] [directory] Default: python server.py 8000 ./images """ import http.server import socketserver import json import os import sys import subprocess import shutil from pathlib import Path from urllib.parse import unquote, urlparse import mimetypes PORT = int(sys.argv[1]) if len(sys.argv) > 1 else 1338 IMAGES_DIR = sys.argv[2] if len(sys.argv) > 2 else './' class GalleryHandler(http.server.SimpleHTTPRequestHandler): def do_GET(self): parsed_path = urlparse(self.path) path = unquote(parsed_path.path) # Serve the main HTML page if path == '/' or path == '/index.html': self.serve_html() # API: Get list of directories elif path == '/api/directories': self.serve_directories() # API: Get list of images in a directory elif path.startswith('/api/images/'): directory = path.replace('/api/images/', '') self.serve_images_list(directory) # API: Get captions for a directory elif path.startswith('/api/captions/'): directory = path.replace('/api/captions/', '') self.serve_captions(directory) # API: Serve thumbnail (images or generated video thumbnail) elif path.startswith('/api/thumbnail/'): parts = path.replace('/api/thumbnail/', '').split('/', 1) if len(parts) == 2: directory, filename = parts self.serve_thumbnail(directory, filename) # API: Serve full image or video elif path.startswith('/api/image/'): parts = path.replace('/api/image/', '').split('/', 1) if len(parts) == 2: directory, filename = parts self.serve_image(directory, filename) # API: Download file (for videos or images) elif path.startswith('/api/download/'): parts = path.replace('/api/download/', '').split('/', 1) if len(parts) == 2: directory, filename = parts self.serve_download(directory, filename) else: self.send_error(404, "File not found") def serve_html(self): """Serve the embedded HTML page""" html_content = """ Image Gallery Viewer

Select a directory and image to begin

""" self.send_response(200) self.send_header('Content-type', 'text/html; charset=utf-8') self.end_headers() self.wfile.write(html_content.encode('utf-8')) def serve_directories(self): """Return list of subdirectories in IMAGES_DIR""" try: images_path = Path(IMAGES_DIR) if not images_path.exists(): directories = [] else: directories = [d.name for d in images_path.iterdir() if d.is_dir()] self.send_json_response(directories) except Exception as e: self.send_error(500, "Error reading directories: {}".format(str(e))) def serve_images_list(self, directory): """Return list of JPG files in the specified directory""" try: images_base = Path(IMAGES_DIR).resolve() dir_path = (images_base / directory).resolve() # Security check - ensure path is within IMAGES_DIR if not str(dir_path).startswith(str(images_base)): self.send_error(403, "Access denied") return if not dir_path.exists(): self.send_json_response([]) return # Include JPEG images and MP4 movies images = [f.name for f in dir_path.iterdir() if f.is_file() and f.suffix.lower() in ['.jpg', '.jpeg', '.mp4']] images.sort() self.send_json_response(images) except Exception as e: self.send_error(500, "Error reading images: {}".format(str(e))) def serve_captions(self, directory): """Return captions from image_captions.json if it exists""" try: images_base = Path(IMAGES_DIR).resolve() dir_path = (images_base / directory).resolve() # Security check - ensure path is within IMAGES_DIR if not str(dir_path).startswith(str(images_base)): self.send_error(403, "Access denied") return caption_file = dir_path / 'image_captions.json' if caption_file.exists(): with open(str(caption_file), 'r', encoding='utf-8') as f: captions = json.load(f) self.send_json_response(captions) else: self.send_json_response({}) except json.JSONDecodeError: self.send_json_response({}) except Exception as e: self.send_error(500, "Error reading captions: {}".format(str(e))) def serve_image(self, directory, filename): """Serve an image file""" try: # Construct the full path images_base = Path(IMAGES_DIR).resolve() image_path = (images_base / directory / filename).resolve() # Security check - ensure path is within IMAGES_DIR if not str(image_path).startswith(str(images_base)): self.send_error(403, "Access denied") return if not image_path.exists() or not image_path.is_file(): self.send_error(404, "Image not found: {}".format(image_path)) return # Determine content type content_type, _ = mimetypes.guess_type(str(image_path)) if not content_type: content_type = 'application/octet-stream' # Send the file (streamed in chunks) self.send_response(200) self.send_header('Content-type', content_type) self.send_header('Content-Length', image_path.stat().st_size) self.end_headers() with open(str(image_path), 'rb') as f: shutil.copyfileobj(f, self.wfile) except Exception as e: import traceback print("Error serving image: {}".format(traceback.format_exc())) self.send_error(500, "Error serving image: {}".format(str(e))) def send_json_response(self, data): """Send JSON response""" json_data = json.dumps(data).encode() self.send_response(200) self.send_header('Content-type', 'application/json') self.send_header('Content-Length', len(json_data)) self.end_headers() self.wfile.write(json_data) def serve_thumbnail(self, directory, filename): """Serve a thumbnail for an image or generate one for a video (mp4)""" try: images_base = Path(IMAGES_DIR).resolve() dir_path = (images_base / directory).resolve() # Security check if not str(dir_path).startswith(str(images_base)): self.send_error(403, "Access denied") return file_path = (dir_path / filename).resolve() if not file_path.exists() or not file_path.is_file(): self.send_error(404, "File not found") return # If it's an image, just serve it (browser will scale) if file_path.suffix.lower() in ['.jpg', '.jpeg', '.png']: # reuse serve_image return self.serve_image(directory, filename) # If it's a video, try to generate a thumbnail using ffmpeg if file_path.suffix.lower() == '.mp4': thumbs_base = images_base / '.thumbnails' / directory thumbs_base.mkdir(parents=True, exist_ok=True) thumb_name = filename + '.jpg' thumb_path = (thumbs_base / thumb_name).resolve() # regenerate if missing or older than source try: if not thumb_path.exists() or thumb_path.stat().st_mtime < file_path.stat().st_mtime: # run ffmpeg to grab frame at 1 second cmd = [ 'ffmpeg', '-y', '-ss', '00:00:01', '-i', str(file_path), '-vframes', '1', '-q:v', '2', str(thumb_path) ] subprocess.run(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, check=False) except Exception: # if ffmpeg fails, continue and attempt to serve if thumb exists pass if thumb_path.exists(): # serve thumbnail image content_type = 'image/jpeg' self.send_response(200) self.send_header('Content-type', content_type) self.send_header('Content-Length', thumb_path.stat().st_size) self.end_headers() with open(str(thumb_path), 'rb') as f: shutil.copyfileobj(f, self.wfile) return # fallback: serve a simple SVG placeholder so UI shows something placeholder_svg = '''''' svg_bytes = placeholder_svg.encode('utf-8') self.send_response(200) self.send_header('Content-type', 'image/svg+xml') self.send_header('Content-Length', str(len(svg_bytes))) self.end_headers() self.wfile.write(svg_bytes) return # Other types: not supported self.send_error(415, 'Unsupported media type') except Exception as e: import traceback print("Error serving thumbnail: {}".format(traceback.format_exc())) self.send_error(500, "Error serving thumbnail: {}".format(str(e))) def serve_download(self, directory, filename): """Serve a file for download with Content-Disposition""" try: images_base = Path(IMAGES_DIR).resolve() file_path = (images_base / directory / filename).resolve() if not str(file_path).startswith(str(images_base)): self.send_error(403, "Access denied") return if not file_path.exists() or not file_path.is_file(): self.send_error(404, "File not found") return content_type, _ = mimetypes.guess_type(str(file_path)) if not content_type: content_type = 'application/octet-stream' self.send_response(200) self.send_header('Content-type', content_type) self.send_header('Content-Length', file_path.stat().st_size) self.send_header('Content-Disposition', 'attachment; filename="{}"'.format(file_path.name)) self.end_headers() with open(str(file_path), 'rb') as f: shutil.copyfileobj(f, self.wfile) except Exception as e: import traceback print("Error serving download: {}".format(traceback.format_exc())) self.send_error(500, "Error serving download: {}".format(str(e))) def log_message(self, format, *args): """Override to customize logging""" print("{} - {}".format(self.address_string(), format % args)) if __name__ == "__main__": # Create images directory if it doesn't exist Path(IMAGES_DIR).mkdir(exist_ok=True) print("Starting Image Gallery Server") print("Port: {}".format(PORT)) print("Images directory: {}".format(Path(IMAGES_DIR).resolve())) print("Open your browser to: http://localhost:{}".format(PORT)) print("Press Ctrl+C to stop the server\n") httpd = socketserver.TCPServer(("", PORT), GalleryHandler) try: httpd.serve_forever() except KeyboardInterrupt: print("\nServer stopped.") finally: httpd.server_close()