Final working bug fixer with fixed detection and suggestion

bug_detector.py

@@ -15,6 +15,5 @@ def detect_bug(code):
 if __name__ == "__main__":
     sample = "def multiply(a, b): return a + b"
     print(detect_bug(sample))
-
 #detects if there's a bug in code
 

fix_generator.py

@@ -2,13 +2,13 @@ def generate_fix(code):
     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()
+        if line.strip().startswith("print ") and not line.strip().startswith("print("):
+            content = line.strip()[6:].strip()
             fixed_lines.append(f"print({content})")
         else:
             fixed_lines.append(line)
     return "\n".join(fixed_lines)
+
 # Optional test
 if __name__ == "__main__":
     buggy = "a = input()\nprint a"