Update CodeTestingAtomicFlow.py

CodeTestingAtomicFlow.py

@@ -5,12 +5,12 @@ from flow_modules.Tachi67.InterpreterFlowModule import InterpreterAtomicFlow
 
 
 class CodeTestingAtomicFlow(InterpreterAtomicFlow):
-    def _wait_for_file_update(self, file_location, last_check_time, check_interval=1):
+    def _open_file_and_wait_for_file_update(self, file_location, check_interval=1):
+        process = subprocess.Popen(["code", "--wait", file_location])
         while True:
-            time.sleep(check_interval)
-            current_time = os.path.getmtime(file_location)
-            if current_time != last_check_time:
+            if process.poll() is not None:
                 break
+            time.sleep(check_interval)
     def _prepare_code_and_lang(self, input_data: Dict[str, Any]):
         file_location = input_data["temp_code_file_location"]
         input_data["language"] = "python"
@@ -20,7 +20,6 @@ class CodeTestingAtomicFlow(InterpreterAtomicFlow):
 
     def _check_input(self, input_data: Dict[str, Any]):
         assert "temp_code_file_location" in input_data, "temp_code_file_location not passed to CodeTestingAtomicFlow"
-        assert "temp_code_file_written_timestamp" in input_data, "temp_code_file_written_timestamp not passed to CodeTestingAtomicFlow"
 
     def _delete_file(self, file_location):
         if os.path.exists(file_location):
@@ -31,8 +30,7 @@ class CodeTestingAtomicFlow(InterpreterAtomicFlow):
             input_data: Dict[str, Any]):
         self._check_input(input_data)
         file_loc = input_data["temp_code_file_location"]
-        last_timestamp = input_data["temp_code_file_written_timestamp"]
-        self._wait_for_file_update(file_loc, last_timestamp)
+        self._open_file_and_wait_for_file_update(file_loc)
         self._prepare_code_and_lang(input_data)
         self._process_input_data(input_data)
         response = self._call()