Update multi_agent.py

multi_agent.py

@@ -11,26 +11,11 @@ def read_image_file(image_file_path: str) -> str:
 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:
+def read_file(file_path: str) -> str:
     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.
-
-    Args:
-        code (str): The Python code to format.
-
-    Returns:
-        str: The formatted Markdown string.
-    """
     markdown_code = '```\n'
     markdown_code += code
     markdown_code += '\n```'
@@ -92,13 +77,15 @@ def run_multi_agent(llm, message):
     print(file_name_py)
     print(file_name_sh)
     
-    #file_path_py = "coding/" + file_name_py
-    #code = read_python_file(file_path_py)
-    #markdown_code = format_as_markdown(code)
-
-    #file_path_sh = "coding/" + file_name_sh
-    #code = read_python_file(file_path_sh)
-    #markdown_code += "<br />" + format_as_markdown(code)
+    file_path_py = "coding/" + file_name_py
+    code_py = read_file(file_path_py)
+    markdown_code_py = format_as_markdown(code_py)
+    print(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)
+    print(markdown_code_sh)
     
     #print("### markdown_code = " + markdown_code)
     
@@ -106,4 +93,4 @@ def run_multi_agent(llm, message):
     #print(chat_result)
     #print("###")
     
-    return markdown_code
+    return markdown_code + "<br />" + markdown_code_sh + "<br />" + markdown_code_py