Update app.py

app.py

@@ -36,7 +36,6 @@ with open(f"{BLOCKS_DIR}/embeddings.json", "r") as f:
 image_paths = [item["file-path"] for item in embedding_json]
 image_embeds = np.array([item["embeddings"] for item in embedding_json])
 
-
 # ============================== #
 #   HELPER: Decode Base64 Image  #
 # ============================== #
@@ -45,16 +44,11 @@ def decode_base64_image(b64_string):
     img = Image.open(BytesIO(img_data)).convert("RGB")
     return img
 
-
 # ============================== #
 #         API ROUTE              #
 # ============================== #
 @app.route("/match", methods=["POST"])
 def match_image():
-    """
-    Input: JSON { "images": ["<base64_img1>", "<base64_img2>", ...] }
-    Output: Best match path + score for each input image
-    """
     data = request.get_json()
     if "images" not in data:
         return jsonify({"error": "No images provided"}), 400
@@ -62,16 +56,14 @@ def match_image():
     results = []
     for b64_img in data["images"]:
         try:
-            # Decode and embed input image
             img = decode_base64_image(b64_img)
-            query_embed = np.array(clip_embd.embed_image([img]))  # embed_image can take PIL images
+            query_embed = np.array(clip_embd.embed_image([img]))
 
-            # Cosine similarity with stored embeddings
             sims = cosine_similarity(query_embed, image_embeds)[0]
             best_idx = np.argmax(sims)
 
             results.append({
-                "input": b64_img[:50] + "...",  # partial preview
+                "input": b64_img[:50] + "...",
                 "best_match": {
                     "name": os.path.basename(image_paths[best_idx]),
                     "path": image_paths[best_idx],
@@ -84,8 +76,16 @@ def match_image():
     return jsonify(results)
 
 
+# ============================== #
+#       SIMPLE HTML UI           #
+# ============================== #
+@app.route("/", methods=["GET", "POST"])
+def index():
+    return render_template("index.html")
+
+
 # ============================== #
 #          MAIN ENTRY            #
 # ============================== #
 if __name__ == "__main__":
-    app.run(debug=True, port=7860)
+    app.run(debug=True, port=7860)