Update CodeFileEditAtomicFlow.py

CodeFileEditAtomicFlow.py

@@ -30,12 +30,11 @@ class CodeFileEditAtomicFlow(AtomicFlow):
         try:
             with open(file_location, "w") as file:
                 file.write(content)
-            file_written_timestamp = os.path.getmtime(file_location)
 
-            return True, f"Code written to {file_location}", file_location, file_written_timestamp
+            return True, f"Code written to {file_location}", file_location
 
         except Exception as e:
-            return False, str(e), file_location, 0
+            return False, str(e), file_location
 
     def _check_input(self, input_data: Dict[str, Any]):
         assert "code" in input_data, "code is not passed to CodeFileEditAtomicFlow"
@@ -65,19 +64,11 @@ class CodeFileEditAtomicFlow(AtomicFlow):
         content_to_write, file_location_to_write = self._generate_input_to_writer(input_data)
 
         # ~~~ Calling the writer function ~~~
-        result, code_editor_output, temp_file_location, file_written_timestamp = self._write_code_content_to_file(
+        result, code_editor_output, temp_file_location = self._write_code_content_to_file(
             file_location_to_write, content_to_write)
 
-        # ~~~ Opening up the file for the user to see ~~~
-        if result:
-            try:
-                subprocess.run(["code", temp_file_location], timeout=10)
-            except (subprocess.CalledProcessError, subprocess.TimeoutExpired):
-                webbrowser.open(temp_file_location)
-
         # ~~~ Generating return variables ~~~
         response = {}
         response["code_editor_output"] = code_editor_output
         response["temp_code_file_location"] = temp_file_location
-        response["temp_code_file_written_timestamp"] = file_written_timestamp
         return response