Update app.py

app.py

@@ -2,21 +2,29 @@ import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-# Load optimized model
+# Use light model for fast inference
 model_name = "Salesforce/codet5-small"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
-# Use GPU if available
+# Auto-detect device
 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() or not error.strip():
-        return "❌ Please enter both code and an error message."
+# Supported languages and file extensions
+language_prompts = {
+    "Python": "Fix the following Python code and provide the corrected version:",
+    "C": "Fix the following C code and provide the corrected version:",
+    "C++": "Fix the following C++ code and provide the corrected version:",
+    "JavaScript": "Fix the following JavaScript code and provide the corrected version:"
+}
+
+def eternos_debugger(code, error, language):
+    if not code.strip() or not language.strip():
+        return "❌ Please enter code and select a language."
+
+    prompt = f"{language_prompts[language]}\n{code}\nError:\n{error}\nCorrected Code:"
 
-    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)
 
     with torch.no_grad():
@@ -25,38 +33,28 @@ def eternos_debugger(code, error):
             max_length=128,
             num_beams=2,
             early_stopping=True,
-            temperature=0.7,
+            temperature=0.7
         )
 
     fixed_code = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return fixed_code.strip()
 
-# Gradio UI
-interface = gr.Interface(
-    fn=eternos_debugger,
-    inputs=[
-        gr.Textbox(label="🐞 Buggy Code", lines=12, placeholder="Paste your Python code here..."),
-        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-small model to analyze your Python code and error, then suggest a possible fix.",
-    examples=[
-        [
-            "def greet(name)\n    print('Hello', name)",
-            "SyntaxError: expected ':'"
-        ],
-        [
-            "def divide(a, b):\n    return a / b\n\nprint(divide(5, 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()
+# Gradio Interface
+with gr.Blocks(theme=gr.themes.Monochrome(), css="body {background-color: black !important;}") as demo:
+    gr.Markdown("<h1 style='color: white;'>⚙️ Eternos — AI Code Debugger</h1>")
+    gr.Markdown("<p style='color: grey;'>Fast AI-powered debugger for Python, C, C++, and JavaScript</p>")
+
+    with gr.Row():
+        code_input = gr.Textbox(label="📝 Your Code", lines=15, placeholder="Paste your buggy code here...")
+        error_input = gr.Textbox(label="⚠️ Error Message", lines=4, placeholder="Paste error (optional)...")
+
+    language_input = gr.Dropdown(choices=["Python", "C", "C++", "JavaScript"], label="🌐 Language", value="Python")
+
+    output_code = gr.Code(label="✅ Suggested Fix")
+
+    run_button = gr.Button("🛠️ Fix Code")
+
+    run_button.click(fn=eternos_debugger, inputs=[code_input, error_input, language_input], outputs=output_code)
+
+demo.launch()
+