Add files via upload

Author: if-ai

IFChatPromptNode.py

@@ -367,8 +367,8 @@ def __init__(self):
         self.random = False
 
         self.comfy_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-        self.rag_dir = os.path.join(folder_paths.base_path, "custom_nodes", "ComfyUI-IF_AI_tools", "IF_AI", "rag")
-        self.presets_dir = os.path.join(folder_paths.base_path, "custom_nodes", "ComfyUI-IF_AI_tools", "IF_AI", "presets")
+        self.rag_dir = os.path.join(folder_paths.user_directory, "IF_AI", "rag")
+        self.presets_dir = os.path.join(folder_paths.user_directory, "IF_AI", "presets")
 
         self.stop_file = os.path.join(self.presets_dir, "stop_strings.json")
         self.assistants_file = os.path.join(self.presets_dir, "assistants.json")

IFDisplayTextWildcardNode.py

@@ -20,8 +20,8 @@ def __init__(self):
 
         # Initialize paths
         self.base_path = folder_paths.base_path
-        self.presets_dir = os.path.join(self.base_path, "custom_nodes", "ComfyUI_IF_AI_tools", "IF_AI", "presets")
-        self.wildcards_dir = os.path.join(self.presets_dir, "wildcards")  # Updated path
+        self.presets_dir = os.path.join(folder_paths.user_directory, "IF_AI", "presets")
+        self.wildcards_dir = os.path.join(self.presets_dir, "wildcards") 
 
         # Load wildcards
         self.wildcards = self.load_wildcards()

IFImagePromptNode.py

@@ -73,7 +73,7 @@ def __init__(self):
         self.strategies = "normal"
         # Initialize paths and load presets
         self.base_path = folder_paths.base_path
-        self.presets_dir = os.path.join(self.base_path, "custom_nodes", "ComfyUI-IF_AI_tools", "IF_AI", "presets")
+        self.presets_dir = os.path.join(folder_paths.user_directory, "IF_AI", "presets")
 
         # Load preset configurations
         self.profiles = self.load_presets(os.path.join(self.presets_dir, "profiles.json"))

IFPromptMkrNode.py

@@ -73,7 +73,7 @@ def __init__(self):
         self.strategies = "normal"
         # Initialize paths and load presets
         self.base_path = folder_paths.base_path
-        self.presets_dir = os.path.join(self.base_path, "custom_nodes", "ComfyUI-IF_AI_tools", "IF_AI", "presets")
+        self.presets_dir = os.path.join(folder_paths.user_directory, "IF_AI", "presets")
 
         # Load preset configurations
         self.profiles = self.load_presets(os.path.join(self.presets_dir, "profiles.json"))

IFSaveTextNode.py

@@ -41,7 +41,7 @@ def process_text(self, question_input, negative_input, response_input, save_file
         return (question_input, response_input, negative_input, turn)
 
     def save_text_to_file(self, turn_data, file_format, save_mode):
-        save_text_dir = folder_paths.get_output_directory()
+        save_text_dir = os.path.join(folder_paths.user_directory, "IF_AI", "saved_outputs")
         os.makedirs(save_text_dir, exist_ok=True)
         file_path = os.path.join(save_text_dir, f"output.{file_format}")
 

agent_tool.py

@@ -23,7 +23,7 @@ def __init__(self, name, description, system_prompt, default_engine, default_mod
 
     def load(self):
         # Construct the path to the ComfyUI-IF_AI_tools directory
-        if_ai_tools_dir = os.path.join(folder_paths.base_path, "custom_nodes",  "ComfyUI-IF_AI_tools")
+        if_ai_tools_dir = os.path.join(folder_paths.base_path, "custom_nodes", "ComfyUI-IF_AI_tools")
 
         # Add the ComfyUI-IF_AI_tools directory to sys.path
         if if_ai_tools_dir not in sys.path:

colpaliRAG_module.py

@@ -36,7 +36,7 @@
 class colpaliRAGapp:
     def __init__(self):
         self.comfy_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-        self.rag_dir = os.path.join(folder_paths.base_path, "custom_nodes",  "ComfyUI-IF_AI_tools", "IF_AI", "rag")
+        self.rag_dir = os.path.join(folder_paths.user_directory, "IF_AI", "rag")
         self._rag_root_dir = None
         self._input_dir = None
         self.graphrag_app = GraphRAGapp()

graphRAG_module.py

@@ -43,7 +43,7 @@
 class GraphRAGapp:
     def __init__(self):
         self.comfy_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-        self.rag_dir = os.path.join(folder_paths.base_path, "custom_nodes",  "ComfyUI-IF_AI_tools", "IF_AI", "rag")
+        self.rag_dir = os.path.join(folder_paths.user_directory, "IF_AI", "rag")
         self._rag_root_dir = None
         self._input_dir = None
         self.embedding_func = None

transformers_api.py

@@ -18,7 +18,7 @@
 import logging
 import os
 import re
-from folder_paths import models_dir
+import folder_paths
 from unittest.mock import patch
 from transformers.dynamic_module_utils import get_imports
 import json
@@ -35,7 +35,7 @@
 
 class TransformersModelManager:
     def __init__(self):
-        self.models_dir = models_dir
+        self.models_dir = os.path.join(folder_paths.models_dir, "LLM")
         self.models = {}
         self.processors = {}
         self.loaded_models = {} 
@@ -52,7 +52,7 @@ def download_model_if_not_exists(self, model_name):
         from huggingface_hub import snapshot_download
 
         model_dir = model_name.rsplit('/', 1)[-1]
-        model_path = os.path.join(self.models_dir, "LLM", model_dir)
+        model_path = os.path.join(self.models_dir, model_dir)
         if not os.path.exists(model_path):
             logger.info(f"Downloading model '{model_name}' to: {model_path}")
             try:

utils.py

@@ -6,7 +6,6 @@
 import torch
 import torchvision
 import cv2
-import folder_paths
 import base64
 import logging
 import datetime