Update share_btn.py

share_btn.py

@@ -65,8 +65,9 @@ share_js = """async () => {
     // const gradioEl = document.querySelector("gradio-app").shadowRoot;
     const inputPromptEl = gradioEl.querySelector('#input-prompt input').value;
     const outputVideoEl = gradioEl.querySelector('#output-video video');
-    //const outputMusic = gradioEl.querySelector('#music-out audio');
-    //const outputMusic_src = gradioEl.querySelector('#music-out audio').src;
+    const outputImgEl = gradioEl.querySelector('#output-img img');
+    const outputMusic = gradioEl.querySelector('#output-music audio');
+    const outputMusic_src = gradioEl.querySelector('#output-music audio').src;
     
     let titleTxt = inputPromptEl;
     //if(titleTxt.length > 100){
@@ -83,14 +84,18 @@ share_js = """async () => {
     loadingIconEl.style.removeProperty('display');
     const outputVideo = await getInputVideoFile(outputVideoEl);
     const urlOutputVideo = await uploadFile(outputVideo);
-    //const outputImg = await getInputImgFile(outputvideoEl);
-    //const urlOutputImg = await uploadFile(outputImg);
-    //const musicFile = await getOutputMusicFile(outputMusic);
-	//const dataOutputMusic = await uploadFile(musicFile);
+    const outputImg = await getInputImgFile(outputImgEl);
+    const urlOutputImg = await uploadFile(outputImg);
+    const musicFile = await getOutputMusicFile(outputMusic);
+	const dataOutputMusic = await uploadFile(musicFile);
     
-    const descriptionMd = `#### Here is my AI generated video:
-    
-<video src='${urlOutputVideo}' style='max-height: 704px;'>
+    const descriptionMd = `#### Here is my AI generated art & music:
+<img src='${urlOutputImg}' style='max-height: 512px;'>
+
+<audio controls>
+  <source src="${dataOutputMusic}" type="audio/wav">
+Your browser does not support the audio element.
+</audio>
 `;
     const params = new URLSearchParams({
         title: titleTxt,