Update app.py

app.py

@@ -36,18 +36,57 @@ 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 (keep these as they were)
-# ... (sanitize_input, is_relevant_topic, redact_sensitive_info, check_response_content, check_confidence)
+# 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)
 
 async def generate_response(prompt):
     try:
-        return await client.text_generation(prompt, max_new_tokens=500, temperature=0.7)
+        response = await client.text_generation(prompt, max_new_tokens=500, temperature=0.7)
+        return 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):
-    # ... (keep the existing post_process_response function)
+    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
 
 # CrewAI setup
 communication_expert_crew = Agent(
@@ -91,24 +130,28 @@ async def zerodha_support(message, history, username):
                 verbose=2
             )
 
-            rephrased_query = crew.kickoff()
+            rephrased_query = await crew.kickoff()
         except Exception as e:
             logger.error(f"Error in CrewAI rephrasing: {e}")
             rephrased_query = sanitized_message  # Fallback to original message if rephrasing fails
 
         # 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()
+        try:
+            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 = await crew.kickoff()
+        except Exception as e:
+            logger.error(f"Error in CrewAI response generation: {e}")
+            response = "I apologize, but I'm having trouble generating a response at the moment. Please try again later."
 
         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."