Update src/utils.py

src/utils.py

@@ -20,11 +20,11 @@ def extract_code(text):
 # Extract file path from a code string that uses os.path.join()
 def extract_file_path(code_str):
     try:
-        match = re.search(r'os\.path\.join\(([^)]+)\)', code_str)
+        match = re.search(r'os\.path\.join\(\s*["\'](.+?)["\']\s*,\s*["\'](.+?)["\']\s*\)', code_str)
         if match:
-            parts = re.findall(r'["\']([^"\']+)["\']', match.group(1))
-            if parts:
-                return os.path.join(*parts)
+            folder = match.group(1)
+            filename = match.group(2)
+            return os.path.join(folder, filename)
         print("No file path found.")
         return None
     except Exception as e: