Update src/utils.py

src/utils.py

@@ -20,17 +20,18 @@ 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\(\s*["\'](.+?)["\']\s*,\s*["\'](.+?)["\']\s*\)', code_str)
+        match = re.search(r'os\.path\.join\(([^)]+)\)', code_str)
         if match:
-            folder = match.group(1)
-            filename = match.group(2)
-            return os.path.join(folder, filename)
+            parts = re.findall(r'["\']([^"\']+)["\']', match.group(1))
+            if parts:
+                return os.path.join(*parts)
         print("No file path found.")
         return None
     except Exception as e:
         print(f"File path extraction error: {e}")
         raise
 
+
 # Execute extracted Python code in a subprocess using the given interpreter
 def execute_code_in_virtualenv(text, python_interpreter=sys.executable):
     if not python_interpreter: