Update multi_agent.py

multi_agent.py

@@ -59,10 +59,16 @@ def run_multi_agent(llm, message):
     for file in os.listdir("coding"):
         if file:
             _, file_extension = os.path.splitext(file)
+            if file_extension == ".py":
+                file_name_py = file
             if file_extension == ".sh":
                 file_name_sh = file
     
     try:
+        file_path_py = "coding/" + file_name_py
+        code_py = read_file(file_path_py)
+        markdown_code_py = format_as_markdown(code_py)
+
         file_path_sh = "coding/" + file_name_sh
         code_sh = read_file(file_path_sh)
         markdown_code_sh = format_as_markdown(code_sh)
@@ -73,7 +79,7 @@ def run_multi_agent(llm, message):
     except Exception as e:
         print(f"An unexpected error occurred: {e}")
 
-    result = markdown_code_png + "\n" + markdown_code_sh
+    result = markdown_code_png + "\n" + markdown_code_py + "\n" + markdown_code_sh
 
     print("===")
     print(result)