Update app.py

app.py

@@ -2,14 +2,14 @@ import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-# Load CodeT5+ model and tokenizer
+# Load CodeT5+ model
 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 for supported languages
+# Prompt templates
 language_prompts = {
     "Python": "Fix the following Python code:\n",
     "C": "Fix the following C code:\n",
@@ -17,12 +17,11 @@ language_prompts = {
     "JavaScript": "Fix the following JavaScript code:\n"
 }
 
-# Debug function
+# Debugging 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(
@@ -35,10 +34,22 @@ 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(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("##  Eternos — A Coder's Friend")
-    gr.Markdown("Supports debugging of Python, The C Family and JavaScript")
+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.Row():
         code_input = gr.Textbox(label="👨‍💻 Your Code", lines=14, placeholder="Paste your buggy code here...")