Gallery front

src/py/webui/upload.py

@@ -0,0 +1,50 @@
+import logging
+from pathlib import Path
+from flask import Blueprint, request, current_app
+from werkzeug.utils import secure_filename
+
+# Create a Flask blueprint for upload routes
+upload_bp = Blueprint("upload", __name__)
+
+# Allowed file extensions for uploads
+ALLOWED_EXTENSIONS = {"png", "jpg", "jpeg", "webp"}
+
+# Function to check if a file has an allowed extension
+def allowed_file(filename: str) -> bool:
+    return "." in filename and filename.rsplit(".", 1)[1].lower() in ALLOWED_EXTENSIONS
+
+# Function to save uploaded file to a given folder
+def save_uploaded_file(file, folder: Path):
+    folder.mkdir(parents=True, exist_ok=True)  # Create folder if it doesn't exist
+    filename = secure_filename(file.filename)  # Sanitize filename
+    file.save(folder / filename)  # Save file to folder
+    logging.info(f"[✓] Uploaded {filename} to {folder}")
+    return filename  # Return saved filename
+
+# Route to handle photo uploads for gallery or hero
+@upload_bp.route("/api/<section>/upload", methods=["POST"])
+def upload_photo(section: str):
+    # Validate section
+    if section not in ["gallery", "hero"]:
+        return {"error": "Invalid section"}, 400
+
+    # Check if the request contains a file
+    if "file" not in request.files:
+        return {"error": "No file part"}, 400
+    file = request.files["file"]
+
+    # Check if a file was actually selected
+    if file.filename == "":
+        return {"error": "No selected file"}, 400
+
+    # Check file type and save it
+    if file and allowed_file(file.filename):
+        PHOTOS_DIR = current_app.config.get("PHOTOS_DIR")  # Get base photos directory from config
+        if not PHOTOS_DIR:
+            return {"error": "Server misconfiguration"}, 500
+        folder = PHOTOS_DIR / section  # Target folder (gallery or hero)
+        filename = save_uploaded_file(file, folder)
+        return {"status": "ok", "filename": filename}
+
+    # If file type is not allowed
+    return {"error": "File type not allowed"}, 400

src/py/webui/webui.py

@@ -0,0 +1,112 @@
+import logging
+from pathlib import Path
+from flask import Flask, jsonify, request, send_from_directory, render_template
+from src.py.builder.gallery_builder import (
+    GALLERY_YAML, GALLERY_DIR, HERO_DIR,
+    load_yaml, save_yaml, get_all_image_paths, update_gallery, update_hero
+)
+from src.py.webui.upload import upload_bp
+
+# --- Logging configuration ---
+# Logs messages to console with INFO level
+logging.basicConfig(level=logging.INFO, format="%(message)s")
+
+# --- Flask app setup ---
+# WEBUI_PATH points to the web UI folder (templates + static)
+WEBUI_PATH = Path(__file__).parents[2] / "webui"
+app = Flask(
+    __name__,
+    template_folder=WEBUI_PATH,  # where Flask looks for templates
+    static_folder=WEBUI_PATH,    # where Flask serves static files
+    static_url_path=""            # URL path prefix for static files
+)
+
+# --- Absolute photos directory ---
+# Used by upload.py and deletion endpoints
+PHOTOS_DIR = Path(__file__).resolve().parents[3] / "config" / "photos"
+app.config["PHOTOS_DIR"] = PHOTOS_DIR
+
+# --- Register upload blueprint ---
+# Handles /api/<section>/upload endpoints for gallery and hero images
+app.register_blueprint(upload_bp)
+
+# --- Existing API routes ---
+
+# Serve main page
+@app.route("/")
+def index():
+    return render_template("index.html")
+
+# Get gallery images (returns JSON array)
+@app.route("/api/gallery", methods=["GET"])
+def get_gallery():
+    data = load_yaml(GALLERY_YAML)
+    return jsonify(data.get("gallery", {}).get("images", []))
+
+# Get hero images (returns JSON array)
+@app.route("/api/hero", methods=["GET"])
+def get_hero():
+    data = load_yaml(GALLERY_YAML)
+    return jsonify(data.get("hero", {}).get("images", []))
+
+# Update gallery images with new JSON data
+@app.route("/api/gallery/update", methods=["POST"])
+def update_gallery_api():
+    images = request.json
+    data = load_yaml(GALLERY_YAML)
+    data["gallery"]["images"] = images
+    save_yaml(data, GALLERY_YAML)
+    return jsonify({"status": "ok"})
+
+# Update hero images with new JSON data
+@app.route("/api/hero/update", methods=["POST"])
+def update_hero_api():
+    images = request.json
+    data = load_yaml(GALLERY_YAML)
+    data["hero"]["images"] = images
+    save_yaml(data, GALLERY_YAML)
+    return jsonify({"status": "ok"})
+
+# Refresh gallery from the folder (rebuild YAML)
+@app.route("/api/gallery/refresh", methods=["POST"])
+def refresh_gallery():
+    update_gallery()
+    return jsonify({"status": "ok"})
+
+# Refresh hero images from the folder
+@app.route("/api/hero/refresh", methods=["POST"])
+def refresh_hero():
+    update_hero()
+    return jsonify({"status": "ok"})
+
+# Delete a gallery image file
+@app.route("/api/gallery/delete", methods=["POST"])
+def delete_gallery_photo():
+    data = request.json
+    src = data.get("src")  # filename only
+    file_path = PHOTOS_DIR / "gallery" / src
+    if file_path.exists():
+        file_path.unlink()  # remove the file
+        return {"status": "ok"}
+    return {"error": "File not found"}, 404
+
+# Delete a hero image file
+@app.route("/api/hero/delete", methods=["POST"])
+def delete_hero_photo():
+    data = request.json
+    src = data.get("src")  # filename only
+    file_path = PHOTOS_DIR / "hero" / src
+    if file_path.exists():
+        file_path.unlink()  # remove the file
+        return {"status": "ok"}
+    return {"error": "File not found"}, 404
+
+# Serve photos from /photos/<section>/<filename>
+@app.route("/photos/<section>/<path:filename>")
+def photos(section, filename):
+    return send_from_directory(PHOTOS_DIR / section, filename)
+
+# --- Main entry point ---
+if __name__ == "__main__":
+    logging.info("Starting WebUI at http://127.0.0.1:5000")
+    app.run(debug=True)