Update app.py

app.py

@@ -1,39 +1,33 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-# Load smaller CodeT5 model (faster)
+# Load optimized model
 model_name = "Salesforce/codet5-small"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
-# Main function
+# Use GPU if available
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model.to(device)
+
+# Eternos Fixer Function
 def eternos_debugger(code, error):
-    if not code.strip():
-        return "❌ Please provide some code."
-    if not error.strip():
-        return "❌ Please provide the error message you encountered."
-    
-    # Smart prompting
-    prompt = (
-        f"You are an expert Python debugger.\n"
-        f"Given the buggy code and the error message, fix the code.\n\n"
-        f"Code:\n{code}\n\n"
-        f"Error:\n{error}\n\n"
-        f"Corrected Code:"
-    )
+    if not code.strip() or not error.strip():
+        return "❌ Please enter both code and an error message."
+
+    prompt = f"Fix this Python code:\n{code}\nError:\n{error}\nFixed Code:"
+    inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512).to(device)
 
-    inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
     with torch.no_grad():
         outputs = model.generate(
             **inputs,
-            max_length=512,
-            num_beams=4,
+            max_length=128,
+            num_beams=2,
             early_stopping=True,
             temperature=0.7,
-            top_p=0.95
         )
-    
+
     fixed_code = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return fixed_code.strip()
 
@@ -45,24 +39,24 @@ interface = gr.Interface(
         gr.Textbox(label="🚨 Error Message", lines=3, placeholder="Paste the error message you got...")
     ],
     outputs=gr.Code(label="✅ Suggested Fixed Code"),
-    title="🛠️ Eternos: AI Code Debugger",
-    description="Eternos uses a CodeT5 model to help debug and fix Python code. Provide your code and the error message to get a fix.",
-    theme="soft",
-    allow_flagging="never",
+    title="⚙️ Eternos – AI Code Debugger",
+    description="Eternos uses a CodeT5-small model to analyze your Python code and error, then suggest a possible fix.",
     examples=[
         [
-            "def add_numbers(a, b)\n    return a + b",
+            "def greet(name)\n    print('Hello', name)",
             "SyntaxError: expected ':'"
         ],
         [
-            "for i in range(5)\n    print(i)",
-            "SyntaxError: expected ':'"
+            "def divide(a, b):\n    return a / b\n\nprint(divide(5, 0))",
+            "ZeroDivisionError: division by zero"
         ],
         [
-            "def divide(a, b):\n    return a / b\nprint(divide(4, 0))",
-            "ZeroDivisionError: division by zero"
+            "def sqr(x):\n    return x*x\n\nprint(sqre(5))",
+            "NameError: name 'sqre' is not defined"
         ]
-    ]
+    ],
+    theme="soft",
+    allow_flagging="never"
 )
 
 interface.launch()