Update app.py

app.py

@@ -102,48 +102,33 @@ def export_to_file(output_text):
         file.write(output_text)
     return "Output exported to output.txt"
 
-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", elem_id="output")],
-    title="",
-    theme=gr.themes.Soft()
-)
-
-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
+app = gr.Blocks()
+
 with app:
-    gr.Button("Copy").click(None, None, None, _js=copy_js)
-    gr.Button("Export").click(None, None, None, _js=export_js)
+    with gr.Column():
+        image_input = gr.Image(type="pil", label="")
+        prompt_input = gr.Textbox(label="Your Prompt", value="Analyze")
+        output_field = gr.Markdown(label="Response")
+        
+        with gr.Row():
+            copy_button = gr.Button("Copy")
+            export_button = gr.Button("Export")
+
+    def process_image(image, prompt):
+        response = ImageChat(image, prompt)
+        return response
+
+    def copy_callback(output_text):
+        import pyperclip
+        pyperclip.copy(output_text)
+        return "Copied to clipboard!"
+
+    def export_callback(output_text):
+        with open("output.txt", "w") as file:
+            file.write(output_text)
+        return "Output exported to output.txt"
+
+    copy_button.click(fn=copy_callback, inputs=[output_field], outputs=[])
+    export_button.click(fn=export_callback, inputs=[output_field], outputs=[])
 
-app.launch()
+    app.launch()