bug fix

src/agents/__init__.py

@@ -1,46 +1,21 @@
 """
-UI Components Module
-
-This module contains UI-related components including styles, visualizations,
-and utility functions for the Streamlit application.
+Agent modules for chatbot implementations
 """
 
-from .styles import get_custom_css
-from .components import (
-    display_chunk_statistics_charts,
-    display_chunk_statistics_table
-)
-from .utils import extract_chunk_statistics
-from .visual_documents import (
-    display_visual_search_results,
-    display_visual_document_statistics,
-    display_visual_document_details
-)
-from .saliency import (
-    generate_tile_aware_saliency,
-    can_generate_saliency,
-    get_saliency_metadata_summary,
-    DEFAULT_ALPHA,
-    DEFAULT_COLORMAP,
-    DEFAULT_THRESHOLD_PERCENTILE
-)
+from .gemini_chatbot import get_gemini_chatbot
+from .visual_chatbot import get_visual_chatbot
+from .multi_agent_chatbot import get_multi_agent_chatbot
+from .smart_chatbot import get_chatbot as get_smart_chatbot
+from .visual_multi_agent_chatbot import get_visual_multi_agent_chatbot
+
+# Alias for backward compatibility
+get_visual_chatbot_v2 = get_visual_multi_agent_chatbot
 
 __all__ = [
-    "get_custom_css",
-    "display_chunk_statistics_charts",
-    "display_chunk_statistics_table",
-    "extract_chunk_statistics",
-    "display_visual_search_results",
-    "display_visual_document_statistics",
-    "display_visual_document_details",
-    # Saliency functions
-    "generate_tile_aware_saliency",
-    "can_generate_saliency",
-    "get_saliency_metadata_summary",
-    "DEFAULT_ALPHA",
-    "DEFAULT_COLORMAP",
-    "DEFAULT_THRESHOLD_PERCENTILE"
+    "get_smart_chatbot",
+    "get_multi_agent_chatbot", 
+    "get_gemini_chatbot",
+    "get_visual_chatbot",
+    "get_visual_multi_agent_chatbot",
+    "get_visual_chatbot_v2"
 ]
-
-
-