Update app.py

app.py

@@ -1,24 +1,25 @@
 import streamlit as st
-from crewai import Crew, Agent, Task, Process, LLM
+from crewai import Crew, Agent, Task, Process
 from langchain_community.tools import DuckDuckGoSearchRun
-from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_groq import ChatGroq  # Import ChatGroq
 import os
-import google.generativeai as genai
 from langchain.tools import BaseTool
 
+# Configure Groq API Key (replace with your actual key)
+GROQ_API_KEY = st.secrets["GROQ_API_KEY"]
+# No need for genai.configure anymore
 
-# Configure Gemini API Key
-GOOGLE_API_KEY = st.secrets["GOOGLE_API_KEY"]
-genai.configure(api_key=GOOGLE_API_KEY)
 
-# Define the LLM using langchain-google-genai
-# The CrewAI LLM class has been deprecated.  Pass the Langchain LLM directly.
-gemini_llm = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.7, convert_system_message_to_human=True, google_api_key=GOOGLE_API_KEY)
+# Define the LLM using langchain-groq
+groq_llm = ChatGroq(
+    model_name="groq/deepseek-coder-33b-instruct", temperature=0.7, groq_api_key=GROQ_API_KEY
+)
 
 
 # Define Tools
 search_tool = DuckDuckGoSearchRun()
 
+
 class ToolWrapper(BaseTool):
     """Wrapper for tools to make them compatible with CrewAI."""
 
@@ -33,7 +34,7 @@ class ToolWrapper(BaseTool):
         return await self.func(*args, **kwargs)
 
 
-# Define Agents - Pass the Langchain LLM directly
+# Define Agents - Pass the Groq LLM directly
 inventory_specialist = Agent(
     role="Inventory Specialist",
     goal="Identify the best matching homes from current inventory based on customer criteria.",
@@ -51,7 +52,7 @@ inventory_specialist = Agent(
             description=search_tool.description,
         )
     ],
-    llm=gemini_llm,  # Pass the Langchain LLM *directly*
+    llm=groq_llm,  # Pass the Groq LLM directly
 )
 
 communication_specialist = Agent(
@@ -64,7 +65,7 @@ communication_specialist = Agent(
     """,
     verbose=True,
     allow_delegation=True,
-    llm=gemini_llm,  # Pass the Langchain LLM *directly*
+    llm=groq_llm,  # Pass the Groq LLM directly
 )
 
 
@@ -102,7 +103,7 @@ def create_communication_task(inventory_results):
 
 
 # Streamlit App
-st.title("AI-Powered Real Estate Lead Response (Gemini)")
+st.title("AI-Powered Real Estate Lead Response (Groq)")
 
 customer_zipcode = st.text_input("Zip Code:", placeholder="Enter zip code")
 customer_bedrooms = st.number_input(
@@ -146,5 +147,5 @@ if st.button("Submit Inquiry"):
             except Exception as e:
                 st.error(f"An error occurred: {e}")
                 st.write(
-                    "Please check your Gemini API key and ensure it has the necessary permissions."
+                    "Please check your Groq API key and ensure it has the necessary permissions."
                 )