Update app.py

app.py

@@ -1,26 +1,12 @@
 # app.py
 import gradio as gr
-from groq import Groq
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
 
-# Groq API configuration
-GROQ_API_KEY = "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"  # Replace with your Groq API key
-client = Groq(api_key=GROQ_API_KEY)
-
-# Function to query Groq API
-def query_groq(prompt):
-    try:
-        chat_completion = client.chat.completions.create(
-            messages=[
-                {
-                    "role": "user",
-                    "content": prompt,
-                }
-            ],
-            model="llama-3.3-70b-versatile",  # Use the correct model
-        )
-        return chat_completion.choices[0].message.content
-    except Exception as e:
-        return f"Error querying Groq API: {str(e)}"
+# Load the CodeParrot model and tokenizer
+model_name = "codeparrot/codeparrot-small"  # Replace with your desired model
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
 # Function to generate smart contract code
 def generate_smart_contract(language, requirements):
@@ -28,14 +14,18 @@ def generate_smart_contract(language, requirements):
         # Create a prompt for the model
         prompt = f"Generate a {language} smart contract with the following requirements: {requirements}"
         
-        # Use Groq API to generate code
-        generated_code = query_groq(prompt)
+        # Tokenize the input
+        inputs = tokenizer(prompt, return_tensors="pt").to("cuda" if torch.cuda.is_available() else "cpu")
+        
+        # Generate code
+        outputs = model.generate(**inputs, max_length=500)  # Adjust max_length as needed
+        generated_code = tokenizer.decode(outputs[0], skip_special_tokens=True)
         
         return generated_code
     except Exception as e:
         return f"Error generating smart contract: {str(e)}"
 
-# Custom CSS for a dark box with green text
+# Custom CSS for a dark box with green text and copy button
 custom_css = """
 body {
     font-family: 'Helvetica Neue', Helvetica, Arial, sans-serif;
@@ -130,6 +120,37 @@ h1 {
     line-height: 1.5;
     overflow-x: auto;
     white-space: pre-wrap; /* Preserve formatting */
+    position: relative;
+}
+
+/* Copy button */
+.copy-button {
+    position: absolute;
+    top: 10px;
+    right: 10px;
+    background: #2575fc;
+    border: none;
+    border-radius: 5px;
+    color: #fff;
+    padding: 5px 10px;
+    font-size: 12px;
+    cursor: pointer;
+    transition: background 0.3s ease;
+}
+
+.copy-button:hover {
+    background: #6a11cb;
+}
+"""
+
+# JavaScript for the copy button
+copy_js = """
+function copyCode() {
+    const codeElement = document.querySelector('.output-box pre');
+    const codeText = codeElement.innerText;
+    navigator.clipboard.writeText(codeText).then(() => {
+        alert('Code copied to clipboard!');
+    });
 }
 """
 
@@ -145,8 +166,10 @@ with gr.Blocks(css=custom_css) as demo:
     gr.Markdown("# Smart Contract Generator")
     gr.Markdown("Generate smart contracts using AI.")
     
-    # Output box
-    output_box = gr.HTML(label="Generated Smart Contract", elem_classes="output-box")
+    # Output box with copy button
+    with gr.Column():
+        output_box = gr.HTML(label="Generated Smart Contract", elem_classes="output-box")
+        copy_button = gr.Button("Copy Code", elem_classes="copy-button")
     
     # Input row
     with gr.Row(equal_height=True, variant="panel"):
@@ -160,6 +183,9 @@ 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()