Update app.py

app.py

@@ -2,22 +2,30 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 
-# Load open-source model
-model_name = "replit/code-v1-3b"  # Can switch to deepseek-ai/deepseek-coder-1.3b-instruct
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
-model.to("cuda" if torch.cuda.is_available() else "cpu")
+# Load a lightweight coding model
+model_id = "replit/code-v1-3b"
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(model_id)
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model.to(device)
 
-# Inference function
-def debug_code(code):
-    prompt = f"### Fix the following code and explain:\n{code}\n### Fixed Code:"
-    inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
-    outputs = model.generate(**inputs, max_new_tokens=256, pad_token_id=tokenizer.eos_token_id)
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
-
-# UI
-gr.Interface(fn=debug_code,
-             inputs=gr.Textbox(lines=12, label="Your Buggy Code"),
-             outputs=gr.Textbox(label="Fixed Code & Explanation"),
-             title="Eternos Lite - Local Debugger").launch()
+# Debug function
+def debug_code(code, language):
+    prompt = f"### Debug this {language} code:\n{code}\n### Provide fixed code and explanation:\n"
+    inputs = tokenizer(prompt, return_tensors="pt").to(device)
+    outputs = model.generate(**inputs, max_new_tokens=300, pad_token_id=tokenizer.eos_token_id)
+    result = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return result
 
+# UI layout
+gr.Interface(
+    fn=debug_code,
+    inputs=[
+        gr.Textbox(lines=12, label="Paste Your Buggy Code Here"),
+        gr.Dropdown(["Python", "C", "C++", "JavaScript"], label="Code Language", value="Python")
+    ],
+    outputs=gr.Textbox(label="Debugged Code & Explanation"),
+    title="🛠 Eternos Basic Debugger",
+    description="Paste your code, choose language, and let AI fix & explain the bugs.",
+    theme="default"
+).launch()