Update app.py

app.py

@@ -56,8 +56,8 @@ classifier.fit(X, np.arange(len(approved_topics)))
 class CommunicationExpertAgent(Agent):
     role: Literal["Communication Expert"] = "Communication Expert"
     goal: Literal["To interpret and rephrase user queries with empathy and respect"] = "To interpret and rephrase user queries with empathy and respect"
-    backstory: Literal["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."] = \
-        "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."
+    backstory: Literal["""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."""] = \
+        """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."""
 
     async def run(self, query):
         sanitized_query = re.sub(r'[<>&\']', '', query)
@@ -79,8 +79,8 @@ class CommunicationExpertAgent(Agent):
 class ResponseExpertAgent(Agent):
     role: Literal["Response Expert"] = "Response Expert"
     goal: Literal["To provide accurate, helpful, and emotionally intelligent responses to user queries"] = "To provide accurate, helpful, and emotionally intelligent responses to user queries"
-    backstory: Literal["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."] = \
-        "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."
+    backstory: Literal["""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."""] = \
+        """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."""
 
     async def run(self, rephrased_query):
         try:
@@ -94,8 +94,8 @@ class ResponseExpertAgent(Agent):
 class PostprocessingAgent(Agent):
     role: Literal["Postprocessing Expert"] = "Postprocessing Expert"
     goal: Literal["To enhance and finalize responses ensuring quality and completeness"] = "To enhance and finalize responses ensuring quality and completeness"
-    backstory: Literal["You are responsible for finalizing communications, adding polite terminations, and ensuring that the responses meet the quality standards expected in customer interactions."] = \
-        "You are responsible for finalizing communications, adding polite terminations, and ensuring that the responses meet the quality standards expected in customer interactions."
+    backstory: Literal["""You are responsible for finalizing communications, adding polite terminations, and ensuring that the responses meet the quality standards expected in customer interactions."""] = \
+        """You are responsible for finalizing communications, adding polite terminations, and ensuring that the responses meet the quality standards expected in customer interactions."""
 
     def run(self, response):
         processed_response = self.add_disclaimers(response)
@@ -114,4 +114,97 @@ class PostprocessingAgent(Agent):
 
 class RelationshipManagerAgent(Agent):
     role: Literal["Relationship Manager"] = "Relationship Manager"
-    goal: Literal["To elevate
+    goal: Literal["To elevate communication to embody top 1% customer service qualities"] = "To elevate communication to embody top 1% customer service qualities"
+    backstory: Literal["""You are an expert in elite-level communication, capable of transforming interactions to build instant rapport, trust, and confidence. Your role is to enhance both user queries and system responses to create a warm, empathetic, and highly professional interaction."""] = \
+        """You are an expert in elite-level communication, capable of transforming interactions to build instant rapport, trust, and confidence. Your role is to enhance both user queries and system responses to create a warm, empathetic, and highly professional interaction."""
+
+    async def rephrase_query(self, query: str) -> str:
+        rephrased_query = f"I understand that you're asking about {query}. "
+        rephrased_query += "Your question is important, and I'm here to provide you with the best possible assistance. "
+        rephrased_query += "To ensure I fully grasp your situation and can offer the most helpful and accurate information, "
+        rephrased_query += f"could you please confirm if you want to know about {query}? "
+        rephrased_query += "Feel free to add any additional details that might help me serve you better."
+        return rephrased_query
+
+    async def enhance_response(self, response: str) -> str:
+        enhanced_response = "I appreciate you bringing this to our attention. "
+        enhanced_response += response
+        enhanced_response += "\n\nI hope this information effectively addresses your query. "
+        enhanced_response += "Is there anything else about this matter that you'd like me to clarify or expand on? "
+        enhanced_response += "I'm here to ensure you have all the information you need to make informed decisions about your investments and interactions with Zerodha."
+        return enhanced_response
+
+    async def run(self, query: str, response: str = "") -> Tuple[str, str]:
+        rephrased_query = await self.rephrase_query(query)
+        if response:
+            enhanced_response = await self.enhance_response(response)
+            return rephrased_query, enhanced_response
+        return rephrased_query, ""
+
+# Instantiate agents
+communication_expert = CommunicationExpertAgent()
+response_expert = ResponseExpertAgent()
+postprocessing_agent = PostprocessingAgent()
+relationship_manager = RelationshipManagerAgent()
+
+async def handle_query(query):
+    try:
+        if not query.strip():
+            return "I'm sorry, but I didn't receive any query. Could you please ask a question about Zerodha's services?"
+        
+        # Step 1: Rephrase the query using RelationshipManagerAgent
+        rephrased_query, _ = await relationship_manager.run(query)
+        
+        # Step 2: Process the rephrased query with CommunicationExpertAgent
+        communication_expert_query = await communication_expert.run(rephrased_query)
+        
+        # Step 3: Generate response using ResponseExpertAgent
+        initial_response = await response_expert.run(communication_expert_query)
+        
+        # Step 4: Postprocess the response
+        postprocessed_response = postprocessing_agent.run(initial_response)
+        
+        # Step 5: Enhance the response using RelationshipManagerAgent
+        _, final_response = await relationship_manager.run(query, postprocessed_response)
+        
+        return final_response
+    except Exception as e:
+        logger.error(f"Error in handle_query: {str(e)}")
+        return "I apologize, but an error occurred while processing your request. Please try again or contact Zerodha support if the issue persists."
+
+# Gradio interface setup
+def setup_interface():
+    with gr.Blocks() as app:
+        gr.Markdown("# Zerodha Support Chatbot")
+        gr.Markdown("Ask questions about Zerodha's services, trading, account management, and more.")
+        
+        with gr.Row():
+            query_input = gr.Textbox(label="Enter your query", placeholder="Type your question here...")
+            submit_button = gr.Button("Submit")
+        
+        response_output = gr.Textbox(label="Response", lines=10)
+        
+        submit_button.click(
+            fn=lambda x: asyncio.run(handle_query(x)),
+            inputs=[query_input],
+            outputs=[response_output]
+        )
+        
+        gr.Examples(
+            examples=[
+                "How do I open a Zerodha account?",
+                "What are the brokerage charges for intraday trading?",
+                "Can you explain how to use the Kite platform?",
+                "I'm having trouble logging into my account. What should I do?",
+                "What are the margin requirements for F&O trading?"
+            ],
+            inputs=[query_input]
+        )
+    
+    return app
+
+app = setup_interface()
+
+if __name__ == "__main__":
+    app.launch()
+```