Update app.py

app.py

@@ -4,8 +4,8 @@ import requests
 import gradio as gr
 import torch
 
-# Load the Hugging Face model and tokenizer (only once)
-model_name = "distilgpt2"  # Smaller and faster model
+# Load the CodeParrot model and tokenizer (only once)
+model_name = "codeparrot/codeparrot-small"  # CodeParrot model
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
@@ -31,9 +31,9 @@ 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 the Hugging Face model to generate code
+    # Use the CodeParrot model to generate code
     inputs = tokenizer(prompt, return_tensors="pt").to("cuda" if torch.cuda.is_available() else "cpu")
-    outputs = model.generate(**inputs, max_length=150)  # Reduced max_length
+    outputs = model.generate(**inputs, max_length=300)  # Increased max_length for better results
     generated_code = tokenizer.decode(outputs[0], skip_special_tokens=True)
     
     # Enhance the code using Groq API
@@ -104,9 +104,7 @@ h1 {
     text-align: center;
     font-size: 2.5em;
     margin-bottom: 20px;
-    background: linear-gradient(135deg, #6a11cb 0%, #2575fc 100%);
-    -webkit-background-clip: text;
-    -webkit-text-fill-color: transparent;
+    color: white; /* White title color */
     transform-style: preserve-3d;
     transform: translateZ(30px);
 }
@@ -132,10 +130,13 @@ h1 {
 def generate_contract(language, requirements):
     return generate_smart_contract(language, requirements)
 
+# Dropdown options for programming languages
+languages = ["Solidity", "Vyper", "Rust", "JavaScript", "Python"]
+
 interface = gr.Interface(
     fn=generate_contract,
     inputs=[
-        gr.Textbox(label="Programming Language", placeholder="e.g., Solidity"),
+        gr.Dropdown(label="Programming Language", choices=languages, value="Solidity"),  # Dropdown menu
         gr.Textbox(label="Requirements", placeholder="e.g., ERC20 token with minting functionality")
     ],
     outputs=gr.Textbox(label="Generated Smart Contract"),