Update app.py

app.py

@@ -10,14 +10,17 @@ from sklearn.naive_bayes import MultinomialNB
 import asyncio
 from crewai import Agent as CrewAgent, Task, Crew
 import autogen
-import openai
+from langchain_openai import ChatOpenAI
 
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
-# Set OpenAI API key from environment variable
-openai.api_key = os.environ.get('OPENAI_API_KEY')
+# Check for OpenAI API key
+if 'OPENAI_API_KEY' not in os.environ:
+    logger.error("OPENAI_API_KEY environment variable is not set.")
+    logger.info("Please set the OPENAI_API_KEY environment variable before running this script.")
+    sys.exit(1)
 
 # Initialize the client with the Mistral-7B-Instruct-v0.2 model
 try:
@@ -92,13 +95,15 @@ def post_process_response(response):
     return response
 
 # CrewAI and AutoGen setup
+chat_model = ChatOpenAI(model="gpt-3.5-turbo")
+
 communication_expert_crew = CrewAgent(
     role='Communication Expert',
     goal='Interpret and rephrase user queries with empathy and respect',
     backstory="""You are an expert in communication, specializing in understanding and rephrasing queries to ensure they are interpreted in the most positive and constructive light. Your role is crucial in setting the tone for respectful and empathetic interactions.""",
     verbose=True,
     allow_delegation=False,
-    tools=[generate_response]
+    llm=chat_model
 )
 
 response_expert_crew = CrewAgent(
@@ -107,11 +112,11 @@ response_expert_crew = CrewAgent(
     backstory="""You are an expert in Zerodha's services and policies, with a keen ability to provide comprehensive and empathetic responses. Your role is to ensure that all user queries are addressed accurately while maintaining a respectful and supportive tone.""",
     verbose=True,
     allow_delegation=False,
-    tools=[generate_response]
+    llm=chat_model
 )
 
 llm_config = {
-    "config_list": [{"model": "gpt-3.5-turbo", "api_key": os.environ.get('OPENAI_API_KEY')}]
+    "config_list": [{"model": "gpt-3.5-turbo"}]
 }
 
 communication_expert_autogen = autogen.AssistantAgent(
@@ -180,39 +185,4 @@ async def zerodha_support(message, history):
             )
             return response_expert_autogen.last_message()["content"]
 
-        response = await get_autogen_response()
-
-        if not check_response_content(response):
-            response += "\n\nPlease note that I cannot provide specific investment advice or guarantee returns. For personalized guidance, please consult with a qualified financial advisor."
-
-        if not check_confidence(response):
-            return "I apologize, but I'm not confident in providing an accurate answer to this query. For the most up-to-date and accurate information, please contact Zerodha's customer support directly."
-
-        final_response = post_process_response(response)
-
-        return final_response
-    except Exception as e:
-        logger.error(f"Error in zerodha_support: {e}")
-        return "I apologize, but an error occurred while processing your request. Please try again later."
-
-# Gradio interface setup
-demo = gr.ChatInterface(
-    zerodha_support,
-    chatbot=gr.Chatbot(height=600),
-    textbox=gr.Textbox(placeholder="Ask your question about Zerodha here...", container=False, scale=7),
-    title="Zerodha Support Assistant",
-    description="Ask questions about Zerodha's services, trading, account management, and more. Our multi-agent system ensures respectful and empathetic responses.",
-    theme="soft",
-    examples=[
-        "How do I open a Zerodha account?",
-        "I'm frustrated with the recent changes to the Kite platform. Can you help?",
-        "What are the risks involved in F&O trading?",
-        "I think there's an error in my account statement. What should I do?",
-        "Can you explain Zerodha's policy on intraday trading margins?",
-        "I'm new to investing. What resources does Zerodha offer for beginners?",
-        "How does Zerodha ensure the security of my investments and personal data?"
-    ],
-)
-
-if __name__ == "__main__":
-    demo.launch()
+        res