Update multi_agent.py

multi_agent.py

@@ -12,18 +12,15 @@ def generate_markdown_code(image_data: str) -> str:
     return f"![Image](data:image/png;base64,{image_data})"
 
 def read_python_file(file_path: str) -> str:
-    """
-    Reads a Python file and returns its contents as a string.
-
-    Args:
-        file_path (str): The path to the Python file.
-
-    Returns:
-        str: The contents of the Python file.
-    """
     with open(file_path, 'r', encoding='utf-8') as file:
         return file.read()
 
+def read_py_file(folder_path: str, extension: str) -> str:
+    for filename in os.listdir(folder_path):
+        if filename.endswith(extension):
+            with open(os.path.join(folder_path, filename), "r", encoding="utf-8") as file:
+                return file.read()
+
 def format_as_markdown(code: str) -> str:
     """
     Formats a string of Python code as Markdown.
@@ -84,15 +81,22 @@ def run_multi_agent(llm, message):
     print(f"Files in {current_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")
     
-    file_path = "coding/ytd_stock_gains.py"
+    file_path = "coding/" + file_name_py
     code = read_python_file(file_path)
     markdown_code = format_as_markdown(code)
+
+    file_path = "coding/" + file_name_sh
+    code = read_python_file(file_path)
+    markdown_code += "<br />" format_as_markdown(code)
     
-    #print("### markdown_code = " + markdown_code)
+    print("### markdown_code = " + markdown_code)
     
     print("###")
-    print(chat_result)
+    #print(chat_result)
     print("###")
     
     return markdown_code