Update Gradio_UI.py

Gradio_UI.py

@@ -11,29 +11,28 @@ class GradioUI:
         self.search_tool = None
         self.doc_qa_tool = None
 
-        # Map tools by known types or names
-        for tool in self.agent.tools:
-            try:
-                tool_name = getattr(tool, "__name__", None)
-                if tool_name == "get_current_time_in_timezone":
-                    self.get_time_tool = tool
-                elif tool_name == "get_current_weather":
-                    self.get_weather_tool = tool
-                elif tool_name == "document_qna_tool":
-                    self.doc_qa_tool = tool
-                elif "text-to-image" in str(tool):
-                    self.image_gen_tool = tool
-                elif "DuckDuckGoSearchTool" in str(tool):
-                    self.search_tool = tool
-            except Exception:
-                pass
-
-        # Fallback in case mapping fails
-        self.get_time_tool = self.get_time_tool or self.agent.tools[0]
-        self.get_weather_tool = self.get_weather_tool or self.agent.tools[1]
-        self.image_gen_tool = self.image_gen_tool or self.agent.tools[2]
-        self.search_tool = self.search_tool or self.agent.tools[3]
-        self.doc_qa_tool = self.doc_qa_tool or self.agent.tools[4]
+        # Try to detect tool types from dictionary-like access
+        tools = getattr(agent, "tools", {})
+
+        for tool in tools.values():  # ✅ Use `.values()` instead of indexing
+            tool_name = getattr(tool, "__name__", None)
+            if tool_name == "get_current_time_in_timezone":
+                self.get_time_tool = tool
+            elif tool_name == "get_current_weather":
+                self.get_weather_tool = tool
+            elif tool_name == "document_qna_tool":
+                self.doc_qa_tool = tool
+            elif "text-to-image" in str(tool):
+                self.image_gen_tool = tool
+            elif "DuckDuckGoSearchTool" in str(tool):
+                self.search_tool = tool
+
+        # Optional fallback with warnings
+        assert self.get_time_tool, "Missing: get_current_time_in_timezone"
+        assert self.get_weather_tool, "Missing: get_current_weather"
+        assert self.doc_qa_tool, "Missing: document_qna_tool"
+        assert self.image_gen_tool, "Missing: image generation tool"
+        assert self.search_tool, "Missing: search tool"
 
     def launch(self):
         with gr.Blocks() as demo: