Hugging Face's logo
Hugging Face
Search models, datasets, users...
Models
Datasets
Spaces
Docs
Solutions
Pricing
Spaces:
stabilityai
/
stable-diffusion Copied
like
2.77k
App
Files and versions
Community
2785
stable-diffusion
/
share_btn.py
multimodalart's picture
multimodalart
HF STAFF
Fix share button (#826)
517a59b
13 days ago
raw
history
blame
contribute
delete
Safe
6.13 kB
community_icon_html = """"""
loading_icon_html = """"""
share_js = """async () => {
async function uploadFile(file){
const UPLOAD_URL = 'https://huggingface.co/uploads';
const response = await fetch(UPLOAD_URL, {
method: 'POST',
headers: {
'Content-Type': file.type,
'X-Requested-With': 'XMLHttpRequest',
},
body: file, /// <- File inherits from Blob
});
const url = await response.text();
return url;
}
const gradioEl = document.querySelector('body > gradio-app');
const imgEls = gradioEl.querySelectorAll('#gallery img');
const promptTxt = gradioEl.querySelector('#prompt-text-input input').value;
const shareBtnEl = gradioEl.querySelector('#share-btn');
const shareIconEl = gradioEl.querySelector('#share-btn-share-icon');
const loadingIconEl = gradioEl.querySelector('#share-btn-loading-icon');
if(!imgEls.length){
return;
};
shareBtnEl.style.pointerEvents = 'none';
shareIconEl.style.display = 'none';
loadingIconEl.style.removeProperty('display');
const 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 = `diffuse-the-rest-${{imgId}}.jpg`;
return new File([blob], fileName, { type: 'image/jpeg' });
})
);
const urls = await Promise.all(files.map((f) => uploadFile(f)));
const htmlImgs = urls.map(url => `
`);
const descriptionMd = `
${htmlImgs.join(`\n`)}
`;
const params = new URLSearchParams({
title: promptTxt,
description: descriptionMd,
});
const paramsStr = params.toString();
window.open(`https://huggingface.co/spaces/stabilityai/stable-diffusion/discussions/new?${paramsStr}`, '_blank');
shareBtnEl.style.removeProperty('pointer-events');
shareIconEl.style.removeProperty('display');
loadingIconEl.style.display = 'none';
}"""