Update multi_agent.py

multi_agent.py

@@ -11,6 +11,34 @@ 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:
+    """
+    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 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 = '```python\n'
+    markdown_code += code
+    markdown_code += '\n```'
+    return markdown_code
+
 def run_multi_agent(llm, message):
     llm_config = {"model": llm}
     
@@ -49,6 +77,12 @@ def run_multi_agent(llm, message):
 
     print("### markdown_code = " + markdown_code)
 
+    file_path = "/home/user/app/coding/ytd_stock_gains.py"
+    code = read_python_file(file_path)
+    markdown_code += format_as_markdown(code)
+
+    print("### markdown_code = " + markdown_code)
+    
     print("###")
     print(chat_result)
     print("###")