Update app.py

app.py

@@ -8,6 +8,7 @@ from tools.web_search import DuckDuckGoSearchTool
 import duckduckgo_search
 import requests
 from bs4 import BeautifulSoup
+import re
 from typing import List, Dict, Any, Optional
 from Gradio_UI import GradioUI
 
@@ -19,16 +20,11 @@ def get_healthy_cheat_meal(diet_type: str) -> List[Dict[str, Any]]:
     When a user provides a diet type—such as Keto, Paleo, Vegan, Vegetarian, or Non-Vegetarian—retrieve and summarize 
     reliable, up-to-date information on healthy sweet dish alternatives.
 
-    Ensure that your responses include:
-    - Clear Recipe Details: List ingredients and nutritional benefits.
-    - Diet-Specific Adaptations: Highlight modifications and substitutions relevant to the specified diet type.
-    - Credible Sources: Rely on trusted culinary and nutritional resources to ensure accuracy and quality.
-
     Args:
         diet_type (str): User-provided diet type.
 
     Returns:
-        List[Dict[str, Any]]: A list of dictionaries with the following structure:
+        A list of at least 5 sweet dishes dictionaries with the following structure:
             {
                 "recipe_name": str,             # The name of the recipe.
                 "source": str                   # A credible source or URL where the recipe was found.
@@ -38,8 +34,15 @@ def get_healthy_cheat_meal(diet_type: str) -> List[Dict[str, Any]]:
     query = f"healthy {diet_type} sweet dish recipe with ingredients and instructions"
     
     try:
-        # Perform web search using DuckDuckGo
-        search_results = duckduckgo_search.ddg(query, max_results=5)
+        # Perform web search using DuckDuckGo with specific parameters
+        search_results = ddg(
+            query,
+            region="us-en",
+            safesearch="moderate",
+            time=None,
+            max_results=5,
+            output='json'
+        )
         
         if not search_results:
             return [{"error": "No recipes found. Try a different diet type or check your query."}]
@@ -47,16 +50,26 @@ def get_healthy_cheat_meal(diet_type: str) -> List[Dict[str, Any]]:
         recipes = []
         
         for result in search_results:
-            recipes.append({
-                "recipe_name": result.get("title", "Unknown Recipe"),
-                "source": result.get("href", "No source available")
-            })
+            # Extract title and link from the results
+            title = result.get('title', '').split(' | ')[0]  # Clean up title by removing site name
+            link = result.get('link', '')
+            
+            if title and link:
+                recipes.append({
+                    "recipe_name": title.strip(),
+                    "source": link
+                })
         
+        # Ensure we have at least some results
+        if not recipes:
+            return [{"error": "No valid recipes found. Please try a different search."}]
+            
         return recipes
     
     except Exception as e:
         return [{"error": f"An error occurred while searching: {str(e)}"}]
 
+        
 # Initialize tools
 final_answer = FinalAnswerTool()
 web_search = DuckDuckGoSearchTool()