Update app.py

app.py

@@ -1,77 +1,53 @@
 import gradio as gr
 import torch
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-# Load StarCoder model
-model_name = "bigcode/starcoder"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+# Load CodeT5+ model and tokenizer
+model_id = "Salesforce/codet5p-770m"
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_id, torch_dtype=torch.float32)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 
-# Prompts per language
+# Prompt templates for supported languages
 language_prompts = {
-    "Python": "### Fix Python code:\n### Buggy code:\n",
-    "C": "### Fix C code:\n### Buggy code:\n",
-    "C++": "### Fix C++ code:\n### Buggy code:\n",
-    "JavaScript": "### Fix JavaScript code:\n### Buggy code:\n"
+    "Python": "Fix the following Python code:\n",
+    "C": "Fix the following C code:\n",
+    "C++": "Fix the following C++ code:\n",
+    "JavaScript": "Fix the following JavaScript code:\n"
 }
 
-# Debugging function
+# Debug function
 def eternos_debugger(code, error, language):
     if not code.strip():
-        return "❌ Please provide code."
-    prompt = f"{language_prompts[language]}{code}\n### Error:\n{error}\n### Corrected code:\n"
-    inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=8192).to(device)
-    outputs = model.generate(
-        **inputs,
-        max_new_tokens=256,
-        temperature=0.2,
-        top_p=0.95,
-        do_sample=False,
-        pad_token_id=tokenizer.eos_token_id
-    )
-    resp = tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True)
-    return resp.strip()
-
-# Gradio UI
-with gr.Blocks(css="""
-#logo-container {
-    display: flex;
-    justify-content: center;
-    margin-top: 10px;
-}
-.title-text {
-    text-align: center;
-    font-size: 28px;
-    font-weight: bold;
-    color: #222;
-    margin-top: 10px;
-    margin-bottom: 30px;
-}
-.gr-button {
-    background-color: #2F80ED !important;
-    color: white !important;
-    font-weight: bold;
-    border-radius: 8px;
-    padding: 10px 16px;
-}
-""") as demo:
-    with gr.Column():
-        
-        
-        gr.Markdown("<div class='title-text'> Eternos Beta — A Coder's Friend </div>")
-
-        with gr.Row():
-            language_input = gr.Dropdown(["Python", "C", "C++", "JavaScript"], label="🌐 Language", value="Python")
-
-        with gr.Row():
-            code_input = gr.Textbox(label="📝 Your Code", lines=12, placeholder="Paste or write buggy code here...")
-            error_input = gr.Textbox(label="⚠️ Error Message (optional)", lines=4, placeholder="Paste the error traceback here (optional)")
-
-        run_btn = gr.Button("🧠 Fix Code")
-        output_code = gr.Code(label="✅ Suggested Fix")
-
-        run_btn.click(fn=eternos_debugger, inputs=[code_input, error_input, language_input], outputs=output_code)
+        return "❌ Please enter some code to debug."
+    prompt = language_prompts[language] + code + "\nError:\n" + error + "\nFixed Code:\n"
+
+    inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512).to(device)
+    with torch.no_grad():
+        outputs = model.generate(
+            **inputs,
+            max_new_tokens=256,
+            temperature=0.3,
+            top_p=0.9,
+            do_sample=True
+        )
+    result = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return result.strip()
+
+# Gradio Interface
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("##  Eternos — A Coder,s Friend")
+    gr.Markdown("Supports Python, C, C++, JavaScript — Powered by CodeT5+")
+
+    with gr.Row():
+        code_input = gr.Textbox(label="👨‍💻 Your Code", lines=14, placeholder="Paste your buggy code here...")
+        error_input = gr.Textbox(label="⚠️ Error Message (optional)", lines=4)
+
+    language_input = gr.Dropdown(["Python", "C", "C++", "JavaScript"], label="🌐 Language", value="Python")
+    output_code = gr.Code(label="✅ Suggested Fix")
+    run_btn = gr.Button("🔍 Debug Code")
+
+    run_btn.click(fn=eternos_debugger, inputs=[code_input, error_input, language_input], outputs=output_code)
 
 demo.launch()