Update share_btn.py

share_btn.py

@@ -22,21 +22,12 @@ share_js = """async () => {
             return url;
         }
 
-        function find_els(parent, selectors, min_size){
-              var elAry = Array.prototype.filter.call(parent.querySelectorAll(selectors),function(el){
-                return (el.offsetWidth>min_size && el.offsetHeight>min_size);
-            });
-            return elAry;
-        }
-
         var gradioEl = document.querySelector('body > gradio-app').shadowRoot;
         if (!gradioEl) {
             gradioEl = document.querySelector('body > gradio-app');
         }
         
-        // const imgEls = gradioEl.querySelectorAll('#gallery img');
         const imgEls = find_els(gradioEl, '#gallery .overflow-hidden', 50);
-        console.log(imgEls);
         const promptTxt = 'my perfect work'; 
 
         const shareBtnEl = gradioEl.querySelector('#share-btn');
@@ -49,17 +40,21 @@ share_js = """async () => {
         shareIconEl.style.display = 'none';
         loadingIconEl.style.removeProperty('display');
 
-        const files = await Promise.all(
+        var files = [];
+        await Promise.all(
             [...imgEls].map(async (imgEl) => {
-                const res = await fetch(imgEl.src);
-                const blob = await res.blob();
-                const imgId = Date.now() % 200;
-                const fileName = `img-cleaner-${{imgId}}.jpg`;
-                return new File([blob], fileName, { type: 'image/jpeg' });
+                if (imgEl.offsetWidth > 50 && imgEl.offsetHeight > 50) {
+                    const res = await fetch(imgEl.src);
+                    const blob = await res.blob();
+                    const imgId = Date.now() % 200;
+                    const fileName = `img-cleaner-${{imgId}}.jpg`;
+                    files.push(new File([blob], fileName, { type: 'image/jpeg'}));
+                }
             })
         );
+
         const urls = await Promise.all(files.map((f) => uploadFile(f)));
-        const htmlImgs = urls.map(url => `<img src='${url}' width='400' height1='400'>`);
+        const htmlImgs = urls.map(url => `<img src='${url}' width='400' height='400'>`);
         const descriptionMd = `<div style='display: flex; flex-wrap: wrap; column-gap: 0.75rem;'>
                 ${htmlImgs.join(`\n`)}
                 </div>`;