Update app.py

app.py

@@ -16,11 +16,6 @@ except ImportError:
     GENAI_AVAILABLE = False
     print("Warning: google-generativeai not installed, will attempt on-demand import")
 
-try:
-    import pyperclip
-except ImportError:
-    pyperclip = None
-
 # --- Environment Configuration ---
 GEMINI_KEY = os.environ.get("GEMINI_KEY", "")
 DEFAULT_PORT = int(os.environ.get("PORT", 7860))
@@ -253,31 +248,60 @@ def build_interface():
             api_name="generate"
         )
         
-        # Implement browser-based clipboard functionality instead of pyperclip
+        # Add a JavaScript function for clipboard functionality
+        app.load(None, None, None, _js="""
+        function setupCopyButton() {
+            // Find the copy button by its text
+            const copyButtons = Array.from(document.querySelectorAll('button')).filter(
+                button => button.textContent.includes('Copy')
+            );
+            
+            if (copyButtons.length > 0) {
+                const copyBtn = copyButtons[0];
+                copyBtn.addEventListener('click', function() {
+                    // Find the prompt output textarea
+                    const textareas = document.querySelectorAll('textarea');
+                    let promptTextarea = null;
+                    
+                    for (let textarea of textareas) {
+                        if (textarea.closest('div').querySelector('label')?.textContent.includes('Optimized Prompt')) {
+                            promptTextarea = textarea;
+                            break;
+                        }
+                    }
+                    
+                    if (promptTextarea && promptTextarea.value) {
+                        navigator.clipboard.writeText(promptTextarea.value)
+                            .then(() => {
+                                const originalText = copyBtn.textContent;
+                                copyBtn.textContent = "✓ Copied!";
+                                setTimeout(() => {
+                                    copyBtn.textContent = originalText;
+                                }, 2000);
+                            })
+                            .catch(err => {
+                                console.error('Failed to copy: ', err);
+                                alert('Copy failed. Please select and copy manually.');
+                            });
+                    }
+                });
+            }
+        }
+        
+        // Run our setup once the DOM is fully loaded
+        if (document.readyState === 'loading') {
+            document.addEventListener('DOMContentLoaded', setupCopyButton);
+        } else {
+            // DOM already loaded, run setup
+            setTimeout(setupCopyButton, 1000);
+        }
+        """)
+        
+        # Add a simple no-op function for the copy button
         copy_btn.click(
-            None,  # No Python function needed
+            lambda x: x,
             inputs=prompt_output,
-            outputs=None,
-            _js="""
-            (text) => {
-                if (!text) return;
-                navigator.clipboard.writeText(text)
-                    .then(() => {
-                        // Show temporary success message
-                        const button = document.querySelector("button:contains('Copy')");
-                        const originalText = button.innerText;
-                        button.innerText = "✓ Copied!";
-                        setTimeout(() => {
-                            button.innerText = originalText;
-                        }, 2000);
-                    })
-                    .catch(err => {
-                        console.error('Failed to copy: ', err);
-                        alert('Copy failed. Please select and copy manually.');
-                    });
-                return;
-            }
-            """
+            outputs=prompt_output
         )
 
     return app