Update app.py

app.py

@@ -75,8 +75,13 @@ class ResponseExpertAgent(Agent):
         "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):
-        response = await hf_client.text_generation(rephrased_query, max_new_tokens=500, temperature=0.7)
-        return response['generated_text']
+        try:
+            logger.info(f"Sending query for generation: {rephrased_query}")
+            response = await hf_client.text_generation(rephrased_query, max_new_tokens=500, temperature=0.7)
+            return response['generated_text']
+        except Exception as e:
+            logger.error(f"Failed to generate text due to: {str(e)}")
+            return "Error in generating response. Please try again."
 
 class PostprocessingAgent(Agent):
     role: Literal["Postprocessing Expert"] = "Postprocessing Expert"