Update app.py

app.py

@@ -8,7 +8,8 @@ import numpy as np
 from sklearn.feature_extraction.text import CountVectorizer
 from sklearn.naive_bayes import MultinomialNB
 import asyncio
-from crewai import Agent as CrewAgent, Task, Crew
+from crewai import Agent, Task, Crew
+import random
 
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -35,38 +36,8 @@ Key points about Zerodha:
 
 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)
-
-approved_topics = ['account opening', 'trading', 'fees', 'platforms', 'funds', 'regulations', 'support']
-vectorizer = CountVectorizer()
-classifier = MultinomialNB()
-
-X = vectorizer.fit_transform(approved_topics)
-y = np.arange(len(approved_topics))
-classifier.fit(X, y)
-
-def is_relevant_topic(query):
-    query_vector = vectorizer.transform([query])
-    prediction = classifier.predict(query_vector)
-    return prediction[0] in range(len(approved_topics))
-
-def redact_sensitive_info(text):
-    text = re.sub(r'\b\d{10,12}\b', '[REDACTED]', text)
-    text = re.sub(r'[A-Z]{5}[0-9]{4}[A-Z]', '[REDACTED]', text)
-    return text
-
-def check_response_content(response):
-    unauthorized_patterns = [
-        r'\b(guarantee|assured|certain)\b.*\b(returns|profit)\b',
-        r'\b(buy|sell)\b.*\b(specific stocks?|shares?)\b'
-    ]
-    return not any(re.search(pattern, response, re.IGNORECASE) for pattern in unauthorized_patterns)
-
-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)
+# Guardrail functions (keep these as they were)
+# ... (sanitize_input, is_relevant_topic, redact_sensitive_info, check_response_content, check_confidence)
 
 async def generate_response(prompt):
     try:
@@ -76,18 +47,10 @@ async def generate_response(prompt):
         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
+    # ... (keep the existing post_process_response function)
 
 # CrewAI setup
-communication_expert_crew = CrewAgent(
+communication_expert_crew = Agent(
     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.""",
@@ -96,7 +59,7 @@ communication_expert_crew = CrewAgent(
     tools=[generate_response]
 )
 
-response_expert_crew = CrewAgent(
+response_expert_crew = Agent(
     role='Response Expert',
     goal='Provide accurate, helpful, and emotionally intelligent responses to user queries',
     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.""",
@@ -106,7 +69,7 @@ response_expert_crew = CrewAgent(
 )
 
 # Main function
-async def zerodha_support(message, history):
+async def zerodha_support(message, history, username):
     try:
         sanitized_message = sanitize_input(message)
         
@@ -133,12 +96,19 @@ async def zerodha_support(message, history):
             logger.error(f"Error in CrewAI rephrasing: {e}")
             rephrased_query = sanitized_message  # Fallback to original message if rephrasing fails
 
-        # Generate response using Mistral model
-        try:
-            response = await generate_response(f"{SHARED_CONTEXT}\n\nUser: {rephrased_query}\nAssistant:")
-        except Exception as e:
-            logger.error(f"Error in response generation: {e}")
-            response = "I apologize, but I'm having trouble generating a response at the moment. Please try again later."
+        # Generate response using Response Expert
+        response_task = Task(
+            description=f"Provide an accurate and helpful response to the user query: '{rephrased_query}'",
+            agent=response_expert_crew
+        )
+
+        crew = Crew(
+            agents=[response_expert_crew],
+            tasks=[response_task],
+            verbose=2
+        )
+
+        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."
@@ -153,24 +123,98 @@ async def zerodha_support(message, history):
         logger.error(f"Error in zerodha_support: {e}")
         return "I apologize, but an error occurred while processing your request. Please try again later."
 
+# Custom CSS for better styling
+custom_css = """
+.container {
+    max-width: 800px;
+    margin: auto;
+    padding: 20px;
+}
+.title {
+    text-align: center;
+    color: #387EF5;
+    font-size: 32px;
+    margin-bottom: 20px;
+}
+.description {
+    text-align: center;
+    color: #555;
+    margin-bottom: 30px;
+}
+.chatbot {
+    border: 1px solid #ddd;
+    border-radius: 10px;
+    overflow: hidden;
+}
+.user-info {
+    background-color: #f0f0f0;
+    padding: 10px;
+    border-radius: 5px;
+    margin-bottom: 20px;
+}
+"""
+
+# Placeholder function for user authentication (you'd implement this with your backend)
+def authenticate(username, password):
+    # This is a mock authentication. In a real scenario, you'd verify against a database.
+    return username == "demo" and password == "password"
+
+# Function to generate a random support ticket number
+def generate_ticket_number():
+    return f"ZRD-{random.randint(100000, 999999)}"
+
 # 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?"
-    ],
-)
+with gr.Blocks(css=custom_css) as demo:
+    gr.HTML("<div class='title'>Zerodha Support Assistant</div>")
+    gr.HTML("<div class='description'>Ask questions about Zerodha's services, trading, account management, and more.</div>")
+    
+    with gr.Tab("Login"):
+        username_input = gr.Textbox(label="Username")
+        password_input = gr.Textbox(label="Password", type="password")
+        login_button = gr.Button("Login")
+        login_message = gr.Textbox(label="Login Status", interactive=False)
+    
+    with gr.Tab("Support Chat"):
+        with gr.Row():
+            with gr.Column(scale=2):
+                chatbot = gr.Chatbot(height=400)
+                message_input = gr.Textbox(label="Your Message", placeholder="Type your question here...")
+                submit_button = gr.Button("Send")
+            
+            with gr.Column(scale=1):
+                with gr.Box():
+                    gr.HTML("<div class='user-info'>")
+                    user_display = gr.Textbox(label="Logged in as", interactive=False)
+                    ticket_number = gr.Textbox(label="Support Ticket", value=generate_ticket_number(), interactive=False)
+                    gr.HTML("</div>")
+                
+                gr.HTML("<h4>Quick Links</h4>")
+                gr.HTML("<ul><li><a href='https://zerodha.com/varsity/' target='_blank'>Zerodha Varsity</a></li><li><a href='https://console.zerodha.com/' target='_blank'>Zerodha Console</a></li></ul>")
+    
+    # Example queries
+    gr.Examples(
+        examples=[
+            "How do I open a Zerodha account?",
+            "What are the brokerage charges for intraday trading?",
+            "How can I withdraw funds from my Zerodha account?",
+            "What is the margin requirement for F&O trading?",
+            "How do I use the Kite mobile app?",
+        ],
+        inputs=message_input,
+    )
+
+    # Login logic
+    def login(username, password):
+        if authenticate(username, password):
+            return "Login successful!", gr.Tabs.update(selected="Support Chat"), username
+        else:
+            return "Login failed. Please try again.", gr.Tabs.update(selected="Login"), ""
+
+    login_button.click(login, inputs=[username_input, password_input], outputs=[login_message, demo.tabs, user_display])
+
+    # Chat logic
+    submit_button.click(zerodha_support, inputs=[message_input, chatbot, user_display], outputs=chatbot)
+    message_input.submit(zerodha_support, inputs=[message_input, chatbot, user_display], outputs=chatbot)
 
 if __name__ == "__main__":
     demo.launch()