Update app.py

app.py

@@ -1,23 +1,40 @@
+import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
-import gradio as gr
 
-# Load DeepSeek Coder model
-model_id = "deepseek-ai/deepseek-coder-1.3b-base"
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
-device = "cuda" if torch.cuda.is_available() else "cpu"
+# Load StarCoder model
+model_name = "bigcode/starcoder"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 
-# Debug function
-def debug_code(code, language):
-    prompt = f"### Task: Fix and explain the following {language} code:\n{code}\n### Output:\n"
-    inputs = tokenizer(prompt, return_tensors="pt").to(device)
-    outputs = model.generate(**inputs, max_new_tokens=300, pad_token_id=tokenizer.eos_token_id)
-    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return response[len(prompt):].strip()
+# Prompts per language
+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"
+}
+
+# Debugging 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 white theme and modern layout
+# Gradio UI
 with gr.Blocks(css="""
 #logo-container {
     display: flex;
@@ -41,16 +58,20 @@ with gr.Blocks(css="""
 }
 """) as demo:
     with gr.Column():
-        with gr.Row(elem_id="logo-container"):
-            gr.Image(value="763f8a67-91ed-4f76-9296-d64a30676efd.png", width=100, height=100, show_label=False, show_download_button=False)
-        gr.Markdown("<div class='title-text'> Eternos Beta — The Coder's friend</div>")
         
-        language = gr.Dropdown(["Python", "C", "C++", "JavaScript"], label="Select Programming Language", value="Python")
-        code_input = gr.Textbox(lines=12, label="Paste Buggy Code", placeholder="Paste or write buggy code here...")
-        output = gr.Textbox(label="Fixed Code & Explanation", lines=14)
+        
+        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")
 
-        debug_btn = gr.Button("🧠 Debug Code")
-        debug_btn.click(fn=debug_code, inputs=[code_input, language], outputs=output)
+        run_btn.click(fn=eternos_debugger, inputs=[code_input, error_input, language_input], outputs=output_code)
 
-# Launch the app
 demo.launch()