Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
@@ -145,6 +144,7 @@ h1 {
 
 # JavaScript for the copy button
 copy_js = """
+<script>
 function copyCode() {
     const codeElement = document.querySelector('.output-box pre');
     const codeText = codeElement.innerText;
@@ -152,6 +152,7 @@ function copyCode() {
         alert('Code copied to clipboard!');
     });
 }
+</script>
 """
 
 # Gradio interface for the app
@@ -166,6 +167,9 @@ with gr.Blocks(css=custom_css) as demo:
     gr.Markdown("# Smart Contract Generator")
     gr.Markdown("Generate smart contracts using AI.")
     
+    # Inject JavaScript
+    gr.HTML(copy_js)
+    
     # Output box with copy button
     with gr.Column():
         output_box = gr.HTML(label="Generated Smart Contract", elem_classes="output-box")
@@ -183,9 +187,6 @@ with gr.Blocks(css=custom_css) as demo:
         inputs=[language_dropdown, requirements_input],
         outputs=output_box
     )
-    
-    # Add JavaScript for the copy button
-    demo.load(fn=None, inputs=None, outputs=None, _js=copy_js)
 
 # Launch the Gradio app
 demo.launch()