Update app.py

app.py

@@ -6,6 +6,7 @@ import yaml
 import json
 from tools.final_answer import FinalAnswerTool
 from tools.web_search import DuckDuckGoSearchTool
+from bs4 import BeautifulSoup
 from duckduckgo_search import DDGS
 import re
 from typing import List, Dict, Any, Optional
@@ -30,42 +31,39 @@ def get_healthy_cheat_meal(diet_type: str) -> List[Dict[str, Any]]:
             - source: URL of the recipe source
     """
     query = f"healthy {diet_type} sweet dish recipe"
-    
+
     try:
-        # Initialize the web search tool
-        search_tool = DuckDuckGoSearchTool()
-        
-        # Get search results
-        search_response = search_tool.run(query)
-        if not search_response:
+        ddgs = DDGS() # Initialize DDGS here
+        search_results = ddgs.text(query, region="us-en", safesearch="moderate", max_results=5)
+
+        if not search_results:
             return [{"error": "No recipes found. Try a different diet type or check your query."}]
-        
+
         recipes = []
-        
-        # Process each search result
-        for result in search_response[:5]:  # Limit to 5 recipes
+
+        for result in search_results:
             try:
-                title = result['title'].split(' - ')[0].strip()
-                url = result['href']
-                
-                # Attempt to fetch and parse recipe details
+                title = result.get('title', '').split(' - ')[0].strip() # Use get and handle missing keys
+                url = result.get('href', '') # Use get and handle missing keys
+
+                if not url: # If URL is missing, skip the result
+                    continue
+
                 response = requests.get(url, timeout=10, headers={
                     'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
                 })
-                
+
                 if response.status_code == 200:
                     soup = BeautifulSoup(response.text, 'html.parser')
-                    
-                    # Extract basic recipe information
+
                     ingredients = []
                     for ingredient in soup.select('.recipe-ingredients li, .ingredients li'):
                         ingredients.append(ingredient.get_text().strip())
-                    
+
                     instructions = []
                     for step in soup.select('.recipe-instructions li, .instructions li'):
                         instructions.append(step.get_text().strip())
-                    
-                    # Create formatted recipe entry
+
                     recipe = {
                         "recipe_name": title,
                         "ingredients": ingredients[:10] if ingredients else ["Please check source for ingredients"],
@@ -74,28 +72,32 @@ def get_healthy_cheat_meal(diet_type: str) -> List[Dict[str, Any]]:
                         "adaptations": f"This recipe is suitable for {diet_type} diet. Adjust sweeteners and ingredients according to your specific dietary needs.",
                         "source": url
                     }
-                    
+
                     recipes.append(recipe)
-                
+
             except Exception as e:
+                print(f"Error processing a recipe: {e}")  # Print the error for debugging
                 continue  # Skip this recipe if there's an error
-        
+            time.sleep(1) # Added a small delay to respect rate limits
+
+
         if not recipes:
             # Fallback with basic information if web scraping fails
-            for result in search_response[:5]:
+            for result in search_results:
                 recipes.append({
-                    "recipe_name": result['title'].split(' - ')[0].strip(),
+                    "recipe_name": result.get('title', '').split(' - ')[0].strip(), # Use get and handle missing keys
                     "ingredients": ["Please check source for full ingredients list"],
                     "instructions": ["Please check source for detailed instructions"],
                     "nutritional_benefits": f"This {diet_type} recipe is designed to be a healthier alternative to traditional sweet dishes.",
                     "adaptations": f"This recipe is suitable for {diet_type} diet. Adjust sweeteners and ingredients according to your specific dietary needs.",
-                    "source": result['link']
+                    "source": result.get('href', '') # Use get and handle missing keys
                 })
-        
+
         return recipes if recipes else [{"error": "Unable to process recipes. Please try again."}]
-        
+
     except Exception as e:
         return [{"error": f"An error occurred while searching: {str(e)}"}]
+
 # def get_healthy_cheat_meal(diet_type: str) -> List[Dict[str, Any]]:
 #     """Search for healthy sweet dish alternatives based on specific diet types.
     
@@ -145,7 +147,7 @@ def get_healthy_cheat_meal(diet_type: str) -> List[Dict[str, Any]]:
         
 # Initialize tools
 final_answer = FinalAnswerTool()
-web_search_tool = DuckDuckGoSearchTool()
+#web_search_tool = DuckDuckGoSearchTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
@@ -167,7 +169,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,web_search_tool,get_healthy_cheat_meal], ## add your tools here (don't remove final answer)
+    tools=[final_answer,get_healthy_cheat_meal], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,