Update Gradio_UI.py

Gradio_UI.py

@@ -15,21 +15,22 @@ class GradioUI:
         tools = agent.tools if isinstance(agent.tools, (list, tuple)) else list(agent.tools)
 
         for tool in tools:
-            func_name = getattr(tool, "__name__", None)
-            raw_func_name = getattr(getattr(tool, "fn", None), "__name__", "")
+            # Get raw function name if possible
+            raw_func = getattr(tool, "fn", None)
+            raw_func_name = getattr(raw_func, "__name__", "").lower()
+            tool_str = str(tool).lower()
         
-            if "time" in func_name or "timezone" in raw_func_name:
+            if "timezone" in raw_func_name or "get_current_time_in_timezone" in raw_func_name:
                 self.get_time_tool = tool
-            elif "weather" in func_name or "weather" in raw_func_name:
+            elif "weather" in raw_func_name or "get_current_weather" in raw_func_name:
                 self.get_weather_tool = tool
-            elif "document_qna_tool" in func_name or "document_qna_tool" in raw_func_name:
+            elif "document_qna_tool" in raw_func_name:
                 self.doc_qa_tool = tool
-            elif "image" in func_name or "image" in raw_func_name:
+            elif "text-to-image" in tool_str or "image" in raw_func_name:
                 self.image_gen_tool = tool
-            elif "search" in func_name or "duckduckgo" in raw_func_name:
+            elif "duckduckgo" in tool_str or "search" in raw_func_name:
                 self.search_tool = tool
 
-
         # Validate all tools are loaded
         assert self.get_time_tool, "Missing: get_current_time_in_timezone"
         assert self.get_weather_tool, "Missing: get_current_weather"