Update generation.html

generation.html

@@ -369,6 +369,19 @@
           alert("Please generate an image first.");
           return;
         }
+        // 4. Clear old results
+        let resultContainer = document.getElementById("lookalike-results");
+        if (!resultContainer) {
+          resultContainer = document.createElement("div");
+          resultContainer.id = "lookalike-results";
+          resultContainer.className = "mt-6 grid grid-cols-2 sm:grid-cols-3 md:grid-cols-4 gap-4";
+          document.body.appendChild(resultContainer);
+        }
+        resultContainer.innerHTML = `
+          <div class="col-span-full flex justify-center items-center">
+            <img src="https://i.gifer.com/YCZH.gif" alt="loading" class="w-20 h-20"/>
+          </div>
+        `;
 
         // 1. Convert current image to base64
         const imageBlob = await fetch(currentImageUrl).then(r => r.blob());
@@ -402,16 +415,6 @@
         if (!searchResp.ok) throw new Error("Search failed");
         const result = await searchResp.json();
 
-        // 4. Clear old results
-        let resultContainer = document.getElementById("lookalike-results");
-        if (!resultContainer) {
-          resultContainer = document.createElement("div");
-          resultContainer.id = "lookalike-results";
-          resultContainer.className = "flex flex-wrap mt-6";
-          document.body.appendChild(resultContainer);
-        }
-        resultContainer.innerHTML = "";
-
         // 5. Process results
         for (const r of result.results) {
           try {
@@ -432,7 +435,12 @@
               body: formData
             });
 
-            let finalBlob = cleanResp.ok ? await cleanResp.blob() : blob;
+            if (!cleanResp.ok) {
+              console.warn("Skipping image, dewatermark failed:", r);
+              continue;
+            }
+            
+            let finalBlob = await cleanResp.blob();
             let imgUrl = URL.createObjectURL(finalBlob);
 
             // Display