Update app.py

app.py

@@ -236,7 +236,7 @@ from langchain_community.document_loaders import PyMuPDFLoader
 from docx import Document as DocxDocument
 import openpyxl
 
-class AnalyseAttachmentTool:
+class AnalyseAttachmentTool(Tool):
     """
     A tool for analyzing various attachment types (PY, PDF, TXT, DOCX, XLSX)
     and extracting their text content.
@@ -311,7 +311,7 @@ from io import BytesIO
 # Define image analysis tool
 import requests
 
-class ImageAnalysisTool:
+class ImageAnalysisTool(Tool):
     """
     A tool for analyzing images using a hosted Hugging Face model.
     """
@@ -420,7 +420,7 @@ class BasicAgent:
 
         # Custom tools - ensure these classes are defined and imported
         self.video_transcription_tool = VideoTranscriptionTool()
-        self.image_analysis_tool_instance = ImageAnalysisTool() # Renamed for clarity
+        self.image_analysis_tool = ImageAnalysisTool() # Renamed for clarity
         self.analyse_attachment_tool = AnalyseAttachmentTool() # Renamed for clarity
         self.code_llama_tool = CodeLlamaTool() # Ensure this class is defined/imported
 
@@ -439,7 +439,7 @@ If the answer is a comma-separated list, apply the above rules for each element
             tools=[
                 self.search_tool, # Use the initialized DuckDuckGoSearchTool instance
                 self.visit_webpage_tool,
-                self.image_analysis_tool_instance # Use the initialized instance of your ImageAnalysisTool
+                self.image_analysis_tool
             ],
             model=self.model, # Use self.model
             max_steps=10,