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;
}
async function getInputImgFile(imgEl){
const res = await fetch(imgEl.src);
const blob = await res.blob();
const imgId = Date.now() % 200;
const isPng = imgEl.src.startsWith(`data:image/png`);
if(isPng){
const fileName = `sd-perception-${{imgId}}.png`;
return new File([blob], fileName, { type: 'image/png' });
}else{
const fileName = `sd-perception-${{imgId}}.jpg`;
return new File([blob], fileName, { type: 'image/jpeg' });
}
}
async function getOutputMusicFile(audioEL){
const res = await fetch(audioEL.src);
const blob = await res.blob();
const audioId = Date.now() % 200;
const fileName = `riffusion-music-${{audioId}}.wav`;
const musicBlob = new File([blob], fileName, { type: 'audio/wav' });
console.log(musicBlob);
return musicBlob;
}
async function audioToBase64(audioFile) {
return new Promise((resolve, reject) => {
let reader = new FileReader();
reader.readAsDataURL(audioFile);
reader.onload = () => resolve(reader.result);
reader.onerror = error => reject(error);
});
}
const gradioEl = document.querySelector('body > gradio-app');
// const gradioEl = document.querySelector("gradio-app").shadowRoot;
const inputPromptEl = gradioEl.querySelector('#prompt-in textarea').value;
const outputImgEl = gradioEl.querySelector('#img-out img');
const outputMusic = gradioEl.querySelector('#music-out audio');
const outputMusic_src = gradioEl.querySelector('#music-out audio').src;
let titleTxt = inputPromptEl;
//if(titleTxt.length > 100){
// titleTxt = titleTxt.slice(0, 100) + ' ...';
//}
const shareBtnEl = gradioEl.querySelector('#share-btn');
const shareIconEl = gradioEl.querySelector('#share-btn-share-icon');
const loadingIconEl = gradioEl.querySelector('#share-btn-loading-icon');
if(!outputMusic){
return;
};
shareBtnEl.style.pointerEvents = 'none';
shareIconEl.style.display = 'none';
loadingIconEl.style.removeProperty('display');
const outputImg = await getInputImgFile(outputImgEl);
const urlOutputImg = await uploadFile(outputImg);
const musicFile = await getOutputMusicFile(outputMusic);
const dataOutputMusic = await uploadFile(musicFile);
const descriptionMd = `#### Spectrogram Image:
#### Spectrogram Sound:
`;
const params = new URLSearchParams({
title: titleTxt,
description: descriptionMd,
});
const paramsStr = params.toString();
window.open(`https://huggingface.co/spaces/nightfury/Riffusion_real-time_image-to-music_generation/discussions/new?${paramsStr}`, '_blank');
shareBtnEl.style.removeProperty('pointer-events');
shareIconEl.style.removeProperty('display');
loadingIconEl.style.display = 'none';
}"""