Update app.py

app.py

@@ -6,21 +6,43 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+import spacy
 
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def recipe_engine(query: str) -> str:
-    """
-    Allows you to find the best recipe from local database. Do not search over internet. Returns a string representation of the result.
-
+def diagnose_disease(symptoms: str) -> str:
+    """Fetches health articles based on symptoms and extracts possible diseases.
+    
     Args:
-        query: The query to perform. This should be a string.
+        symptoms: A comma-separated list of symptoms (e.g., 'fever, headache, sore throat').
     """
-    output = "This is my test tools"
+    try:
+        search_tool = DuckDuckGoSearchTool()
+        query = f"Possible diseases for symptoms: {symptoms}"
+        search_results = search_tool.forward(query)
+
+        if not search_results:
+            return "No relevant medical information found. Please consult a doctor."
+
+        articles = search_results.split("\n\n")[:10]  # Get top 10 articles
+        possible_diseases = set()
+        
+        for article in articles:
+            doc = nlp(article)
+            for ent in doc.ents:
+                if ent.label_ == "DISEASE":
+                    possible_diseases.add(ent.text)
+
+        if not possible_diseases:
+            return "No specific diseases could be identified. Consider seeking medical advice."
+
+        return f"Based on your symptoms, possible conditions could be: {', '.join(possible_diseases)}."
+
+    except Exception as e:
+        return f"An error occurred: {e}"
 
-    return output
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: