Update app.py

app.py

@@ -9,7 +9,6 @@ from sklearn.feature_extraction.text import CountVectorizer
 from sklearn.naive_bayes import MultinomialNB
 import asyncio
 from crewai import Agent as CrewAgent, Task, Crew
-import autogen
 import openai
 
 # Set up logging
@@ -32,6 +31,20 @@ except Exception as e:
     logger.error(f"Failed to initialize InferenceClient: {e}")
     sys.exit(1)
 
+# Shared context for agents
+SHARED_CONTEXT = """You are part of a multi-agent system designed to provide respectful, empathetic, and accurate support for Zerodha, a leading Indian financial services company. Your role is crucial in ensuring all interactions uphold the highest standards of customer service while maintaining Zerodha's excellent reputation.
+
+Key points about Zerodha:
+1. India's largest discount broker, known for innovative technology and low-cost trading.
+2. Flat fee structure: ₹20 per executed order for intraday and F&O trades, zero brokerage for delivery equity investments.
+3. Main trading platform: Kite (web and mobile).
+4. Coin platform for commission-free direct mutual fund investments.
+5. Extensive educational resources through Varsity.
+6. Additional tools: Sentinel (price alerts) and ChartIQ (advanced charting).
+7. Console for account management and administrative tasks.
+
+Always prioritize user safety, ethical investing practices, and transparent communication. Never provide information that could mislead users or bring disrepute to Zerodha."""
+
 # Guardrail functions
 def sanitize_input(input_text):
     return re.sub(r'[<>&\']', '', input_text)
@@ -65,23 +78,73 @@ def check_confidence(response):
     uncertain_phrases = ["I'm not sure", "It's possible", "I don't have enough information"]
     return not any(phrase.lower() in response.lower() for phrase in uncertain_phrases)
 
-async def generate_response(prompt):
+# Agent setup using CrewAI only
+communication_expert_crew = CrewAgent(
+    role='Communication Expert',
+    goal='Interpret and rephrase user queries with empathy and respect',
+    backstory=SHARED_CONTEXT,
+    verbose=True,
+    allow_delegation=False
+)
+
+response_expert_crew = CrewAgent(
+    role='Response Expert',
+    goal='Provide accurate, helpful, and emotionally intelligent responses to user queries',
+    backstory=SHARED_CONTEXT,
+    verbose=True,
+    allow_delegation=False
+)
+
+# Main function for handling user queries
+async def zerodha_support(message, history):
     try:
-        return await client.text_generation(prompt, max_new_tokens=500, temperature=0.7)
+        sanitized_message = sanitize_input(message)
+        
+        if not is_relevant_topic(sanitized_message):
+            return "I'm sorry, but I can only assist with queries related to Zerodha's services and trading. Could you please ask a question about your Zerodha account, trading, or our platforms?"
+        
+        sanitized_message = redact_sensitive_info(sanitized_message)
+
+        # CrewAI for rephrasing query
+        rephrase_task = Task(
+            description=f"Rephrase the following user query with empathy and respect: '{sanitized_message}'",
+            agent=communication_expert_crew
+        )
+        
+        crew = Crew(
+            agents=[communication_expert_crew],
+            tasks=[rephrase_task],
+            verbose=2
+        )
+
+        rephrased_query = await crew.kickoff()
+
+        # CrewAI for generating the response
+        response_task = Task(
+            description=f"Generate a comprehensive and empathetic response to: '{rephrased_query}'",
+            agent=response_expert_crew
+        )
+        
+        response_crew = Crew(
+            agents=[response_expert_crew],
+            tasks=[response_task],
+            verbose=2
+        )
+
+        response = await response_crew.kickoff()
+
+        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 generating response: {e}")
-        return "I apologize, but I'm having trouble generating a response at the moment. Please try again later."
-
-def post_process_response(response):
-    response = re.sub(r'\b(stupid|dumb|idiotic|foolish)\b', 'mistaken', response, flags=re.IGNORECASE)
-    
-    if not re.search(r'(Thank you|Is there anything else|Hope this helps|Let me know if you need more information)\s*$', response, re.IGNORECASE):
-        response += "\n\nIs there anything else I can help you with regarding Zerodha's services?"
-    
-    if re.search(r'\b(invest|trade|buy|sell|market)\b', response, re.IGNORECASE):
-        response += "\n\nPlease note that this information is for educational purposes only and should not be considered as financial advice. Always do your own research and consider consulting with a qualified financial advisor before making investment decisions."
-    
-    return response
+        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(