From 5e80166b63a0f1b0d793f0bffbe0a016650fe80c Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Fri, 21 Jul 2023 18:52:00 +0530
Subject: [PATCH 01/23] Added functions to write various file types and a file
 handler too

---
 superagi/resource_manager/file_manager.py | 58 +++++++++++++++++++----
 1 file changed, 48 insertions(+), 10 deletions(-)

diff --git a/superagi/resource_manager/file_manager.py b/superagi/resource_manager/file_manager.py
index e9059ac20..7a48da53d 100644
--- a/superagi/resource_manager/file_manager.py
+++ b/superagi/resource_manager/file_manager.py
@@ -7,11 +7,16 @@
 from superagi.models.agent import Agent
 from superagi.models.agent_execution import AgentExecution
 from superagi.types.storage_types import StorageType
+
+class UnsupportedFileTypeError(Exception):
+    pass
+
 class FileManager:
     def __init__(self, session: Session, agent_id: int = None, agent_execution_id: int = None):
         self.session = session
         self.agent_id = agent_id
         self.agent_execution_id = agent_execution_id
+        
     def write_binary_file(self, file_name: str, data):
         if self.agent_id is not None:
             final_path = ResourceHelper.get_agent_write_resource_path(file_name,
@@ -31,6 +36,7 @@ def write_binary_file(self, file_name: str, data):
             return f"Binary {file_name} saved successfully"
         except Exception as err:
             return f"Error write_binary_file: {err}"
+        
     def write_to_s3(self, file_name, final_path):
         with open(final_path, 'rb') as img:
             resource = ResourceHelper.make_written_file_resource(file_name=file_name,
@@ -54,15 +60,9 @@ def write_file(self, file_name: str, content):
                                                                                                    self.agent_execution_id))
         else:
             final_path = ResourceHelper.get_resource_path(file_name)
-        try:
-            with open(final_path, mode="w") as file:
-                file.write(content)
-                file.close()
-            self.write_to_s3(file_name, final_path)
-            logger.info(f"{file_name} - File written successfully")
-            return f"{file_name} - File written successfully"
-        except Exception as err:
-            return f"Error write_file: {err}"
+        
+        self.save_file_by_type(file_name=file_name, file_path=final_path, content=content)
+        
     def write_csv_file(self, file_name: str, csv_data):
         if self.agent_id is not None:
             final_path = ResourceHelper.get_agent_write_resource_path(file_name,
@@ -84,13 +84,30 @@ def write_csv_file(self, file_name: str, csv_data):
         except Exception as err:
             return f"Error write_csv_file: {err}"
         
-
+    def write_pdf_file(self, file_name: str ,file_path: str, content):
+        pass
+    
+    def write_docx_file(self, file_name: str ,file_path: str, content):
+        pass
+    
+    def write_txt_file(self, file_name: str ,file_path: str, content):
+        try:
+            with open(file_path, mode="w") as file:
+                file.write(content)
+                file.close()
+            self.write_to_s3(file_name, file_path)
+            logger.info(f"{file_name} - File written successfully")
+            return f"{file_name} - File written successfully"
+        except Exception as err:
+            return f"Error write_file: {err}"
+    
     def get_agent_resource_path(self, file_name: str):
         return ResourceHelper.get_agent_write_resource_path(file_name, agent=Agent.get_agent_from_id(self.session,
                                                                                                      self.agent_id),
                                                             agent_execution=AgentExecution
                                                             .get_agent_execution_from_id(self.session,
                                                                                          self.agent_execution_id))
+        
     def read_file(self, file_name: str):
         if self.agent_id is not None:
             final_path = self.get_agent_resource_path(file_name)
@@ -104,6 +121,7 @@ def read_file(self, file_name: str):
             return content
         except Exception as err:
             return f"Error while reading file {file_name}: {err}"
+        
     def get_files(self):
         """
         Gets all file names generated by the CodingTool.
@@ -122,3 +140,23 @@ def get_files(self):
             logger.error(f"Error while accessing files in {final_path}: {err}")
             files = []
         return files
+
+    def save_file_by_type(self, file_name: str, file_path: str, content):
+        
+        # Extract the file type from the file_name
+        file_type = file_name.split('.')[-1].lower()
+        
+        # Dictionary to map file extensions to corresponding functions
+        file_type_handlers = {
+            'txt': write_txt_file,
+            'pdf': write_pdf_file,
+            'docx': write_docx_file, 
+            'csv': write_csv_file
+            # NOTE: Add more file types and corresponding functions as needed, These functions should be defined 
+        }
+        
+        if file_path in file_type_handlers:
+            file_type_handlers[file_type](file_name, file_path, content)
+        else:
+            raise UnsupportedFileTypeError(f"Unsupported file type: {file_type}. Cannot save the file.")
+        
\ No newline at end of file

From 2738a9a6c0a28512712691268b6bfe96ecd6755c Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Wed, 26 Jul 2023 11:53:51 +0530
Subject: [PATCH 02/23] FileManager updated to handle and save HTMLs.

---
 superagi/resource_manager/file_manager.py | 32 +++++++++++------------
 1 file changed, 16 insertions(+), 16 deletions(-)

diff --git a/superagi/resource_manager/file_manager.py b/superagi/resource_manager/file_manager.py
index 7a48da53d..5ebf1215f 100644
--- a/superagi/resource_manager/file_manager.py
+++ b/superagi/resource_manager/file_manager.py
@@ -50,7 +50,7 @@ def write_to_s3(self, file_name, final_path):
                 s3_helper = S3Helper()
                 s3_helper.upload_file(img, path=resource.path)
 
-    def write_file(self, file_name: str, content):
+    def write_file(self, file_name: str, content, return_file_path: bool = False):
         if self.agent_id is not None:
             final_path = ResourceHelper.get_agent_write_resource_path(file_name,
                                                                       agent=Agent.get_agent_from_id(self.session,
@@ -61,18 +61,17 @@ def write_file(self, file_name: str, content):
         else:
             final_path = ResourceHelper.get_resource_path(file_name)
         
-        self.save_file_by_type(file_name=file_name, file_path=final_path, content=content)
+        try:
+            self.save_file_by_type(file_name=file_name, file_path=final_path, content=content)
+        except Exception as err:
+            return f"Error write_file: {err}"
         
-    def write_csv_file(self, file_name: str, csv_data):
-        if self.agent_id is not None:
-            final_path = ResourceHelper.get_agent_write_resource_path(file_name,
-                                                                      agent=Agent.get_agent_from_id(self.session,
-                                                                                                    self.agent_id),
-                                                                      agent_execution=AgentExecution
-                                                                      .get_agent_execution_from_id(self.session,
-                                                                                                   self.agent_execution_id))
+        if return_file_path:
+            return final_path
         else:
-            final_path = ResourceHelper.get_resource_path(file_name)
+            return f"{file_name} - File written successfully"
+        
+    def write_csv_file(self, file_name: str, final_path: str, csv_data) -> str:
         try:
             with open(final_path, mode="w") as file:
                 writer = csv.writer(file, lineterminator="\n")
@@ -90,14 +89,14 @@ def write_pdf_file(self, file_name: str ,file_path: str, content):
     def write_docx_file(self, file_name: str ,file_path: str, content):
         pass
     
-    def write_txt_file(self, file_name: str ,file_path: str, content):
+    def write_txt_file(self, file_name: str ,file_path: str, content) -> str:
         try:
             with open(file_path, mode="w") as file:
                 file.write(content)
                 file.close()
             self.write_to_s3(file_name, file_path)
             logger.info(f"{file_name} - File written successfully")
-            return f"{file_name} - File written successfully"
+            return file_path
         except Exception as err:
             return f"Error write_file: {err}"
     
@@ -146,17 +145,18 @@ def save_file_by_type(self, file_name: str, file_path: str, content):
         # Extract the file type from the file_name
         file_type = file_name.split('.')[-1].lower()
         
-        # Dictionary to map file extensions to corresponding functions
+        # Dictionary to map file types to corresponding functions
         file_type_handlers = {
             'txt': write_txt_file,
             'pdf': write_pdf_file,
             'docx': write_docx_file, 
-            'csv': write_csv_file
+            'csv': write_csv_file,
+            'html': write_txt_file
             # NOTE: Add more file types and corresponding functions as needed, These functions should be defined 
         }
         
         if file_path in file_type_handlers:
-            file_type_handlers[file_type](file_name, file_path, content)
+            return file_type_handlers[file_type](file_name, file_path, content)
         else:
             raise UnsupportedFileTypeError(f"Unsupported file type: {file_type}. Cannot save the file.")
         
\ No newline at end of file

From 0fdeff9ead9065a6a8494388d2fda99a5d863199 Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Fri, 28 Jul 2023 18:25:06 +0530
Subject: [PATCH 03/23] adding PDF + DOCX support to save images

---
 Dockerfile                                    |  3 +
 DockerfileCelery                              |  3 +
 requirements.txt                              |  2 +
 superagi/resource_manager/file_manager.py     | 66 +++++++++++++++----
 .../tools/file/prompts/add_images_to_html.txt |  4 ++
 .../file/prompts/content_to_html_prompt.txt   |  7 ++
 superagi/tools/file/write_file.py             | 60 ++++++++++++++++-
 .../stable_diffusion_image_gen.py             | 10 ++-
 8 files changed, 135 insertions(+), 20 deletions(-)
 create mode 100644 superagi/tools/file/prompts/add_images_to_html.txt
 create mode 100644 superagi/tools/file/prompts/content_to_html_prompt.txt

diff --git a/Dockerfile b/Dockerfile
index 775a0d5dc..9d0804851 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,6 +3,9 @@ WORKDIR /app
 COPY requirements.txt .
 
 #RUN apt-get update && apt-get install --no-install-recommends -y git wget libpq-dev gcc python3-dev && pip install psycopg2
+RUN apt-get update 
+RUN apt-get install -y wkhtmltopdf
+
 RUN pip install --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt
 
diff --git a/DockerfileCelery b/DockerfileCelery
index 0f8625679..a09c6160b 100644
--- a/DockerfileCelery
+++ b/DockerfileCelery
@@ -3,6 +3,9 @@ FROM python:3.9
 WORKDIR /app
 
 #RUN apt-get update && apt-get install --no-install-recommends -y git wget libpq-dev gcc python3-dev && pip install psycopg2
+RUN apt-get update 
+RUN apt-get install -y wkhtmltopdf
+
 RUN pip install --upgrade pip
 
 COPY requirements.txt .
diff --git a/requirements.txt b/requirements.txt
index f393a996d..c5a4cc8c3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -153,3 +153,5 @@ EbookLib==0.18
 html2text==2020.1.16
 duckduckgo-search==3.8.3 
 google-generativeai==0.1.0
+pdfkit==1.0.0
+htmldocx==0.0.6
diff --git a/superagi/resource_manager/file_manager.py b/superagi/resource_manager/file_manager.py
index 5ebf1215f..fe7c74558 100644
--- a/superagi/resource_manager/file_manager.py
+++ b/superagi/resource_manager/file_manager.py
@@ -7,10 +7,15 @@
 from superagi.models.agent import Agent
 from superagi.models.agent_execution import AgentExecution
 from superagi.types.storage_types import StorageType
+import pdfkit
+from htmldocx import HtmlToDocx
 
 class UnsupportedFileTypeError(Exception):
     pass
 
+class FileNotCreatedError(Exception):
+    pass
+
 class FileManager:
     def __init__(self, session: Session, agent_id: int = None, agent_execution_id: int = None):
         self.session = session
@@ -62,7 +67,8 @@ def write_file(self, file_name: str, content, return_file_path: bool = False):
             final_path = ResourceHelper.get_resource_path(file_name)
         
         try:
-            self.save_file_by_type(file_name=file_name, file_path=final_path, content=content)
+            res = self.save_file_by_type(file_name=file_name, file_path=final_path, content=content)
+            print(f"---------------{res}-------------")
         except Exception as err:
             return f"Error write_file: {err}"
         
@@ -81,24 +87,55 @@ def write_csv_file(self, file_name: str, final_path: str, csv_data) -> str:
             logger.info(f"{file_name} - File written successfully")
             return f"{file_name} - File written successfully"
         except Exception as err:
-            return f"Error write_csv_file: {err}"
+            raise FileNotCreatedError(f"Error write_file: {err}") from err
         
     def write_pdf_file(self, file_name: str ,file_path: str, content):
-        pass
-    
+        logger.info("Inside writing PDF")
+        # Saving the HTML file
+        html_file_path = f"{file_path[:-4]}.html"
+        self.write_txt_file(file_name=html_file_path.split('/')[-1], file_path=html_file_path, content=content)
+        
+        # Convert HTML file to a PDF file
+        try:
+            options = {
+                'quiet': '',
+                'page-size': 'Letter',
+                'margin-top': '0.75in',
+                'margin-right': '0.75in',
+                'margin-bottom': '0.75in',
+                'margin-left': '0.75in',
+                'enable-local-file-access': ''
+            }
+            config = pdfkit.configuration(wkhtmltopdf = "/usr/bin/wkhtmltopdf")
+            pdfkit.from_file(html_file_path, file_path, options = options, configuration = config)
+            return file_path
+        
+        except Exception as err:
+            raise FileNotCreatedError(f"Error write_file: {err}") from err
+            
     def write_docx_file(self, file_name: str ,file_path: str, content):
-        pass
-    
+        # Saving the HTML file
+        html_file_path = f"{file_path[:-4]}.html"
+        self.write_txt_file(file_name=html_file_path.split('/')[-1], file_path=html_file_path, content=content)
+
+        # Convert HTML file to a DOCx file
+        try:
+            new_parser = HtmlToDocx()
+            new_parser.parse_html_file(html_file_path, file_path)
+
+            return file_path
+        except Exception as err:
+            raise FileNotCreatedError(f"Error write_file: {err}") from err
+        
     def write_txt_file(self, file_name: str ,file_path: str, content) -> str:
         try:
             with open(file_path, mode="w") as file:
                 file.write(content)
                 file.close()
             self.write_to_s3(file_name, file_path)
-            logger.info(f"{file_name} - File written successfully")
             return file_path
         except Exception as err:
-            return f"Error write_file: {err}"
+            raise FileNotCreatedError(f"Error write_file: {err}") from err
     
     def get_agent_resource_path(self, file_name: str):
         return ResourceHelper.get_agent_write_resource_path(file_name, agent=Agent.get_agent_from_id(self.session,
@@ -147,15 +184,16 @@ def save_file_by_type(self, file_name: str, file_path: str, content):
         
         # Dictionary to map file types to corresponding functions
         file_type_handlers = {
-            'txt': write_txt_file,
-            'pdf': write_pdf_file,
-            'docx': write_docx_file, 
-            'csv': write_csv_file,
-            'html': write_txt_file
+            'txt': self.write_txt_file,
+            'pdf': self.write_pdf_file,
+            'docx': self.write_docx_file, 
+            'doc': self.write_docx_file,
+            'csv': self.write_csv_file,
+            'html': self.write_txt_file
             # NOTE: Add more file types and corresponding functions as needed, These functions should be defined 
         }
         
-        if file_path in file_type_handlers:
+        if file_type in file_type_handlers:
             return file_type_handlers[file_type](file_name, file_path, content)
         else:
             raise UnsupportedFileTypeError(f"Unsupported file type: {file_type}. Cannot save the file.")
diff --git a/superagi/tools/file/prompts/add_images_to_html.txt b/superagi/tools/file/prompts/add_images_to_html.txt
new file mode 100644
index 000000000..57dda0a49
--- /dev/null
+++ b/superagi/tools/file/prompts/add_images_to_html.txt
@@ -0,0 +1,4 @@
+Now, you will be provided with few image path locations. You will have to attach the following images in appropriate locations inside the html code.  
+Remember to maintain the elegancy and styling of the User Interface generated. Make sure you attach all the images provided to you.
+
+The relevant paths of the images are provided below:
diff --git a/superagi/tools/file/prompts/content_to_html_prompt.txt b/superagi/tools/file/prompts/content_to_html_prompt.txt
new file mode 100644
index 000000000..e450f46e4
--- /dev/null
+++ b/superagi/tools/file/prompts/content_to_html_prompt.txt
@@ -0,0 +1,7 @@
+You are an HTML code generating AI Agent. Your task is to generate a well formatted and well styled HTML file for a given content. 
+Remember to style the HTML beautifully, for which you can add the <style> element in the html itself. Make sure that the User Interface looks elegant to the user. 
+Make sure you have covered all of the content which is provided to you. 
+
+{embedding_image}.
+
+Content to be used:`{content}`
\ No newline at end of file
diff --git a/superagi/tools/file/write_file.py b/superagi/tools/file/write_file.py
index f8b61aeaf..3ebe93ff5 100644
--- a/superagi/tools/file/write_file.py
+++ b/superagi/tools/file/write_file.py
@@ -5,7 +5,9 @@
 # from superagi.helper.s3_helper import upload_to_s3
 from superagi.resource_manager.file_manager import FileManager
 from superagi.tools.base_tool import BaseTool
-
+from superagi.llms.base_llm import BaseLlm
+from superagi.helper.prompt_reader import PromptReader
+from superagi.tools.tool_response_query_manager import ToolResponseQueryManager
 
 # from superagi.helper.s3_helper import upload_to_s3
 
@@ -27,12 +29,14 @@ class WriteFileTool(BaseTool):
         args_schema : The args schema.
         resource_manager: File resource manager.
     """
+    llm: Optional[BaseLlm] = None
     name: str = "Write File"
     args_schema: Type[BaseModel] = WriteFileInput
     description: str = "Writes text to a file"
     agent_id: int = None
     resource_manager: Optional[FileManager] = None
-
+    tool_response_manager: Optional[ToolResponseQueryManager] = None
+    
     class Config:
         arbitrary_types_allowed = True
 
@@ -47,4 +51,54 @@ def _execute(self, file_name: str, content: str):
         Returns:
             success message if message is file written successfully or failure message if writing file fails.
         """
-        return self.resource_manager.write_file(file_name, content)
+        file_type = file_name.split('.')[-1].lower()
+        print(f"--------------{file_type}----------------")
+        if file_type not in ['pdf', 'docx', 'doc']:
+            return self.resource_manager.write_file(file_name, content)
+        
+        html_code_content = self._convert_content_into_html(content=content)
+        return self.resource_manager.write_file(file_name, html_code_content)
+
+    def _convert_content_into_html(self, content: str) -> str:
+        """
+        Converts the content into an HTML file
+        Args:
+            content (str): Content to be beautified and formatted
+        Returns:
+            HTML Code (str): HTML code of the formated content 
+        """
+        prompt = PromptReader.read_tools_prompt(__file__, "content_to_html_prompt.txt")
+        prompt = prompt.replace("{content}", content)
+
+        if image_file_paths := self._get_file_path_from_image_generation_tool():
+            embedding_image_prompt = PromptReader.read_tools_prompt(__file__, "add_images_to_html.txt")
+            for idx, image_path in enumerate(image_file_paths):
+                embedding_image_prompt += f"\n{idx+1}. {image_path}"
+            prompt = prompt.replace("{embedding_image}", embedding_image_prompt)
+        else:
+            prompt = prompt.replace("{embedding_image}", "")
+
+        messages = [{"role": "system", "content": prompt}]
+        result = self.llm.chat_completion(messages, max_tokens=self.max_token_limit)
+        return result['content']
+    
+    def _get_file_path_from_image_generation_tool(self) -> list:
+        """
+        Parses the output of the previous tool (Stable diffusion) and returns the path of the image file
+
+        Args:
+
+        Returns:
+            The path of the image file generated by the image generation toolkit
+        """
+        
+        last_tool_response = self.tool_response_manager.get_last_response()
+        file_path="resources"+last_tool_response.partition("['")[2].partition("']")[0]
+        image_file_paths = []
+
+        if ',' in file_path:
+            # Split the string based on the comma and get the first element (substring before the comma)
+            file_path = file_path.split(',')[0].strip()
+            image_file_paths = file_path[:-1]
+        print(image_file_paths)
+        return image_file_paths
\ No newline at end of file
diff --git a/superagi/tools/image_generation/stable_diffusion_image_gen.py b/superagi/tools/image_generation/stable_diffusion_image_gen.py
index 983d7ec5e..47641d308 100644
--- a/superagi/tools/image_generation/stable_diffusion_image_gen.py
+++ b/superagi/tools/image_generation/stable_diffusion_image_gen.py
@@ -8,6 +8,7 @@
 
 from superagi.resource_manager.file_manager import FileManager
 from superagi.tools.base_tool import BaseTool
+from superagi.helper.resource_helper import ResourceHelper
 
 
 class StableDiffusionImageGenInput(BaseModel):
@@ -55,7 +56,7 @@ def _execute(self, prompt: str, image_names: list, width: int = 512, height: int
         data = response.json()
 
         artifacts = data['artifacts']
-        base64_strings = []
+        base64_strings, image_paths = [],[]
         for artifact in artifacts:
             base64_strings.append(artifact['base64'])
 
@@ -68,8 +69,11 @@ def _execute(self, prompt: str, image_names: list, width: int = 512, height: int
             final_img.save(img_byte_arr, format=image_format)
 
             self.resource_manager.write_binary_file(image_names[i], img_byte_arr.getvalue())
-
-        return "Images downloaded and saved successfully"
+            
+        for image in image_names:
+            image_paths.append(ResourceHelper.get_resource_path(image))
+        return f"Images downloaded and saved successfully at the following locations: {image_paths}"
+        
 
     def call_stable_diffusion(self, api_key, width, height, num, prompt, steps):
         engine_id = self.get_tool_config("ENGINE_ID")

From a0ca1253821b2e3ac4bec4e25be92d3e217f4ebb Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Mon, 31 Jul 2023 00:56:36 +0530
Subject: [PATCH 04/23] Added Wkhtmltopdf package installation run commands in
 docker

---
 Dockerfile       | 4 ++++
 DockerfileCelery | 3 +--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 86f98bf95..92d6707d4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -7,6 +7,8 @@ RUN apt-get update && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
+RUN apt-get update && apt-get install -y wkhtmltopdf
+
 RUN python -m venv /opt/venv
 ENV PATH="/opt/venv/bin:$PATH"
 
@@ -26,6 +28,8 @@ RUN apt-get update && \
     apt-get install --no-install-recommends -y libpq-dev && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
+    
+RUN apt-get update && apt-get install -y wkhtmltopdf
 
 COPY --from=compile-image /opt/venv /opt/venv
 COPY --from=compile-image /app /app
diff --git a/DockerfileCelery b/DockerfileCelery
index 00b25c1ab..ba608dcbd 100644
--- a/DockerfileCelery
+++ b/DockerfileCelery
@@ -3,8 +3,7 @@ FROM python:3.9
 WORKDIR /app
 
 #RUN apt-get update && apt-get install --no-install-recommends -y git wget libpq-dev gcc python3-dev && pip install psycopg2
-RUN apt-get update 
-RUN apt-get install -y wkhtmltopdf
+RUN apt-get update && apt-get install -y wkhtmltopdf
 
 RUN pip install --upgrade pip
 

From cccd599bf3081781ac2d2cd66119a80a0a5751ee Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Mon, 31 Jul 2023 00:58:01 +0530
Subject: [PATCH 05/23] Added get_all_responses feature for extractng the
 response for particular tools

---
 superagi/models/agent_execution_feed.py       | 29 ++++++++++++++++++-
 superagi/tools/tool_response_query_manager.py |  3 ++
 2 files changed, 31 insertions(+), 1 deletion(-)

diff --git a/superagi/models/agent_execution_feed.py b/superagi/models/agent_execution_feed.py
index d6dbd8c8c..5f767754b 100644
--- a/superagi/models/agent_execution_feed.py
+++ b/superagi/models/agent_execution_feed.py
@@ -45,8 +45,35 @@ def get_last_tool_response(cls, session: Session, agent_execution_id: int, tool_
             AgentExecutionFeed.role == "system").order_by(AgentExecutionFeed.created_at.desc()).all()
 
         for agent_execution_feed in agent_execution_feeds:
-            if tool_name and not agent_execution_feed.feed.startswith("Tool " + tool_name):
+            if tool_name and not agent_execution_feed.feed.startswith(
+                f"Tool {tool_name}"
+            ):
                 continue
             if agent_execution_feed.feed.startswith("Tool"):
                 return agent_execution_feed.feed
         return ""
+    
+    @classmethod
+    def get_all_tools_response(cls, session: Session, agent_execution_id: int, tool_names: list = None):
+        """
+        Returns all system generated responses by the tools
+        Args:
+            session : Current Agent Session running
+            agent_execution_id (int) : Agent Execution id 
+            tool name (list) : Tool Name(s), whose responses are to be queried
+        Returns:
+            list: List of responses queried
+        """
+        agent_execution_feeds = session.query(AgentExecutionFeed).filter(
+            AgentExecutionFeed.agent_execution_id == agent_execution_id,
+            AgentExecutionFeed.role == "system").all()
+
+        return [
+            agent_execution_feed.feed
+            for agent_execution_feed in agent_execution_feeds
+            if tool_names
+            and any(
+                agent_execution_feed.feed.startswith(f"Tool {x}")
+                for x in tool_names
+            )
+        ]
\ No newline at end of file
diff --git a/superagi/tools/tool_response_query_manager.py b/superagi/tools/tool_response_query_manager.py
index 5a2387648..93a4f9180 100644
--- a/superagi/tools/tool_response_query_manager.py
+++ b/superagi/tools/tool_response_query_manager.py
@@ -10,3 +10,6 @@ def __init__(self, session: Session, agent_execution_id: int):
 
     def get_last_response(self, tool_name: str = None):
         return AgentExecutionFeed.get_last_tool_response(self.session, self.agent_execution_id, tool_name)
+    
+    def get_all_responses(self, tool_names: list = None):
+        return AgentExecutionFeed.get_all_tools_response(self.session, self.agent_execution_id, tool_names)

From 285a13248d440107c9060a57d6591686e764bfa5 Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Mon, 31 Jul 2023 00:59:10 +0530
Subject: [PATCH 06/23] Added Image embedding feature, this will extract and
 embed the images generated by the agent during the run

---
 superagi/resource_manager/file_manager.py |  1 -
 superagi/tools/file/write_file.py         | 61 ++++++++++++++++-------
 2 files changed, 43 insertions(+), 19 deletions(-)

diff --git a/superagi/resource_manager/file_manager.py b/superagi/resource_manager/file_manager.py
index 1c4add39e..1491f5f1b 100644
--- a/superagi/resource_manager/file_manager.py
+++ b/superagi/resource_manager/file_manager.py
@@ -90,7 +90,6 @@ def write_csv_file(self, file_name: str, final_path: str, csv_data) -> str:
             raise FileNotCreatedError(f"Error write_file: {err}") from err
         
     def write_pdf_file(self, file_name: str ,file_path: str, content):
-        logger.info("Inside writing PDF")
         # Saving the HTML file
         html_file_path = f"{file_path[:-4]}.html"
         self.write_txt_file(file_name=html_file_path.split('/')[-1], file_path=html_file_path, content=content)
diff --git a/superagi/tools/file/write_file.py b/superagi/tools/file/write_file.py
index 3ebe93ff5..05b896358 100644
--- a/superagi/tools/file/write_file.py
+++ b/superagi/tools/file/write_file.py
@@ -1,4 +1,5 @@
 from typing import Type, Optional
+import base64
 
 from pydantic import BaseModel, Field
 
@@ -32,7 +33,7 @@ class WriteFileTool(BaseTool):
     llm: Optional[BaseLlm] = None
     name: str = "Write File"
     args_schema: Type[BaseModel] = WriteFileInput
-    description: str = "Writes text to a file"
+    description: str = "Writes content in a file. The content can carry text and images."
     agent_id: int = None
     resource_manager: Optional[FileManager] = None
     tool_response_manager: Optional[ToolResponseQueryManager] = None
@@ -52,18 +53,19 @@ def _execute(self, file_name: str, content: str):
             success message if message is file written successfully or failure message if writing file fails.
         """
         file_type = file_name.split('.')[-1].lower()
-        print(f"--------------{file_type}----------------")
+        
         if file_type not in ['pdf', 'docx', 'doc']:
             return self.resource_manager.write_file(file_name, content)
         
-        html_code_content = self._convert_content_into_html(content=content)
+        html_code_content = self._convert_content_into_html(content=content, formated_for=file_type)
         return self.resource_manager.write_file(file_name, html_code_content)
 
-    def _convert_content_into_html(self, content: str) -> str:
+    def _convert_content_into_html(self, content: str, formated_for: str) -> str:
         """
         Converts the content into an HTML file
         Args:
             content (str): Content to be beautified and formatted
+            formated_for (str): HTML content to be specifically formatted for a specific document type
         Returns:
             HTML Code (str): HTML code of the formated content 
         """
@@ -77,28 +79,51 @@ def _convert_content_into_html(self, content: str) -> str:
             prompt = prompt.replace("{embedding_image}", embedding_image_prompt)
         else:
             prompt = prompt.replace("{embedding_image}", "")
-
+            
         messages = [{"role": "system", "content": prompt}]
         result = self.llm.chat_completion(messages, max_tokens=self.max_token_limit)
+        
+        if formated_for == 'pdf':
+            return self._html_formatting_for_pdf(content=result["content"], image_list=image_file_paths)
         return result['content']
     
     def _get_file_path_from_image_generation_tool(self) -> list:
         """
-        Parses the output of the previous tool (Stable diffusion) and returns the path of the image file
-
-        Args:
+        Parses the output of the Image Generation tool (Stable diffusion / Dalle) and returns the path of the image file
 
         Returns:
             The path of the image file generated by the image generation toolkit
         """
         
-        last_tool_response = self.tool_response_manager.get_last_response()
-        file_path="resources"+last_tool_response.partition("['")[2].partition("']")[0]
-        image_file_paths = []
-
-        if ',' in file_path:
-            # Split the string based on the comma and get the first element (substring before the comma)
-            file_path = file_path.split(',')[0].strip()
-            image_file_paths = file_path[:-1]
-        print(image_file_paths)
-        return image_file_paths
\ No newline at end of file
+        tool_responses = self.tool_response_manager.get_all_responses(tool_names=["Stable Diffusion", "Dalle"])
+    
+        all_image_paths = []
+        for tool_response in tool_responses:
+            # Iterating through all the tool responses to find the image paths in it. 
+            file_path = tool_response.partition("['")[2].partition("']")[0]
+            file_path = file_path.replace("'", "").replace('"', '').replace(" ","").strip()
+            print("file path after partition -> ", file_path)
+
+            file_path = file_path.split(',') if ',' in file_path else [file_path]
+            all_image_paths.extend(file_path)
+
+        return all_image_paths
+    
+    def _image_to_base64(self, image_path):
+        with open(image_path, "rb") as image_file:
+            encoded_image = base64.b64encode(image_file.read()).decode("utf-8")
+        return encoded_image
+    
+    def _html_formatting_for_pdf(self, content: str, image_list: list) -> str:
+        """
+        Converts image paths into base64 inputs in an HTML file
+        Args:
+            content (str): HTML code
+            image_list (list): List of images to be converted and replaced
+        Returns:
+            HTML Code (str): Formatted HTML code 
+        """
+        
+        for image_path in image_list:
+            content = content.replace(f"{image_path}", f"data:image/png;base64,{self._image_to_base64(image_path=image_path)}")
+        return content
\ No newline at end of file

From 74c938c00f86928b324eefad19700c5db2f215b9 Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Thu, 10 Aug 2023 13:02:27 +0530
Subject: [PATCH 07/23] renaming functions and refactoring

---
 superagi/models/agent_execution_feed.py                      | 2 +-
 superagi/tools/file/write_file.py                            | 2 +-
 .../tools/image_generation/stable_diffusion_image_gen.py     | 5 +----
 superagi/tools/tool_response_query_manager.py                | 4 ++--
 4 files changed, 5 insertions(+), 8 deletions(-)

diff --git a/superagi/models/agent_execution_feed.py b/superagi/models/agent_execution_feed.py
index 5f767754b..f981a9f2c 100644
--- a/superagi/models/agent_execution_feed.py
+++ b/superagi/models/agent_execution_feed.py
@@ -54,7 +54,7 @@ def get_last_tool_response(cls, session: Session, agent_execution_id: int, tool_
         return ""
     
     @classmethod
-    def get_all_tools_response(cls, session: Session, agent_execution_id: int, tool_names: list = None):
+    def get_tools_response(cls, session: Session, agent_execution_id: int, tool_names: list = None):
         """
         Returns all system generated responses by the tools
         Args:
diff --git a/superagi/tools/file/write_file.py b/superagi/tools/file/write_file.py
index 05b896358..62e47f6ae 100644
--- a/superagi/tools/file/write_file.py
+++ b/superagi/tools/file/write_file.py
@@ -95,7 +95,7 @@ def _get_file_path_from_image_generation_tool(self) -> list:
             The path of the image file generated by the image generation toolkit
         """
         
-        tool_responses = self.tool_response_manager.get_all_responses(tool_names=["Stable Diffusion", "Dalle"])
+        tool_responses = self.tool_response_manager.get_responses(tool_names=["Stable Diffusion", "Dalle"])
     
         all_image_paths = []
         for tool_response in tool_responses:
diff --git a/superagi/tools/image_generation/stable_diffusion_image_gen.py b/superagi/tools/image_generation/stable_diffusion_image_gen.py
index fce54c51a..e8bb6f1a7 100644
--- a/superagi/tools/image_generation/stable_diffusion_image_gen.py
+++ b/superagi/tools/image_generation/stable_diffusion_image_gen.py
@@ -57,10 +57,7 @@ def _execute(self, prompt: str, image_names: list, width: int = 512, height: int
         data = response.json()
 
         artifacts = data['artifacts']
-        base64_strings, image_paths = [],[]
-        for artifact in artifacts:
-            base64_strings.append(artifact['base64'])
-
+        base64_strings = [artifact['base64'] for artifact in artifacts]
         image_paths=[]
         for i in range(num):
             image_base64 = base64_strings[i]
diff --git a/superagi/tools/tool_response_query_manager.py b/superagi/tools/tool_response_query_manager.py
index 93a4f9180..b327db39c 100644
--- a/superagi/tools/tool_response_query_manager.py
+++ b/superagi/tools/tool_response_query_manager.py
@@ -11,5 +11,5 @@ def __init__(self, session: Session, agent_execution_id: int):
     def get_last_response(self, tool_name: str = None):
         return AgentExecutionFeed.get_last_tool_response(self.session, self.agent_execution_id, tool_name)
     
-    def get_all_responses(self, tool_names: list = None):
-        return AgentExecutionFeed.get_all_tools_response(self.session, self.agent_execution_id, tool_names)
+    def get_responses(self, tool_names: list = None):
+        return AgentExecutionFeed.get_tools_response(self.session, self.agent_execution_id, tool_names)

From 1c4bbba89a4d4535eb6ede39a3b37f2ed2408a2a Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Thu, 10 Aug 2023 13:02:27 +0530
Subject: [PATCH 08/23] renaming functions and refactoring

---
 superagi/models/agent_execution_feed.py       | 15 ++++
 superagi/tools/file/write_file.py             | 80 ++++++++++++++++++-
 .../stable_diffusion_image_gen.py             |  6 +-
 superagi/tools/tool_response_query_manager.py |  3 +
 4 files changed, 99 insertions(+), 5 deletions(-)

diff --git a/superagi/models/agent_execution_feed.py b/superagi/models/agent_execution_feed.py
index 356f6cfc7..269260998 100644
--- a/superagi/models/agent_execution_feed.py
+++ b/superagi/models/agent_execution_feed.py
@@ -52,6 +52,21 @@ def get_last_tool_response(cls, session: Session, agent_execution_id: int, tool_
             if agent_execution_feed.feed.startswith("Tool"):
                 return agent_execution_feed.feed
         return ""
+    
+    @classmethod
+    def get_tools_response(cls, session: Session, agent_execution_id: int, tool_names: list = None):
+        """
+        Returns all system generated responses by the tools
+        Args:
+            session : Current Agent Session running
+            agent_execution_id (int) : Agent Execution id 
+            tool name (list) : Tool Name(s), whose responses are to be queried
+        Returns:
+            list: List of responses queried
+        """
+        agent_execution_feeds = session.query(AgentExecutionFeed).filter(
+            AgentExecutionFeed.agent_execution_id == agent_execution_id,
+            AgentExecutionFeed.role == "system").all()
 
     @classmethod
     def fetch_agent_execution_feeds(cls, session, agent_execution_id: int):
diff --git a/superagi/tools/file/write_file.py b/superagi/tools/file/write_file.py
index f8b61aeaf..548f83573 100644
--- a/superagi/tools/file/write_file.py
+++ b/superagi/tools/file/write_file.py
@@ -1,10 +1,14 @@
 from typing import Type, Optional
+import base64
 
 from pydantic import BaseModel, Field
 
 # from superagi.helper.s3_helper import upload_to_s3
 from superagi.resource_manager.file_manager import FileManager
 from superagi.tools.base_tool import BaseTool
+from superagi.llms.base_llm import BaseLlm
+from superagi.helper.prompt_reader import PromptReader
+from superagi.tools.tool_response_query_manager import ToolResponseQueryManager
 
 
 # from superagi.helper.s3_helper import upload_to_s3
@@ -47,4 +51,78 @@ def _execute(self, file_name: str, content: str):
         Returns:
             success message if message is file written successfully or failure message if writing file fails.
         """
-        return self.resource_manager.write_file(file_name, content)
+        file_type = file_name.split('.')[-1].lower()
+        
+        if file_type not in ['pdf', 'docx', 'doc']:
+            return self.resource_manager.write_file(file_name, content)
+        
+        html_code_content = self._convert_content_into_html(content=content, formated_for=file_type)
+        return self.resource_manager.write_file(file_name, html_code_content)
+
+    def _convert_content_into_html(self, content: str, formated_for: str) -> str:
+        """
+        Converts the content into an HTML file
+        Args:
+            content (str): Content to be beautified and formatted
+            formated_for (str): HTML content to be specifically formatted for a specific document type
+        Returns:
+            HTML Code (str): HTML code of the formated content 
+        """
+        prompt = PromptReader.read_tools_prompt(__file__, "content_to_html_prompt.txt")
+        prompt = prompt.replace("{content}", content)
+
+        if image_file_paths := self._get_file_path_from_image_generation_tool():
+            embedding_image_prompt = PromptReader.read_tools_prompt(__file__, "add_images_to_html.txt")
+            for idx, image_path in enumerate(image_file_paths):
+                embedding_image_prompt += f"\n{idx+1}. {image_path}"
+            prompt = prompt.replace("{embedding_image}", embedding_image_prompt)
+        else:
+            prompt = prompt.replace("{embedding_image}", "")
+            
+        messages = [{"role": "system", "content": prompt}]
+        result = self.llm.chat_completion(messages, max_tokens=self.max_token_limit)
+        
+        if formated_for == 'pdf':
+            return self._html_formatting_for_pdf(content=result["content"], image_list=image_file_paths)
+        return result['content']
+    
+    def _get_file_path_from_image_generation_tool(self) -> list:
+        """
+        Parses the output of the Image Generation tool (Stable diffusion / Dalle) and returns the path of the image file
+
+        Returns:
+            The path of the image file generated by the image generation toolkit
+        """
+        
+        tool_responses = self.tool_response_manager.get_responses(tool_names=["Stable Diffusion", "Dalle"])
+    
+        all_image_paths = []
+        for tool_response in tool_responses:
+            # Iterating through all the tool responses to find the image paths in it. 
+            file_path = tool_response.partition("['")[2].partition("']")[0]
+            file_path = file_path.replace("'", "").replace('"', '').replace(" ","").strip()
+            print("file path after partition -> ", file_path)
+
+            file_path = file_path.split(',') if ',' in file_path else [file_path]
+            all_image_paths.extend(file_path)
+
+        return all_image_paths
+    
+    def _image_to_base64(self, image_path):
+        with open(image_path, "rb") as image_file:
+            encoded_image = base64.b64encode(image_file.read()).decode("utf-8")
+        return encoded_image
+    
+    def _html_formatting_for_pdf(self, content: str, image_list: list) -> str:
+        """
+        Converts image paths into base64 inputs in an HTML file
+        Args:
+            content (str): HTML code
+            image_list (list): List of images to be converted and replaced
+        Returns:
+            HTML Code (str): Formatted HTML code 
+        """
+        
+        for image_path in image_list:
+            content = content.replace(f"{image_path}", f"data:image/png;base64,{self._image_to_base64(image_path=image_path)}")
+        return content
diff --git a/superagi/tools/image_generation/stable_diffusion_image_gen.py b/superagi/tools/image_generation/stable_diffusion_image_gen.py
index b3c5d2c96..8a71f7c42 100644
--- a/superagi/tools/image_generation/stable_diffusion_image_gen.py
+++ b/superagi/tools/image_generation/stable_diffusion_image_gen.py
@@ -57,10 +57,8 @@ def _execute(self, prompt: str, image_names: list, width: int = 512, height: int
         data = response.json()
 
         artifacts = data['artifacts']
-        base64_strings = []
-        for artifact in artifacts:
-            base64_strings.append(artifact['base64'])
-
+        base64_strings = [artifact['base64'] for artifact in artifacts]
+        image_paths=[]
         for i in range(num):
             image_base64 = base64_strings[i]
             img_data = base64.b64decode(image_base64)
diff --git a/superagi/tools/tool_response_query_manager.py b/superagi/tools/tool_response_query_manager.py
index 5a2387648..b327db39c 100644
--- a/superagi/tools/tool_response_query_manager.py
+++ b/superagi/tools/tool_response_query_manager.py
@@ -10,3 +10,6 @@ def __init__(self, session: Session, agent_execution_id: int):
 
     def get_last_response(self, tool_name: str = None):
         return AgentExecutionFeed.get_last_tool_response(self.session, self.agent_execution_id, tool_name)
+    
+    def get_responses(self, tool_names: list = None):
+        return AgentExecutionFeed.get_tools_response(self.session, self.agent_execution_id, tool_names)

From 580317106f96dce69aceba936d1ec5a0c5f6b10a Mon Sep 17 00:00:00 2001
From: Fluder-Paradyne <121793617+Fluder-Paradyne@users.noreply.github.com>
Date: Thu, 10 Aug 2023 13:28:11 +0530
Subject: [PATCH 09/23] Update Dockerfile

---
 Dockerfile | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 92d6707d4..77bc5cd7d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,12 +3,10 @@ FROM python:3.10-slim-bullseye AS compile-image
 WORKDIR /app
 
 RUN apt-get update && \
-    apt-get install --no-install-recommends -y wget libpq-dev gcc g++ python3-dev && \
+    apt-get install --no-install-recommends -y wget libpq-dev gcc g++ python3-dev wkhtmltopdf && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
-RUN apt-get update && apt-get install -y wkhtmltopdf
-
 RUN python -m venv /opt/venv
 ENV PATH="/opt/venv/bin:$PATH"
 
@@ -25,11 +23,9 @@ FROM python:3.10-slim-bullseye AS build-image
 WORKDIR /app
 
 RUN apt-get update && \
-    apt-get install --no-install-recommends -y libpq-dev && \
+    apt-get install --no-install-recommends -y libpq-dev wkhtmltopdf && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
-    
-RUN apt-get update && apt-get install -y wkhtmltopdf
 
 COPY --from=compile-image /opt/venv /opt/venv
 COPY --from=compile-image /app /app

From 31b20a68371ac3439044a35ad5b4713d4c470cbe Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Sun, 20 Aug 2023 14:10:18 +0530
Subject: [PATCH 10/23] removing unsused classmethods

---
 superagi/models/agent_execution_feed.py       | 24 -------------------
 superagi/tools/tool_response_query_manager.py |  5 +---
 2 files changed, 1 insertion(+), 28 deletions(-)

diff --git a/superagi/models/agent_execution_feed.py b/superagi/models/agent_execution_feed.py
index e19e3a56c..e3f208036 100644
--- a/superagi/models/agent_execution_feed.py
+++ b/superagi/models/agent_execution_feed.py
@@ -55,30 +55,6 @@ def get_last_tool_response(cls, session: Session, agent_execution_id: int, tool_
                 return agent_execution_feed.feed
         return ""
     
-    @classmethod
-    def get_tools_response(cls, session: Session, agent_execution_id: int, tool_names: list = None):
-        """
-        Returns all system generated responses by the tools
-        Args:
-            session : Current Agent Session running
-            agent_execution_id (int) : Agent Execution id 
-            tool name (list) : Tool Name(s), whose responses are to be queried
-        Returns:
-            list: List of responses queried
-        """
-        agent_execution_feeds = session.query(AgentExecutionFeed).filter(
-            AgentExecutionFeed.agent_execution_id == agent_execution_id,
-            AgentExecutionFeed.role == "system").all()
-        return [
-            agent_execution_feed.feed
-            for agent_execution_feed in agent_execution_feeds
-            if tool_names
-            and any(
-                agent_execution_feed.feed.startswith(f"Tool {x}")
-                for x in tool_names
-            )
-        ]
-        
     @classmethod
     def fetch_agent_execution_feeds(cls, session, agent_execution_id: int):
         agent_execution = AgentExecution.find_by_id(session, agent_execution_id)
diff --git a/superagi/tools/tool_response_query_manager.py b/superagi/tools/tool_response_query_manager.py
index b327db39c..218ff53bb 100644
--- a/superagi/tools/tool_response_query_manager.py
+++ b/superagi/tools/tool_response_query_manager.py
@@ -9,7 +9,4 @@ def __init__(self, session: Session, agent_execution_id: int):
         self.agent_execution_id = agent_execution_id
 
     def get_last_response(self, tool_name: str = None):
-        return AgentExecutionFeed.get_last_tool_response(self.session, self.agent_execution_id, tool_name)
-    
-    def get_responses(self, tool_names: list = None):
-        return AgentExecutionFeed.get_tools_response(self.session, self.agent_execution_id, tool_names)
+        return AgentExecutionFeed.get_last_tool_response(self.session, self.agent_execution_id, tool_name)
\ No newline at end of file

From 9e2b4d6564fa284c4187da0110dab191367eeeeb Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Sun, 20 Aug 2023 14:12:32 +0530
Subject: [PATCH 11/23] Finding generated images and attached files in the
 write tool. Images are fetched inorder to be embedded in the respective file
 type.

---
 superagi/tools/file/write_file.py | 105 ++++++++++++++++++++++--------
 1 file changed, 79 insertions(+), 26 deletions(-)

diff --git a/superagi/tools/file/write_file.py b/superagi/tools/file/write_file.py
index e12d0ee75..4cc5c3cc3 100644
--- a/superagi/tools/file/write_file.py
+++ b/superagi/tools/file/write_file.py
@@ -1,16 +1,17 @@
 from typing import Type, Optional
 import base64
+import os
 
 from pydantic import BaseModel, Field
 
-# from superagi.helper.s3_helper import upload_to_s3
+from superagi.helper.prompt_reader import PromptReader
+from superagi.helper.resource_helper import ResourceHelper
 from superagi.resource_manager.file_manager import FileManager
 from superagi.tools.base_tool import BaseTool
-from superagi.llms.base_llm import BaseLlm
-from superagi.helper.prompt_reader import PromptReader
 from superagi.tools.tool_response_query_manager import ToolResponseQueryManager
-
-# from superagi.helper.s3_helper import upload_to_s3
+from superagi.llms.base_llm import BaseLlm
+from superagi.models.agent import Agent
+from superagi.models.agent_execution import AgentExecution
 
 
 class WriteFileInput(BaseModel):
@@ -18,7 +19,6 @@ class WriteFileInput(BaseModel):
     file_name: str = Field(..., description="Name of the file to write. Only include the file name. Don't include path.")
     content: str = Field(..., description="File content to write")
 
-
 class WriteFileTool(BaseTool):
     """
     Write File tool
@@ -34,10 +34,9 @@ class WriteFileTool(BaseTool):
     name: str = "Write File"
     args_schema: Type[BaseModel] = WriteFileInput
     description: str = "Writes content in a file. The content can carry text and images."
-    agent_id: int = None
     resource_manager: Optional[FileManager] = None
-    tool_response_manager: Optional[ToolResponseQueryManager] = None
-    
+    agent_id:int =None
+    agent_execution_id:int =None
     class Config:
         arbitrary_types_allowed = True
 
@@ -52,15 +51,23 @@ def _execute(self, file_name: str, content: str):
         Returns:
             success message if message is file written successfully or failure message if writing file fails.
         """
+        print("write init -->", file_name)
+        
+        attached_files = self._get_attached_files()
         file_type = file_name.split('.')[-1].lower()
         
+        print("Attached Files --> ", attached_files)
+        
         if file_type not in ['pdf', 'docx', 'doc']:
             return self.resource_manager.write_file(file_name, content)
         
-        html_code_content = self._convert_content_into_html(content=content, formated_for=file_type)
+        html_code_content = self._convert_content_into_html(content=content, attached_files = attached_files,formated_for=file_type)
+        
+        print("html code -->", html_code_content)
+        
         return self.resource_manager.write_file(file_name, html_code_content)
 
-    def _convert_content_into_html(self, content: str, formated_for: str) -> str:
+    def _convert_content_into_html(self, content: str, attached_files: list, formated_for: str) -> str:
         """
         Converts the content into an HTML file
         Args:
@@ -72,7 +79,7 @@ def _convert_content_into_html(self, content: str, formated_for: str) -> str:
         prompt = PromptReader.read_tools_prompt(__file__, "content_to_html_prompt.txt")
         prompt = prompt.replace("{content}", content)
 
-        if image_file_paths := self._get_file_path_from_image_generation_tool():
+        if image_file_paths := self._get_file_path_of_images(attached_files=attached_files):
             embedding_image_prompt = PromptReader.read_tools_prompt(__file__, "add_images_to_html.txt")
             for idx, image_path in enumerate(image_file_paths):
                 embedding_image_prompt += f"\n{idx+1}. {image_path}"
@@ -87,27 +94,47 @@ def _convert_content_into_html(self, content: str, formated_for: str) -> str:
             return self._html_formatting_for_pdf(content=result["content"], image_list=image_file_paths)
         return result['content']
     
-    def _get_file_path_from_image_generation_tool(self) -> list:
+    def _get_file_path_of_images(self, attached_files: list):
         """
-        Parses the output of the Image Generation tool (Stable diffusion / Dalle) and returns the path of the image file
+        Filters Images from the attached files list and finds out the corresponding paths. 
+
+        Args:
+            attached_files: List of names of files generated
 
         Returns:
-            The path of the image file generated by the image generation toolkit
+            Full paths of the image files
         """
-        
-        tool_responses = self.tool_response_manager.get_responses(tool_names=["Stable Diffusion", "Dalle"])
+        image_extensions, image_paths = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp'],[]
+        for file_name in attached_files:
+            path = self._get_file_path(file_name=file_name)
+            if not os.path.exists(path):
+                continue
+            _, file_extension = os.path.splitext(path)
+            # Check if the file extension is in the list of image extensions
+            if file_extension.lower() in image_extensions:
+                image_paths.append(path)
+        print("Image Paths", image_paths)
+        return image_paths
     
-        all_image_paths = []
-        for tool_response in tool_responses:
-            # Iterating through all the tool responses to find the image paths in it. 
-            file_path = tool_response.partition("['")[2].partition("']")[0]
-            file_path = file_path.replace("'", "").replace('"', '').replace(" ","").strip()
-            print("file path after partition -> ", file_path)
+    def _get_file_path(self, file_name: str) -> list:
+        """
+        Gets the path of the file
 
-            file_path = file_path.split(',') if ',' in file_path else [file_path]
-            all_image_paths.extend(file_path)
+        Args:
+            file_name: Name of file 
 
-        return all_image_paths
+        Returns:
+            Full paths of the file
+        """
+        session = self.toolkit_config.session
+
+        return ResourceHelper().get_agent_read_resource_path(
+            file_name,
+            agent=Agent.get_agent_from_id(session, self.agent_id),
+            agent_execution=AgentExecution.get_agent_execution_from_id(
+                session, self.agent_execution_id
+            ),
+        )
     
     def _image_to_base64(self, image_path):
         with open(image_path, "rb") as image_file:
@@ -127,3 +154,29 @@ def _html_formatting_for_pdf(self, content: str, image_list: list) -> str:
         for image_path in image_list:
             content = content.replace(f"{image_path}", f"data:image/png;base64,{self._image_to_base64(image_path=image_path)}")
         return content
+
+    def _get_attached_files(self):
+        output_directory = ResourceHelper.get_root_output_dir()
+        if "{agent_id}" not in output_directory:
+            return []
+        output_directory = ResourceHelper.get_formatted_agent_level_path(agent=Agent
+                                                                        .get_agent_from_id(session=self
+                                                                                           .toolkit_config.session,
+                                                                                           agent_id=self.agent_id),
+                                                                        path=output_directory)
+        agent_execution=AgentExecution.get_agent_execution_from_id(session=self.toolkit_config.session, agent_execution_id=self.agent_execution_id)
+        if agent_execution is not None and "{agent_execution_id}" in output_directory:
+            output_directory = ResourceHelper.get_formatted_agent_execution_level_path(agent_execution=agent_execution, path=output_directory)
+
+        print("Output directory --> ", output_directory)
+        return self._list_files(output_directory)
+    
+    def _list_files(self, directory):
+        found_files = []
+        for root, dirs, files in os.walk(directory):
+            found_files.extend(
+                file
+                for file in files
+                if not file.startswith(".") and "__pycache__" not in root
+            )
+        return found_files
\ No newline at end of file

From a24df274fa2020689a700720d58be2a08b511d8a Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Wed, 23 Aug 2023 19:56:22 +0530
Subject: [PATCH 12/23] Adding the filename and paths to the Resource Manager
 S3 storage

---
 superagi/resource_manager/file_manager.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/superagi/resource_manager/file_manager.py b/superagi/resource_manager/file_manager.py
index 1491f5f1b..93bb52fd7 100644
--- a/superagi/resource_manager/file_manager.py
+++ b/superagi/resource_manager/file_manager.py
@@ -68,8 +68,7 @@ def write_file(self, file_name: str, content, return_file_path: bool = False):
             final_path = ResourceHelper.get_resource_path(file_name)
         
         try:
-            res = self.save_file_by_type(file_name=file_name, file_path=final_path, content=content)
-            print(f"---------------{res}-------------")
+            self.save_file_by_type(file_name=file_name, file_path=final_path, content=content)
         except Exception as err:
             return f"Error write_file: {err}"
         
@@ -107,6 +106,7 @@ def write_pdf_file(self, file_name: str ,file_path: str, content):
             }
             config = pdfkit.configuration(wkhtmltopdf = "/usr/bin/wkhtmltopdf")
             pdfkit.from_file(html_file_path, file_path, options = options, configuration = config)
+            self.write_to_s3(file_name, file_path)
             return file_path
         
         except Exception as err:
@@ -121,7 +121,7 @@ def write_docx_file(self, file_name: str ,file_path: str, content):
         try:
             new_parser = HtmlToDocx()
             new_parser.parse_html_file(html_file_path, file_path)
-
+            self.write_to_s3(file_name, file_path)
             return file_path
         except Exception as err:
             raise FileNotCreatedError(f"Error write_file: {err}") from err

From 0dc6fa60311b24c76adffd7c0424f242d77a8969 Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Wed, 23 Aug 2023 19:56:34 +0530
Subject: [PATCH 13/23] Code Cleanup

---
 superagi/tools/file/write_file.py | 7 -------
 1 file changed, 7 deletions(-)

diff --git a/superagi/tools/file/write_file.py b/superagi/tools/file/write_file.py
index 4cc5c3cc3..4fcc3d75f 100644
--- a/superagi/tools/file/write_file.py
+++ b/superagi/tools/file/write_file.py
@@ -51,20 +51,15 @@ def _execute(self, file_name: str, content: str):
         Returns:
             success message if message is file written successfully or failure message if writing file fails.
         """
-        print("write init -->", file_name)
         
         attached_files = self._get_attached_files()
         file_type = file_name.split('.')[-1].lower()
         
-        print("Attached Files --> ", attached_files)
-        
         if file_type not in ['pdf', 'docx', 'doc']:
             return self.resource_manager.write_file(file_name, content)
         
         html_code_content = self._convert_content_into_html(content=content, attached_files = attached_files,formated_for=file_type)
         
-        print("html code -->", html_code_content)
-        
         return self.resource_manager.write_file(file_name, html_code_content)
 
     def _convert_content_into_html(self, content: str, attached_files: list, formated_for: str) -> str:
@@ -113,7 +108,6 @@ def _get_file_path_of_images(self, attached_files: list):
             # Check if the file extension is in the list of image extensions
             if file_extension.lower() in image_extensions:
                 image_paths.append(path)
-        print("Image Paths", image_paths)
         return image_paths
     
     def _get_file_path(self, file_name: str) -> list:
@@ -168,7 +162,6 @@ def _get_attached_files(self):
         if agent_execution is not None and "{agent_execution_id}" in output_directory:
             output_directory = ResourceHelper.get_formatted_agent_execution_level_path(agent_execution=agent_execution, path=output_directory)
 
-        print("Output directory --> ", output_directory)
         return self._list_files(output_directory)
     
     def _list_files(self, directory):

From 3c0a5782a7753ca3a25e9d7867e40e0c05f1d7f3 Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Thu, 24 Aug 2023 19:13:37 +0530
Subject: [PATCH 14/23] added logger: Fix for the failing TEST

---
 superagi/resource_manager/file_manager.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/superagi/resource_manager/file_manager.py b/superagi/resource_manager/file_manager.py
index 93bb52fd7..dbd0e4d16 100644
--- a/superagi/resource_manager/file_manager.py
+++ b/superagi/resource_manager/file_manager.py
@@ -66,16 +66,17 @@ def write_file(self, file_name: str, content, return_file_path: bool = False):
                                                                                                    self.agent_execution_id))
         else:
             final_path = ResourceHelper.get_resource_path(file_name)
-        
+
         try:
             self.save_file_by_type(file_name=file_name, file_path=final_path, content=content)
         except Exception as err:
             return f"Error write_file: {err}"
+
+        logger.info(f"{file_name} - File written successfully")
         
         if return_file_path:
             return final_path
-        else:
-            return f"{file_name} - File written successfully"
+        return f"{file_name} - File written successfully"
         
     def write_csv_file(self, file_name: str, final_path: str, csv_data) -> str:
         try:

From 4993251bb3a2f8e514c9da7c69d8492e65962711 Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Sat, 26 Aug 2023 16:03:08 +0530
Subject: [PATCH 15/23] Exceptions dir created to store custom exceptions

---
 superagi/exceptions/__init__.py    |  0
 superagi/exceptions/file_errors.py | 10 ++++++++++
 2 files changed, 10 insertions(+)
 create mode 100644 superagi/exceptions/__init__.py
 create mode 100644 superagi/exceptions/file_errors.py

diff --git a/superagi/exceptions/__init__.py b/superagi/exceptions/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/superagi/exceptions/file_errors.py b/superagi/exceptions/file_errors.py
new file mode 100644
index 000000000..354135196
--- /dev/null
+++ b/superagi/exceptions/file_errors.py
@@ -0,0 +1,10 @@
+
+class UnsupportedFileTypeError(Exception):
+    def __init__(self, file_name: str, supported_types: list):
+        message = f"Unsupported file type for '{file_name}'. Supported types are: {', '.join(supported_types)}"
+        super().__init__(message)
+
+class FileNotCreatedError(Exception):
+    def __init__(self, file_name: str):
+        message = f"Failed to create the file '{file_name}'."
+        super().__init__(message)
\ No newline at end of file

From 677c28e01bfffdb20a8971e189f138ab308d3c81 Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Sat, 26 Aug 2023 16:04:09 +0530
Subject: [PATCH 16/23] review changes

---
 superagi/models/agent_execution_feed.py   |  4 +--
 superagi/resource_manager/file_manager.py | 30 +++++++++--------------
 superagi/tools/file/write_file.py         | 27 ++++++--------------
 3 files changed, 19 insertions(+), 42 deletions(-)

diff --git a/superagi/models/agent_execution_feed.py b/superagi/models/agent_execution_feed.py
index aa7c3a148..00f50ee60 100644
--- a/superagi/models/agent_execution_feed.py
+++ b/superagi/models/agent_execution_feed.py
@@ -47,9 +47,7 @@ def get_last_tool_response(cls, session: Session, agent_execution_id: int, tool_
             AgentExecutionFeed.role == "system").order_by(AgentExecutionFeed.created_at.desc()).all()
 
         for agent_execution_feed in agent_execution_feeds:
-            if tool_name and not agent_execution_feed.feed.startswith(
-                f"Tool {tool_name}"
-            ):
+            if tool_name and not agent_execution_feed.feed.startswith("Tool " + tool_name):
                 continue
             if agent_execution_feed.feed.startswith("Tool"):
                 return agent_execution_feed.feed
diff --git a/superagi/resource_manager/file_manager.py b/superagi/resource_manager/file_manager.py
index dbd0e4d16..93b6396e7 100644
--- a/superagi/resource_manager/file_manager.py
+++ b/superagi/resource_manager/file_manager.py
@@ -1,22 +1,19 @@
 import csv
 from sqlalchemy.orm import Session
-from superagi.config.config import get_config
 import os
+
+from superagi.config.config import get_config
 from superagi.helper.resource_helper import ResourceHelper
 from superagi.helper.s3_helper import S3Helper
 from superagi.lib.logger import logger
 from superagi.models.agent import Agent
 from superagi.models.agent_execution import AgentExecution
 from superagi.types.storage_types import StorageType
+from superagi.exceptions.file_errors import UnsupportedFileTypeError, FileNotCreatedError
+
 import pdfkit
 from htmldocx import HtmlToDocx
 
-class UnsupportedFileTypeError(Exception):
-    pass
-
-class FileNotCreatedError(Exception):
-    pass
-
 class FileManager:
     def __init__(self, session: Session, agent_id: int = None, agent_execution_id: int = None):
         self.session = session
@@ -56,7 +53,7 @@ def write_to_s3(self, file_name, final_path):
                 s3_helper = S3Helper()
                 s3_helper.upload_file(img, path=resource.path)
 
-    def write_file(self, file_name: str, content, return_file_path: bool = False):
+    def write_file(self, file_name: str, content):
         if self.agent_id is not None:
             final_path = ResourceHelper.get_agent_write_resource_path(file_name,
                                                                       agent=Agent.get_agent_from_id(self.session,
@@ -73,9 +70,6 @@ def write_file(self, file_name: str, content, return_file_path: bool = False):
             return f"Error write_file: {err}"
 
         logger.info(f"{file_name} - File written successfully")
-        
-        if return_file_path:
-            return final_path
         return f"{file_name} - File written successfully"
         
     def write_csv_file(self, file_name: str, final_path: str, csv_data) -> str:
@@ -87,7 +81,7 @@ def write_csv_file(self, file_name: str, final_path: str, csv_data) -> str:
             logger.info(f"{file_name} - File written successfully")
             return f"{file_name} - File written successfully"
         except Exception as err:
-            raise FileNotCreatedError(f"Error write_file: {err}") from err
+            raise FileNotCreatedError(file_name=file_name) from err
         
     def write_pdf_file(self, file_name: str ,file_path: str, content):
         # Saving the HTML file
@@ -111,7 +105,7 @@ def write_pdf_file(self, file_name: str ,file_path: str, content):
             return file_path
         
         except Exception as err:
-            raise FileNotCreatedError(f"Error write_file: {err}") from err
+            raise FileNotCreatedError(file_name=file_name) from err
             
     def write_docx_file(self, file_name: str ,file_path: str, content):
         # Saving the HTML file
@@ -125,7 +119,7 @@ def write_docx_file(self, file_name: str ,file_path: str, content):
             self.write_to_s3(file_name, file_path)
             return file_path
         except Exception as err:
-            raise FileNotCreatedError(f"Error write_file: {err}") from err
+            raise FileNotCreatedError(file_name=file_name) from err
         
     def write_txt_file(self, file_name: str ,file_path: str, content) -> str:
         try:
@@ -135,7 +129,7 @@ def write_txt_file(self, file_name: str ,file_path: str, content) -> str:
             self.write_to_s3(file_name, file_path)
             return file_path
         except Exception as err:
-            raise FileNotCreatedError(f"Error write_file: {err}") from err
+            raise FileNotCreatedError(file_name=file_name) from err
     
     def get_agent_resource_path(self, file_name: str):
         return ResourceHelper.get_agent_write_resource_path(file_name, agent=Agent.get_agent_from_id(self.session,
@@ -193,8 +187,6 @@ def save_file_by_type(self, file_name: str, file_path: str, content):
             # NOTE: Add more file types and corresponding functions as needed, These functions should be defined 
         }
         
-        if file_type in file_type_handlers:
-            return file_type_handlers[file_type](file_name, file_path, content)
-        else:
-            raise UnsupportedFileTypeError(f"Unsupported file type: {file_type}. Cannot save the file.")
+        if file_type not in file_type_handlers:
+            raise UnsupportedFileTypeError(file_name=file_name, supported_types=list(file_type_handlers))
         
\ No newline at end of file
diff --git a/superagi/tools/file/write_file.py b/superagi/tools/file/write_file.py
index 4fcc3d75f..6b0618982 100644
--- a/superagi/tools/file/write_file.py
+++ b/superagi/tools/file/write_file.py
@@ -101,7 +101,13 @@ def _get_file_path_of_images(self, attached_files: list):
         """
         image_extensions, image_paths = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp'],[]
         for file_name in attached_files:
-            path = self._get_file_path(file_name=file_name)
+            path = ResourceHelper().get_agent_read_resource_path(
+                    file_name,
+                    agent=Agent.get_agent_from_id(self.toolkit_config.session, self.agent_id),
+                    agent_execution=AgentExecution.get_agent_execution_from_id(
+                        self.toolkit_config.session, self.agent_execution_id
+                    ),
+                )
             if not os.path.exists(path):
                 continue
             _, file_extension = os.path.splitext(path)
@@ -110,25 +116,6 @@ def _get_file_path_of_images(self, attached_files: list):
                 image_paths.append(path)
         return image_paths
     
-    def _get_file_path(self, file_name: str) -> list:
-        """
-        Gets the path of the file
-
-        Args:
-            file_name: Name of file 
-
-        Returns:
-            Full paths of the file
-        """
-        session = self.toolkit_config.session
-
-        return ResourceHelper().get_agent_read_resource_path(
-            file_name,
-            agent=Agent.get_agent_from_id(session, self.agent_id),
-            agent_execution=AgentExecution.get_agent_execution_from_id(
-                session, self.agent_execution_id
-            ),
-        )
     
     def _image_to_base64(self, image_path):
         with open(image_path, "rb") as image_file:

From 6606919829cb17fb37212647b2a95e19a9b67515 Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Mon, 4 Sep 2023 13:33:10 +0530
Subject: [PATCH 17/23] updated the name of exception file

---
 superagi/exceptions/{file_errors.py => file_exceptions.py} | 0
 1 file changed, 0 insertions(+), 0 deletions(-)
 rename superagi/exceptions/{file_errors.py => file_exceptions.py} (100%)

diff --git a/superagi/exceptions/file_errors.py b/superagi/exceptions/file_exceptions.py
similarity index 100%
rename from superagi/exceptions/file_errors.py
rename to superagi/exceptions/file_exceptions.py

From 91adecf51dc4e5be0dd465d30383a92d2435bafe Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Mon, 4 Sep 2023 13:33:31 +0530
Subject: [PATCH 18/23] update exception import

---
 superagi/resource_manager/file_manager.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/superagi/resource_manager/file_manager.py b/superagi/resource_manager/file_manager.py
index 93b6396e7..45ed90d7b 100644
--- a/superagi/resource_manager/file_manager.py
+++ b/superagi/resource_manager/file_manager.py
@@ -9,7 +9,7 @@
 from superagi.models.agent import Agent
 from superagi.models.agent_execution import AgentExecution
 from superagi.types.storage_types import StorageType
-from superagi.exceptions.file_errors import UnsupportedFileTypeError, FileNotCreatedError
+from superagi.exceptions.file_exceptions import UnsupportedFileTypeError, FileNotCreatedError
 
 import pdfkit
 from htmldocx import HtmlToDocx

From 224b22735f95c2a93b143786840b927ac2229915 Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Mon, 4 Sep 2023 13:33:59 +0530
Subject: [PATCH 19/23] Added unit test cases for file manager

---
 .../resource_manager/test_file_manager.py     | 32 ++++++++++++++++++-
 1 file changed, 31 insertions(+), 1 deletion(-)

diff --git a/tests/unit_tests/resource_manager/test_file_manager.py b/tests/unit_tests/resource_manager/test_file_manager.py
index b382520d9..f0d4fb657 100644
--- a/tests/unit_tests/resource_manager/test_file_manager.py
+++ b/tests/unit_tests/resource_manager/test_file_manager.py
@@ -26,7 +26,7 @@ def test_write_binary_file(resource_manager):
         logger_mock.assert_called_once_with("Binary test.png saved successfully")
 
 
-def test_write_file(resource_manager):
+def test_write_txt_file(resource_manager):
     with patch.object(ResourceHelper, 'get_resource_path', return_value='test_path'), \
             patch.object(ResourceHelper, 'make_written_file_resource',
                          return_value=Resource(name='test.txt', storage_type='S3')), \
@@ -35,3 +35,33 @@ def test_write_file(resource_manager):
         result = resource_manager.write_file('test.txt', 'content')
         assert result == "test.txt - File written successfully"
         logger_mock.assert_called_once_with("test.txt - File written successfully")
+
+def test_write_pdf_file(resource_manager):
+    with patch.object(ResourceHelper, 'get_resource_path', return_value='test_path'), \
+            patch.object(ResourceHelper, 'make_written_file_resource',
+                         return_value=Resource(name='test.pdf', storage_type='S3')), \
+            patch.object(S3Helper, 'upload_file'), \
+            patch.object(logger, 'info') as logger_mock:
+        result = resource_manager.write_file('test.pdf', 'content')
+        assert result == "test.pdf - File written successfully"
+        logger_mock.assert_called_once_with("test.pdf - File written successfully")
+        
+def test_write_docx_file(resource_manager):
+    with patch.object(ResourceHelper, 'get_resource_path', return_value='test_path'), \
+            patch.object(ResourceHelper, 'make_written_file_resource',
+                         return_value=Resource(name='test.docx', storage_type='S3')), \
+            patch.object(S3Helper, 'upload_file'), \
+            patch.object(logger, 'info') as logger_mock:
+        result = resource_manager.write_file('test.docx', 'content')
+        assert result == "test.docx - File written successfully"
+        logger_mock.assert_called_once_with("test.docx - File written successfully")
+
+def test_read_file(resource_manager):
+    with patch.object(ResourceHelper, 'get_resource_path', return_value='test_path'), \
+            patch.object(logger, 'info') as logger_mock, \
+            patch('builtins.open', create=True) as mock_open:
+        mock_file = mock_open.return_value
+        mock_file.read.return_value = 'file content'
+        result = resource_manager.read_file('test.txt')
+        assert result == 'file content'
+        logger_mock.assert_called_once_with("test.txt - File read successfully")
\ No newline at end of file

From 82ce44bb47d6a1b58326d5b5311e40981d7d8e86 Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Mon, 4 Sep 2023 13:35:58 +0530
Subject: [PATCH 20/23] Added CSV test file UT case

---
 .../unit_tests/resource_manager/test_file_manager.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/tests/unit_tests/resource_manager/test_file_manager.py b/tests/unit_tests/resource_manager/test_file_manager.py
index f0d4fb657..8d10f9884 100644
--- a/tests/unit_tests/resource_manager/test_file_manager.py
+++ b/tests/unit_tests/resource_manager/test_file_manager.py
@@ -55,7 +55,17 @@ def test_write_docx_file(resource_manager):
         result = resource_manager.write_file('test.docx', 'content')
         assert result == "test.docx - File written successfully"
         logger_mock.assert_called_once_with("test.docx - File written successfully")
-
+        
+def test_write_csv_file(resource_manager):
+    with patch.object(ResourceHelper, 'get_resource_path', return_value='test_path'), \
+            patch.object(ResourceHelper, 'make_written_file_resource',
+                         return_value=Resource(name='test.csv', storage_type='S3')), \
+            patch.object(S3Helper, 'upload_file'), \
+            patch.object(logger, 'info') as logger_mock:
+        result = resource_manager.write_file('test.csv', 'content')
+        assert result == "test.csv - File written successfully"
+        logger_mock.assert_called_once_with("test.csv - File written successfully")
+        
 def test_read_file(resource_manager):
     with patch.object(ResourceHelper, 'get_resource_path', return_value='test_path'), \
             patch.object(logger, 'info') as logger_mock, \

From 7cf317d86e04310475055d107cf10e8a61aab19c Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Mon, 4 Sep 2023 17:11:52 +0530
Subject: [PATCH 21/23] code cleanup

---
 .../unit_tests/resource_manager/test_file_manager.py | 12 +-----------
 1 file changed, 1 insertion(+), 11 deletions(-)

diff --git a/tests/unit_tests/resource_manager/test_file_manager.py b/tests/unit_tests/resource_manager/test_file_manager.py
index 8d10f9884..a33f4980f 100644
--- a/tests/unit_tests/resource_manager/test_file_manager.py
+++ b/tests/unit_tests/resource_manager/test_file_manager.py
@@ -64,14 +64,4 @@ def test_write_csv_file(resource_manager):
             patch.object(logger, 'info') as logger_mock:
         result = resource_manager.write_file('test.csv', 'content')
         assert result == "test.csv - File written successfully"
-        logger_mock.assert_called_once_with("test.csv - File written successfully")
-        
-def test_read_file(resource_manager):
-    with patch.object(ResourceHelper, 'get_resource_path', return_value='test_path'), \
-            patch.object(logger, 'info') as logger_mock, \
-            patch('builtins.open', create=True) as mock_open:
-        mock_file = mock_open.return_value
-        mock_file.read.return_value = 'file content'
-        result = resource_manager.read_file('test.txt')
-        assert result == 'file content'
-        logger_mock.assert_called_once_with("test.txt - File read successfully")
\ No newline at end of file
+        logger_mock.assert_called_once_with("test.csv - File written successfully")
\ No newline at end of file

From 6f4fdf5412a3437798fad661b4acd587ad8536a9 Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Wed, 6 Sep 2023 13:47:53 +0530
Subject: [PATCH 22/23] reverting the few unecessary changes

---
 superagi/models/agent_execution_feed.py                     | 2 +-
 .../tools/image_generation/stable_diffusion_image_gen.py    | 6 ++++--
 superagi/tools/tool_response_query_manager.py               | 2 +-
 3 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/superagi/models/agent_execution_feed.py b/superagi/models/agent_execution_feed.py
index 00f50ee60..c9831dd69 100644
--- a/superagi/models/agent_execution_feed.py
+++ b/superagi/models/agent_execution_feed.py
@@ -52,7 +52,7 @@ def get_last_tool_response(cls, session: Session, agent_execution_id: int, tool_
             if agent_execution_feed.feed.startswith("Tool"):
                 return agent_execution_feed.feed
         return ""
-    
+
     @classmethod
     def fetch_agent_execution_feeds(cls, session, agent_execution_id: int):
         agent_execution = AgentExecution.find_by_id(session, agent_execution_id)
diff --git a/superagi/tools/image_generation/stable_diffusion_image_gen.py b/superagi/tools/image_generation/stable_diffusion_image_gen.py
index 8a71f7c42..b3c5d2c96 100644
--- a/superagi/tools/image_generation/stable_diffusion_image_gen.py
+++ b/superagi/tools/image_generation/stable_diffusion_image_gen.py
@@ -57,8 +57,10 @@ def _execute(self, prompt: str, image_names: list, width: int = 512, height: int
         data = response.json()
 
         artifacts = data['artifacts']
-        base64_strings = [artifact['base64'] for artifact in artifacts]
-        image_paths=[]
+        base64_strings = []
+        for artifact in artifacts:
+            base64_strings.append(artifact['base64'])
+
         for i in range(num):
             image_base64 = base64_strings[i]
             img_data = base64.b64decode(image_base64)
diff --git a/superagi/tools/tool_response_query_manager.py b/superagi/tools/tool_response_query_manager.py
index 3aebc6eab..a3d681ab6 100644
--- a/superagi/tools/tool_response_query_manager.py
+++ b/superagi/tools/tool_response_query_manager.py
@@ -11,7 +11,7 @@ def __init__(self, session: Session, agent_execution_id: int,memory:VectorStore)
 
     def get_last_response(self, tool_name: str = None):
         return AgentExecutionFeed.get_last_tool_response(self.session, self.agent_execution_id, tool_name)
-      
+
     def get_relevant_response(self, query: str,metadata:dict, top_k: int = 5):
         documents = self.memory.get_matching_text(query, metadata=metadata)
         relevant_responses = ""

From 7c834c0ac81e7dfd290ca6e89d7c7252a218053c Mon Sep 17 00:00:00 2001
From: Arkajit-Datta <arkajitdatta2010@gmail.com>
Date: Wed, 6 Sep 2023 14:06:58 +0530
Subject: [PATCH 23/23] removing changes

---
 superagi/models/agent_execution_feed.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/superagi/models/agent_execution_feed.py b/superagi/models/agent_execution_feed.py
index c9831dd69..61f449585 100644
--- a/superagi/models/agent_execution_feed.py
+++ b/superagi/models/agent_execution_feed.py
@@ -66,4 +66,3 @@ def fetch_agent_execution_feeds(cls, session, agent_execution_id: int):
             return agent_feeds
         else:
             return agent_feeds[2:]
-