Update app.py

app.py

@@ -4,19 +4,17 @@ from langchain_community.tools import DuckDuckGoSearchRun
 from langchain_google_genai import ChatGoogleGenerativeAI
 import os
 import google.generativeai as genai
-from langchain.tools import BaseTool  # Import BaseTool
+from langchain.tools import BaseTool
 
 
-# Configure Gemini API Key (replace with your actual key)
-# Use Streamlit secrets for secure storage.
+# Configure Gemini API Key
 GOOGLE_API_KEY = st.secrets["GOOGLE_API_KEY"]
 genai.configure(api_key=GOOGLE_API_KEY)
 
 
-# Define Tools - Corrected: Wrap the tool in a list if it's a single tool
+# Define Tools
 search_tool = DuckDuckGoSearchRun()
 
-
 class ToolWrapper(BaseTool):
     """Wrapper for tools to make them compatible with CrewAI."""
 
@@ -31,7 +29,7 @@ class ToolWrapper(BaseTool):
         return await self.func(*args, **kwargs)
 
 
-# Define Agents - Corrected: Use ToolWrapper
+# Define Agents -  Specify the 'provider' in ChatGoogleGenerativeAI
 inventory_specialist = Agent(
     role="Inventory Specialist",
     goal="Identify the best matching homes from current inventory based on customer criteria.",
@@ -48,10 +46,10 @@ inventory_specialist = Agent(
             func=search_tool.run,
             description=search_tool.description,
         )
-    ],  # Corrected Tool Usage
+    ],
     llm=ChatGoogleGenerativeAI(
-        model="gemini-pro", temperature=0.7, convert_system_message_to_human=True
-    ),  # Use Gemini model
+        model="gemini-pro", temperature=0.7, convert_system_message_to_human=True, client=genai
+    ),  # Corrected: Pass genai client
 )
 
 communication_specialist = Agent(
@@ -63,14 +61,14 @@ communication_specialist = Agent(
     You use persuasive language and a friendly tone to create a positive customer experience.  You are a world-class copywriter.
     """,
     verbose=True,
-    allow_delegation=True,  # Allow delegation for potential follow-up tasks
+    allow_delegation=True,
     llm=ChatGoogleGenerativeAI(
-        model="gemini-pro", temperature=0.7, convert_system_message_to_human=True
-    ),  # Use Gemini model
+        model="gemini-pro", temperature=0.7, convert_system_message_to_human=True, client=genai
+    ),  # Corrected: Pass genai client
 )
 
 
-# Define Tasks - ADDED expected_output
+# Define Tasks
 def create_inventory_task(customer_request):
     return Task(
         description=f"""Analyze the customer's request: '{customer_request}'.
@@ -80,7 +78,7 @@ def create_inventory_task(customer_request):
         Do NOT make up any information.  If you don't find enough information, say so, but still provide the best possible matches based on what you *can* find.
         """,
         agent=inventory_specialist,
-        expected_output="A list of 3-5 homes with their features and prices, or a message indicating no matches found.",  # Added expected_output
+        expected_output="A list of 3-5 homes with their features and prices, or a message indicating no matches found.",
     )
 
 
@@ -99,7 +97,7 @@ def create_communication_task(inventory_results):
         - Include a signature with contact information (e.g., name, title, phone number, email address).  Use placeholder information for the signature (like "John Doe, Real Estate Agent, (555) 123-4567, [email protected]").
         """,
         agent=communication_specialist,
-        expected_output="A complete, formatted email ready to be sent to the customer.",  # Added expected_output
+        expected_output="A complete, formatted email ready to be sent to the customer.",
     )
 
 
@@ -128,14 +126,14 @@ if st.button("Submit Inquiry"):
             inventory_task = create_inventory_task(customer_request)
             communication_task = create_communication_task(
                 inventory_task.output
-            )  # Pass output for sequential execution
+            )
 
             # Create Crew
             real_estate_crew = Crew(
                 agents=[inventory_specialist, communication_specialist],
                 tasks=[inventory_task, communication_task],
-                process=Process.sequential,  # Important: Run tasks in sequence
-                verbose=True,  # Increase verbosity for debugging
+                process=Process.sequential,
+                verbose=2,
             )
 
             # Run the Crew
@@ -143,7 +141,7 @@ if st.button("Submit Inquiry"):
                 results = real_estate_crew.kickoff()
                 st.success("Response generated!")
                 st.write("**Draft Email Response:**")
-                st.write(results)  # Display the final output (email)
+                st.write(results)
                 st.balloons()
             except Exception as e:
                 st.error(f"An error occurred: {e}")