Upgraded to CodeT5 for full code bug fixing

README.md

@@ -1,10 +1,11 @@
 ---
-title: Bug Fixer
-emoji: "🔧"
-colorFrom: red
-colorTo: blue
+title: Bug Fixer - CodeT5
+emoji: 🧠
+colorFrom: pink
+colorTo: purple
 sdk: gradio
 sdk_version: 4.33.0
 app_file: app_ui.py
 pinned: false
----
+---
+s

app_ui.py

@@ -1,27 +1,18 @@
 import gradio as gr
-from bug_detector import detect_bug
-from fix_generator import generate_fix
+from full_code_fixer import fix_code
 
-def bug_fixer(code):
-    bug_status = detect_bug(code)
-    fixed_code = generate_fix(code)
-    return bug_status, fixed_code
+def fix_script(code):
+    fixed = fix_code(code)
+    return "Analyzed", fixed
 
-title = "🔧 Bug Detection and Fixing Tool"
+with gr.Blocks() as demo:
+    gr.Markdown("## 🛠️ Bug Detection and Fixing Tool - Full Script Version")
+    code_input = gr.Textbox(lines=20, label="Paste Your Python Code Here")
+    bug_status = gr.Textbox(label="Status")
+    fixed_code = gr.Textbox(lines=20, label="Suggested Fix")
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("""
-        # 🔧 Bug Detection and Fixing Tool
-        **Final Year Project - Guided by: Prof. Alakananda K P**  
-        **Team: Akanksha K P and Divyashree N**
-    """)
+    gr.Button("Detect and Fix").click(fix_script, inputs=code_input, outputs=[bug_status, fixed_code])
 
-    code_input = gr.Textbox(lines=8, label="Enter Buggy Code")
-    bug_output = gr.Textbox(label="Detected Bug")
-    fix_output = gr.Textbox(label="Suggested Fix")
-    btn = gr.Button("Detect and Fix")
+demo.launch()
 
-    btn.click(fn=bug_fixer, inputs=code_input, outputs=[bug_output, fix_output])
-
-demo.launch(share=True)
 #for the gradio user interface

fix_generator.py

@@ -1,20 +1,16 @@
 def generate_fix(code):
-    lines = code.split("\n")
+    lines = code.split('\n')
     fixed_lines = []
     for line in lines:
-        line = line.strip()
-        if line.startswith("print ") and not line.startswith("print("):
-            content = line[6:].strip()
-            fixed_lines.append(f"print({content})")
+        stripped = line.strip()
+        # Fix print statements missing parentheses
+        if stripped.startswith("print ") and "(" not in stripped:
+            expression = stripped[6:]
+            fixed_line = f"print({expression})"
+            fixed_lines.append(fixed_line)
         else:
             fixed_lines.append(line)
     return "\n".join(fixed_lines)
 
-# Optional test
-if __name__ == "__main__":
-    buggy = "a = input()\nprint a"
-    print(generate_fix(buggy))
-
-
 
 #genrate the fixed version of code using an ML Model

full_code_fixer.py

@@ -0,0 +1,18 @@
+# full_code_fixer.py
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+model_name = "Salesforce/codet5-base"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+
+def fix_code(code):
+    input_text = f"fix Python: {code}"
+    inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True)
+    outputs = model.generate(**inputs, max_length=512, num_beams=4, early_stopping=True)
+    fixed_code = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return fixed_code
+
+# Optional test
+if __name__ == "__main__":
+    buggy_code = "def add(a,b):\n return a*b"  # wrong logic
+    print(fix_code(buggy_code))