Update multi_agent.py

multi_agent.py

@@ -1,4 +1,4 @@
-import base64, json, os
+import base64, datetime, json, os
 
 from autogen import ConversableAgent, AssistantAgent
 from autogen.coding import LocalCommandLineCodeExecutor
@@ -20,6 +20,24 @@ def format_as_markdown(code: str) -> str:
     markdown_code += code
     markdown_code += '\n```'
     return markdown_code
+
+def get_latest_file(directory, file_extension):
+    latest_file = None
+    latest_date = datetime.datetime.min
+
+    for file in os.listdir(directory):
+        if file:
+            _, file_ext = os.path.splitext(file)
+
+            if file_ext == file_extension:
+                file_path = os.path.join(directory, file)
+                file_date = datetime.datetime.fromtimestamp(os.path.getmtime(file_path))
+
+                if file_date > latest_date:
+                    latest_date = file_date
+                    latest_file = file
+
+    return latest_file
     
 def run_multi_agent(llm, task):
     llm_config = {"model": llm}
@@ -58,13 +76,13 @@ def run_multi_agent(llm, task):
             chat += f"**{message['role'].replace('assistant', 'Code Executor').replace('user', 'Code Writer')}**\n{message['content']}\n\n"
         first_message = False
 
-    file_name_png = ""
-    
-    for file in os.listdir("coding"):
-        if file:
-            _, file_extension = os.path.splitext(file)
-            if file_extension == ".png":
-                file_name_png = file
+    #for file in os.listdir("coding"):
+    #    if file:
+    #        _, file_extension = os.path.splitext(file)
+    #        if file_extension == ".png":
+    #            file_name_png = file
+
+    file_name_png = get_latest_file("coding", ".png")
     
     image_data = read_image_file(f"/home/user/app/coding/{file_name_png}")
     markdown_code_png = generate_markdown_image(image_data)