Update app.py

app.py

@@ -106,13 +106,44 @@ def process_image(image, prompt):
     response = ImageChat(image, prompt)
     return response
 
+# Custom JS for copy and export buttons
+copy_js = """
+async function copyOutput() {
+    const outputElement = document.getElementById('output');
+    const text = outputElement.innerText;
+    await navigator.clipboard.writeText(text);
+    alert('Copied to clipboard!');
+}
+"""
+
+export_js = """
+function exportOutput() {
+    const outputElement = document.getElementById('output');
+    const text = outputElement.innerText;
+    const blob = new Blob([text], { type: 'text/plain' });
+    const url = URL.createObjectURL(blob);
+    const a = document.createElement('a');
+    a.href = url;
+    a.download = 'output.txt';
+    document.body.appendChild(a);
+    a.click();
+    document.body.removeChild(a);
+}
+"""
+
 app = gr.Interface(
     fn=process_image,
     inputs=[gr.Image(type="pil", label=""), gr.Textbox(label="Your Prompt", value="Analyze")],
-    outputs=[gr.HTML(label="Response")],
+    outputs=[gr.HTML(label="Response", elem_id="output")],
     title="",
     theme=gr.themes.Soft()
 )
 
-# Add custom JS for copy and export buttons
-app.launch(share=True, inline=True, server_name="0.0.0.0", server_port=7860)
+app.launch(share=False, inline=True, server_name="0.0.0.0", server_port=7860)
+
+# Include the buttons and custom JS in the launch method
+with app:
+    gr.Button("Copy").click(None, None, None, _js=copy_js)
+    gr.Button("Export").click(None, None, None, _js=export_js)
+
+app.launch()