Added multi web search switching capability, automated

app.py

@@ -18,9 +18,46 @@ Perplex_Assistant_Prompt = """You are a helpful AI assistant that searches the w
 # Set up API key in environment variable as expected by HfApiModel
 os.environ["HUGGINGFACE_API_TOKEN"] = os.getenv("HUGGINGFACE_API_KEY", "")
 
-# Initialize the standard search tools
-ddg_search_tool = DuckDuckGoSearchTool(max_results=10)  # Default is 10 results
-# google_search_tool = GoogleSearchTool()
+# Initialize search tools with fallback capability
+try:
+    # Try DuckDuckGo first (default)
+    print("Initializing DuckDuckGo search tool...")
+    ddg_search_tool = DuckDuckGoSearchTool(max_results=10)
+    
+    # Test the tool with a simple query
+    test_result = ddg_search_tool("test query")
+    print("DuckDuckGo search tool initialized successfully.")
+    
+    # Use DuckDuckGo as the primary search tool
+    primary_search_tool = ddg_search_tool
+    search_tool_name = "DuckDuckGo"
+    
+except Exception as e:
+    print(f"Error initializing DuckDuckGo search tool: {str(e)}")
+    print("Falling back to Google search tool...")
+    
+    try:
+        # Import GoogleSearchTool only if needed
+        from smolagents import GoogleSearchTool
+        google_search_tool = GoogleSearchTool()
+        
+        # Test the Google search tool
+        test_result = google_search_tool("test query")
+        print("Google search tool initialized successfully.")
+        
+        # Use Google as the fallback search tool
+        primary_search_tool = google_search_tool
+        search_tool_name = "Google"
+        
+    except Exception as google_error:
+        print(f"Error initializing Google search tool: {str(google_error)}")
+        print("WARNING: No working search tool available. Agent functionality will be limited.")
+        # Create a minimal replacement that returns an explanatory message
+        def search_fallback(query):
+            return f"Search functionality unavailable. Both DuckDuckGo and Google search tools failed to initialize. Query was: {query}"
+        
+        primary_search_tool = search_fallback
+        search_tool_name = "Unavailable"
 
 # Initialize the VisitWebpageTool
 visit_webpage_tool = VisitWebpageTool()
@@ -378,22 +415,6 @@ model.__call__ = try_model_call_with_fallbacks
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-@tool
-def web_search(query: str) -> str:
-    """A tool that searches the web for information.
-    
-    Args:
-        query: The search query string
-        
-    Returns:
-        Search results as text
-    """
-    try:
-        # Use the already initialized DuckDuckGo search tool
-        return ddg_search_tool(query)
-    except Exception as e:
-        return f"Error performing web search for '{query}': {str(e)}"
-
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
@@ -402,14 +423,12 @@ agent = CodeAgent(
     tools=[
         final_answer, 
         Sonar_Web_Search_Tool,
-        web_search,  # Added explicit web_search tool
-        ddg_search_tool,  # Added DuckDuckGo search tool
-#        google_search_tool,  # Added Google search tool 
+        primary_search_tool,  # Use whichever search tool was successfully initialized
         get_current_time_in_timezone, 
         image_generation_tool,
         Dataset_Creator_Tool,
         Check_Dataset_Validity,
-        visit_webpage_tool,  # Added webpage visiting capability
+        visit_webpage_tool,
     ],  
     max_steps=6,
     verbosity_level=1,
@@ -420,6 +439,8 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
+print(f"Agent initialized with {search_tool_name} as primary search tool")
+
 # To fix the TypeError in Gradio_UI.py, you would need to modify that file
 # For now, we'll just use the agent directly
 try: