upload flask server

flask_sku_server.py

@@ -0,0 +1,119 @@
+from flask import Flask, jsonify, request
+import json
+import os
+import random
+from collections import OrderedDict
+
+app = Flask(__name__)
+JSON_PATH = r"C:\Users\ASUS\Desktop\Computer_vision_binder\output_cleaning_imagesCarrefourksa_fields.json"
+
+
+def load_products():
+    if os.path.exists(JSON_PATH):
+        with open(JSON_PATH, 'r', encoding='utf-8') as file:
+            return json.load(file)
+    return []
+
+
+def save_products(products_list):
+    cleaned = []
+    for p in products_list:
+        ordered = OrderedDict()
+        # Desired field order
+        field_order = [
+            "product_id",
+            "image_id",
+            "sor_cat",
+            "sor_subcat",
+            "sor_subsubcat",
+            "sor_prdocut_name",
+            "sor_product_url",
+            "sor_image_url"
+        ]
+        for key in field_order:
+            if key in p:
+                ordered[key] = p[key]
+        # SKU always last
+        ordered["binder_cv_sku"] = p.get("binder_cv_sku", None)
+        cleaned.append(ordered)
+
+    with open(JSON_PATH, 'w', encoding='utf-8') as file:
+        json.dump(cleaned, file, indent=2, ensure_ascii=False)
+
+
+def generate_unique_sku(existing_skus):
+    while True:
+        sku = f"{random.randint(0, 999999999):09d}"
+        if sku not in existing_skus:
+            return sku
+
+
+@app.route('/all-products', methods=['GET'])
+def get_all_products():
+    products = load_products()
+
+    sku_map = {}
+    existing_skus = set()
+    updated = False
+
+    for p in products:
+        pid = p.get("product_id")
+        sku = p.get("binder_cv_sku")
+        if sku:
+            sku_map[pid] = sku
+            existing_skus.add(sku)
+
+    for p in products:
+        pid = p.get("product_id")
+        if not p.get("binder_cv_sku"):
+            if pid in sku_map:
+                p["binder_cv_sku"] = sku_map[pid]
+            else:
+                new_sku = generate_unique_sku(existing_skus)
+                sku_map[pid] = new_sku
+                p["binder_cv_sku"] = new_sku
+                existing_skus.add(new_sku)
+                updated = True
+
+    if updated:
+        save_products(products)
+
+    return jsonify(products)
+
+
+@app.route('/add-product', methods=['POST'])
+def add_product():
+    new_product = request.get_json()
+    if not new_product or "product_id" not in new_product or "image_id" not in new_product:
+        return jsonify({"error": "Missing product_id or image_id"}), 400
+
+    products = load_products()
+
+    # Check if image_id already exists — avoid duplicates
+    existing_image_ids = {p['image_id'] for p in products}
+    if new_product["image_id"] in existing_image_ids:
+        return jsonify({"message": "Image already exists. Skipped."}), 200
+
+    # Map product_id to existing SKUs
+    existing_skus = {p['binder_cv_sku'] for p in products if "binder_cv_sku" in p}
+    sku_map = {p['product_id']: p['binder_cv_sku'] for p in products if "binder_cv_sku" in p}
+
+    pid = new_product["product_id"]
+
+    # Assign SKU: reuse if product_id exists, else generate new SKU
+    if pid in sku_map:
+        new_product["binder_cv_sku"] = sku_map[pid]
+    else:
+        new_sku = generate_unique_sku(existing_skus)
+        new_product["binder_cv_sku"] = new_sku
+        sku_map[pid] = new_sku
+
+    # Append and save
+    products.append(new_product)
+    save_products(products)
+
+    return jsonify({"message": "Product added successfully", "binder_cv_sku": new_product["binder_cv_sku"]})
+
+
+if __name__ == '__main__':
+    app.run(debug=True)