Update app.py

app.py

@@ -2,14 +2,14 @@ import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-# Load CodeT5+ model
+# 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)
 
-# Prompt templates
+# Prompt templates for supported languages
 language_prompts = {
     "Python": "Fix the following Python code:\n",
     "C": "Fix the following C code:\n",
@@ -17,11 +17,12 @@ language_prompts = {
     "JavaScript": "Fix the following JavaScript code:\n"
 }
 
-# Debugging function
+# Debug function
 def eternos_debugger(code, error, language):
     if not code.strip():
         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(
@@ -34,22 +35,10 @@ def eternos_debugger(code, error, language):
     result = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return result.strip()
 
-# Custom CSS to set background color
-custom_css = """
-body {
-    background-color: #cbedec !important;
-    color: #000000 !important;
-}
-textarea, .prose textarea, .prose code, .gr-code, .gr-box {
-    background-color: white !important;
-    color: black !important;
-}
-"""
-
 # Gradio Interface
-with gr.Blocks(css=custom_css) as demo:
-    gr.Markdown("## 🛠️ <span style='color:#003333;'>Eternos — AI Code Debugger</span>")
-    gr.Markdown("<span style='color:#004444;'>Supports Python, C, C++, JavaScript — Powered by CodeT5+</span>")
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("## 🛠️ Eternos — AI Code Debugger")
+    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...")