Update ParseFeedbackAtomicFlow.py

ParseFeedbackAtomicFlow.py

@@ -58,13 +58,13 @@ class ParseFeedbackAtomicFlow(AtomicFlow):
         elif code_file_exists and plan_file_exists:
             raise AssertionError("Both code file and plan file are passed to ParseFeedbackAtomic, which one to parse?")
 
-    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 run(
             self,
             input_data: Dict[str, Any]
@@ -73,16 +73,14 @@ class ParseFeedbackAtomicFlow(AtomicFlow):
         self._check_input(input_data)
         if "temp_code_file_location" in input_data:
             temp_code_file_location = input_data['temp_code_file_location']
-            last_check_time = input_data["temp_code_file_written_timestamp"]
-            self._wait_for_file_update(temp_code_file_location, last_check_time)
+            self._open_file_and_wait_for_file_update(temp_code_file_location, last_check_time)
             code_content = self._read_content(temp_code_file_location, "code")
             user_thoughts = self._parse_user_thoughts(temp_code_file_location)
             response = {"code": code_content, "feedback": user_thoughts}
 
         elif "temp_plan_file_location" in input_data:
             plan_file_location = input_data["temp_plan_file_location"]
-            last_check_time = input_data["temp_plan_file_written_timestamp"]
-            self._wait_for_file_update(plan_file_location, last_check_time)
+            self._open_file_and_wait_for_file_update(plan_file_location, last_check_time)
             plan_content = self._read_content(plan_file_location, "plan")
             user_thoughts = self._parse_user_thoughts(plan_file_location)
             response = {"plan": plan_content, "feedback": user_thoughts}