Update app.py

app.py

@@ -11,7 +11,9 @@ import pandas as pd  # If you're working with DataFrames
 import matplotlib.figure  # If you're using matplotlib figures
 import numpy as np
 
-from custom_agent import CustomHfAgent
+from custom_agent import CustomHfAgent 
+from tool_loader import ToolLoader
+from tool_config import tool_names   
 
 
 # For Altair charts
@@ -30,6 +32,7 @@ import pydeck as pdk
 import logging
 import streamlit as st
 from transformers import load_tool, Agent
+from tool_loader import ToolLoader
 
 # Configure the logging settings for transformers
 transformers_logger = logging.getLogger("transformers.file_utils")
@@ -39,20 +42,6 @@ transformers_logger.setLevel(logging.INFO)  # Set the desired logging level
 import time  
 import torch
 
-class ToolLoader:
-    def __init__(self, tool_names):
-        self.tools = self.load_tools(tool_names)
-
-    def load_tools(self, tool_names):
-        loaded_tools = []
-        for tool_name in tool_names:
-            try:
-                tool = load_tool(tool_name)
-                loaded_tools.append(tool)
-            except Exception as e:
-                log_response(f"Error loading tool '{tool_name}': {e}")
-        return loaded_tools
-
 
 def handle_submission(user_message, selected_tools, url_endpoint):
     
@@ -84,26 +73,7 @@ def log_response(response):
     print(response)
 
     
-# Define the tool names to load
-tool_names = [
-    "Chris4K/random-character-tool",
-    "Chris4K/text-generation-tool",
-    "Chris4K/sentiment-tool", 
-    "Chris4K/token-counter-tool", 
-    "Chris4K/most-downloaded-model",
-    "Chris4K/rag-tool",
-    "Chris4K/word-counter-tool",
-    "Chris4K/sentence-counter-tool",
-    "Chris4K/EmojifyTextTool",
-    "Chris4K/NamedEntityRecognitionTool",
-    "Chris4K/TextDownloadTool",
-    "Chris4K/source-code-retriever-tool",
-    "Chris4K/text-to-image",
-    "Chris4K/text-to-video",
-    "Chris4K/image-transformation",
-    "Chris4K/latent-upscaler-tool"
-    # More cool tools to come
-]
+
 
 # Create tool loader instance
 tool_loader = ToolLoader(tool_names)