Update share_btn.py

share_btn.py

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