Update multi_agent.py

multi_agent.py

@@ -74,19 +74,23 @@ def run_multi_agent(llm, message):
 
     print("### markdown_code = " + markdown_code)
 
-    current_folder = os.getcwd()
-    print(current_folder)
     files_in_folder = os.listdir("coding")
 
-    print(f"Files in {current_folder}:")
+    file_name_py = ""
+    file_name_sh = ""
+    
+    print(f"Files in {files_in_folder}:")
     for file in files_in_folder:
         print(file)
-
-    #file_name_py = read_py_file("coding", ".py")
-    #file_name_sh = read_py_file("coding", ".sh")
-
-    #print(file_name_py)
-    #print(file_name_sh)
+        if file :
+            _, file_extension = os.path.splitext(file)
+            if file_extension == "py":
+                file_name_py = file
+            if file_extension == "sh":
+                file_name_sh = file
+    
+    print(file_name_py)
+    print(file_name_sh)
     
     #file_path_py = "coding/" + file_name_py
     #code = read_python_file(file_path_py)