Update share_btn.py

share_btn.py

@@ -38,12 +38,9 @@ share_js = """async () => {
 	}
     
     const gradioEl = document.querySelector("gradio-app").shadowRoot || document.querySelector('body > gradio-app');
-    const randomLoRA_1 = gradioEl.querySelector("#component-7 img")
-    const randomLoRA_1_title = gradioEl.querySelector("#component-7 [data-testid="block-label"] span").innerHTML
-    const randomLoRA_2 = gradioEl.querySelector("#component-7 img")
-    const randomLoRA_2_title = gradioEl.querySelector("#component-7 [data-testid="block-label"] span").innerHTML
-    const prompt = gradioEl.querySelector("#prompt textarea").value
-    const outputImgEl = gradioEl.querySelector('#component-22 img');
+    const selectedLoRA = gradioEl.querySelector('#selected_lora').innerHTML;
+    const inputPrompt = gradioEl.querySelector('#prompt input').value;
+    const outputImgEl = gradioEl.querySelector('#result-image img');
     
     const shareBtnEl = gradioEl.querySelector('#share-btn');
     const shareIconEl = gradioEl.querySelector('#share-btn-share-icon');
@@ -54,34 +51,22 @@ share_js = """async () => {
     loadingIconEl.style.removeProperty('display');
     
     const inputFile = await getInputImgFile(outputImgEl);
-    const urlOutputImg = await uploadFile(inputFile);
-
-    const inputFile = await getInputImgFile(randomLoRA_1);
-    const urlLoRA1 = await uploadFile(inputFile);
-
-    const inputFile = await getInputImgFile(randomLoRA_2);
-    const urlLoRA2 = await uploadFile(inputFile);
+    const urlInputImg = await uploadFile(inputFile);
     
     const descriptionMd = `
-
-| {randomLoRA_1_title} |   | {randomLoRA_2_title}|
-|---------|-----------|---------|
-| ![{randomLoRA_1_title}]({urlLoRA1}) | + | ![{randomLoRA_1_title}]({urlLoRA2}) |
-
+${selectedLoRA}
 ### Prompt
 ${inputPrompt}
-
 #### Generated Image:
 <img src="${urlInputImg}" />
 `;
-    console.log(descriptionMd)
     const params = new URLSearchParams({
-        title: prompt,
+        title: inputPrompt,
         description: descriptionMd,
         preview: true
     });
 	const paramsStr = params.toString();
-	window.open(`https://huggingface.co/spaces/multimodalart/lora-roulette/discussions/new?${paramsStr}`, '_blank');
+	window.open(`https://huggingface.co/spaces/multimodalart/LoraTheExplorer/discussions/new?${paramsStr}`, '_blank');
     shareBtnEl.style.removeProperty('pointer-events');
     shareIconEl.style.removeProperty('display');
     loadingIconEl.style.display = 'none';