Fix share button selectors

app.py

@@ -64,8 +64,8 @@ with gr.Blocks(css=css) as demo:
                             )
         with gr.Row():
             diffuse_btn = gr.Button(value="Diffuse it!")
-      with gr.Column():
-        sd_output = gr.Gallery().style(grid=2, height="auto", elem_id="generated-gallery")
+      with gr.Column(elem_id="generated-gallery"):
+        sd_output = gr.Gallery().style(grid=2, height="auto")
         with gr.Group(elem_id="share-btn-container"):
             community_icon = gr.HTML(community_icon_html)
             loading_icon = gr.HTML(loading_icon_html)

share_btn.py

@@ -24,8 +24,9 @@ share_js = """async () => {
 	}
 
     const gradioEl = document.querySelector('body > gradio-app');
+    // const gradioEl = document.querySelector("gradio-app").shadowRoot;
     const imgEls = gradioEl.querySelectorAll('#generated-gallery img');
-    const promptTxt = gradioEl.querySelector('#translated input').value;
+    const promptTxt = gradioEl.querySelector('#translated textarea').value;
     const shareBtnEl = gradioEl.querySelector('#share-btn');
     const shareIconEl = gradioEl.querySelector('#share-btn-share-icon');
     const loadingIconEl = gradioEl.querySelector('#share-btn-loading-icon');