Update app.py

app.py

@@ -4,10 +4,24 @@ import os
 from datetime import datetime
 import json
 from crewai import Agent, Task, Crew
-from langchain.tools import DuckDuckGoSearchRun
-from langchain.llms import OpenAI
+from langchain.tools import BaseTool
+from langchain_community.tools import DuckDuckGoSearchRun
+from langchain_openai import OpenAI
+
+class SearchTool(BaseTool):
+    name = "Search"
+    description = "Search the internet for information"
+    
+    def __init__(self):
+        super().__init__()
+        self.search = DuckDuckGoSearchRun()
+    
+    def _run(self, query: str) -> str:
+        return self.search.run(query)
+        
+    def _arun(self, query: str) -> str:
+        raise NotImplementedError("Async not implemented")
 
-# Set up OpenAI API key from Streamlit secrets
 def setup_api_keys():
     """Setup API keys from Hugging Face Spaces secrets"""
     try:
@@ -21,13 +35,12 @@ def setup_api_keys():
         st.error(f"Error setting up API keys: {str(e)}")
         return False
 
-# Initialize agents with API key check
 def initialize_agents():
     """Initialize CrewAI agents with proper API key handling"""
     if not os.environ.get('OPENAI_API_KEY'):
         return None
     
-    search_tool = DuckDuckGoSearchRun()
+    search_tool = SearchTool()
     llm = OpenAI(temperature=0.7)
     
     quote_curator = Agent(
@@ -59,6 +72,8 @@ def initialize_agents():
     
     return quote_curator, content_creator, content_validator
 
+
+
 def main():
     st.set_page_config(
         page_title="Spiritual Content Generator",