Update app.py

app.py

@@ -2,14 +2,14 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 
-# Load DeepSeek Coder
+# 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"
 model.to(device)
 
-# Debugging function
+# 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)
@@ -17,16 +17,40 @@ def debug_code(code, language):
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return response[len(prompt):].strip()
 
-# Gradio UI
-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="Fixed Code & Explanation"),
-    title="🛠 Eternos Debugger — DeepSeek Edition",
-    description="Paste code, select language, and get a fix + explanation. Powered by DeepSeek Coder.",
-    theme="default"
-).launch()
+# Gradio UI with white theme and modern layout
+with gr.Blocks(css="""
+#logo-container {
+    display: flex;
+    justify-content: center;
+    margin-top: 10px;
+}
+.title-text {
+    text-align: center;
+    font-size: 28px;
+    font-weight: bold;
+    color: #222;
+    margin-top: 10px;
+    margin-bottom: 30px;
+}
+.gr-button {
+    background-color: #2F80ED !important;
+    color: white !important;
+    font-weight: bold;
+    border-radius: 8px;
+    padding: 10px 16px;
+}
+""") 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)
 
+        debug_btn = gr.Button("🧠 Debug Code")
+        debug_btn.click(fn=debug_code, inputs=[code_input, language], outputs=output)
+
+# Launch the app
+demo.launch()