Update Gradio_UI.py

Gradio_UI.py

@@ -15,19 +15,21 @@ class GradioUI:
         tools = agent.tools if isinstance(agent.tools, (list, tuple)) else list(agent.tools)
 
         for tool in tools:
-            # Match tools based on description (from docstring) or fallback string
-            desc = getattr(tool, "description", "") or str(tool)
-            if "current local time" in desc or "timezone" in desc:
+            func_name = getattr(tool, "__name__", None)
+            raw_func_name = getattr(getattr(tool, "fn", None), "__name__", "")
+        
+            if "time" in func_name or "timezone" in raw_func_name:
                 self.get_time_tool = tool
-            elif "weather" in desc.lower():
+            elif "weather" in func_name or "weather" in raw_func_name:
                 self.get_weather_tool = tool
-            elif "question" in desc.lower() and "pdf" in desc.lower():
+            elif "document_qna_tool" in func_name or "document_qna_tool" in raw_func_name:
                 self.doc_qa_tool = tool
-            elif "text-to-image" in desc.lower() or "image" in str(tool).lower():
+            elif "image" in func_name or "image" in raw_func_name:
                 self.image_gen_tool = tool
-            elif "duckduckgo" in str(tool).lower() or "search" in desc.lower():
+            elif "search" in func_name or "duckduckgo" 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"