Update multi_agent.py

multi_agent.py

@@ -4,6 +4,32 @@ from autogen.coding import LocalCommandLineCodeExecutor
 import markdown, os
 #from IPython.display import Image
 
+def read_image_file(image_file_path: str) -> str:
+    """
+    Read a PNG image file from the local filesystem and encode it as a base64 string.
+
+    Args:
+        image_file_path (str): The local file path of the image.
+
+    Returns:
+        str: The base64-encoded image data.
+    """
+    with open(image_file_path, "rb") as image_file:
+        image_data = image_file.read()
+        return base64.b64encode(image_data).decode("utf-8")
+
+def generate_markdown_code(image_data: str) -> str:
+    """
+    Generate the markdown code to render an image.
+
+    Args:
+        image_data (str): The base64-encoded image data.
+
+    Returns:
+        str: The markdown code to render the image.
+    """
+    return f"![Image](data:image/png;base64,{image_data})"
+
 def run_multi_agent(llm, message):
     llm_config = {"model": llm}
     
@@ -48,24 +74,28 @@ def run_multi_agent(llm, message):
 
     #print("### "+png_file)
 
-    current_folder = os.getcwd()
-    files_in_folder = os.listdir(current_folder)
+    #current_folder = os.getcwd()
+    #files_in_folder = os.listdir(current_folder)
 
-    print(f"Files in {current_folder}:")
-    for file in files_in_folder:
-        print(file)
+    #print(f"Files in {current_folder}:")
+    #for file in files_in_folder:
+    #    print(file)
     
-    image_path = "/home/user/app/coding/ytd_stock_gains.png"
-    image_path_2 = "/coding/ytd_stock_gains.png"
-    image_path_5 = "coding/ytd_stock_gains.png"
-    image_path_3 = "/app/coding/ytd_stock_gains.png"
-    image_path_4 = "/user/app/coding/ytd_stock_gains.png"
+    #image_path = "/home/user/app/coding/ytd_stock_gains.png"
+    #image_path_2 = "/coding/ytd_stock_gains.png"
+    #image_path_5 = "coding/ytd_stock_gains.png"
+    #image_path_3 = "/app/coding/ytd_stock_gains.png"
+    #image_path_4 = "/user/app/coding/ytd_stock_gains.png"
+
+    image_data = read_image_file("/home/user/app/coding/ytd_stock_gains.png")
+    markdown_code = generate_markdown_code(image_data)
 
-    image_path = os.path.join('coding', 'ytd_stock_gains.png')
-    image_markdown = f"![YTD Stock Gains]({image_path})"
+    return markdown_code
+    #image_path = os.path.join('coding', 'ytd_stock_gains.png')
+    #image_markdown = f"![YTD Stock Gains]({image_path})"
 
-    markdown_text = "![YTD Stock Gains](coding/ytd_stock_gains.png)"
-    html = markdown.markdown(markdown_text)
-    print(html)
+    #markdown_text = "![YTD Stock Gains](coding/ytd_stock_gains.png)"
+    #html = markdown.markdown(markdown_text)
+    #print(html)
 
-    return html #f"![YTD Stock Gains]({image_path})<br />![YTD Stock Gains]({image_path_2})<br />![YTD Stock Gains]({image_path_3})<br />![YTD Stock Gains]({image_path_4})<br />![YTD Stock Gains]({image_path_5})<br />"
+    #return html #f"![YTD Stock Gains]({image_path})<br />![YTD Stock Gains]({image_path_2})<br />![YTD Stock Gains]({image_path_3})<br />![YTD Stock Gains]({image_path_4})<br />![YTD Stock Gains]({image_path_5})<br />"