Reworked flow

src/py/webui/webui.py

@@ -2,111 +2,106 @@ 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
+    GALLERY_YAML, load_yaml, save_yaml, 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"
+WEBUI_PATH = Path(__file__).parents[2] / "webui"  # Path to static/templates
 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
+    template_folder=WEBUI_PATH,
+    static_folder=WEBUI_PATH,
+    static_url_path=""
 )
 
-# --- Absolute photos directory ---
-# Used by upload.py and deletion endpoints
+# --- Photos directory (configurable) ---
 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 ---
+# --- Routes ---
 
-# Serve main page
 @app.route("/")
 def index():
+    """Serve the main HTML page."""
     return render_template("index.html")
 
-# Get gallery images (returns JSON array)
 @app.route("/api/gallery", methods=["GET"])
 def get_gallery():
+    """Return JSON list of gallery images from YAML."""
     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():
+    """Return JSON list of hero images from YAML."""
     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():
+    """Update gallery images in YAML from frontend JSON."""
     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():
+    """Update hero images in YAML from frontend JSON."""
     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():
+    """Refresh gallery YAML from photos/gallery folder."""
     update_gallery()
     return jsonify({"status": "ok"})
 
-# Refresh hero images from the folder
 @app.route("/api/hero/refresh", methods=["POST"])
 def refresh_hero():
+    """Refresh hero YAML from photos/hero folder."""
     update_hero()
     return jsonify({"status": "ok"})
 
-# Delete a gallery image file
 @app.route("/api/gallery/delete", methods=["POST"])
 def delete_gallery_photo():
+    """Delete a gallery photo from disk and return status."""
     data = request.json
-    src = data.get("src")  # filename only
+    src = data.get("src")
     file_path = PHOTOS_DIR / "gallery" / src
     if file_path.exists():
-        file_path.unlink()  # remove the file
+        file_path.unlink()
         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():
+    """Delete a hero photo from disk and return status."""
     data = request.json
-    src = data.get("src")  # filename only
+    src = data.get("src")
     file_path = PHOTOS_DIR / "hero" / src
     if file_path.exists():
-        file_path.unlink()  # remove the file
+        file_path.unlink()
         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):
+    """Serve uploaded photos from disk."""
     return send_from_directory(PHOTOS_DIR / section, filename)
 
-# --- Main entry point ---
+# --- Run server ---
 if __name__ == "__main__":
     logging.info("Starting WebUI at http://127.0.0.1:5000")
     app.run(debug=True)