Update researchsimulation/InteractiveInterviewChatbot.py

researchsimulation/InteractiveInterviewChatbot.py

@@ -300,27 +300,48 @@ A culturally authentic and conversational response to the question: '{agent_ques
         )
         logging.debug(f"Crew initialized for agent '{agent_name}' with 1 task and sequential process")
         
-        try:
-            # Start the task
-            crew_output = crew.kickoff()
-            logging.info(f"Task execution completed for agent '{agent_name}'")
-        
-            task_output = question_task.output
-            logging.debug(f"Raw output from agent '{agent_name}': {task_output.raw}")
-        
-            # Collect and format the response
-            if task_output.raw:
-                formatted_response = f"**{agent_name}**: {task_output.raw}"
-                responses.append(formatted_response)
-                logging.info(f"Formatted response from agent '{agent_name}' added to responses")
-            else:
-                fallback_response = f"**{agent_name}**: I wasn't able to answer right now - can you try again?"
-                responses.append(fallback_response)
-                logging.warning(f"No output from agent '{agent_name}'. Added fallback response.")
-        
-        except Exception as e:
-            logging.error(f"Error during task execution for agent '{agent_name}': {str(e)}", exc_info=True)
-            error_response = f"**PreData Moderator**: An error occurred while processing {agent_name}'s response. Please try again."
-            responses.append(error_response)
-    
+        max_attempts = 3
+        attempt = 0
+        validated = False
+        validated_answer = None
+        while attempt < max_attempts and not validated:
+            try:
+                crew_output = crew.kickoff()
+                logging.info(f"Task execution completed for agent '{agent_name}' (attempt {attempt+1})")
+                task_output = question_task.output
+                logging.debug(f"Raw output from agent '{agent_name}': {getattr(task_output, 'raw', str(task_output))}")
+                answer = task_output.raw if hasattr(task_output, 'raw') else str(task_output)
+                # Validate the response using validate_response from validation_utils
+                is_valid = validate_response(
+                    question=agent_question,
+                    answer=answer,
+                    user_profile_str=str(user_profile),
+                    fast_facts_str="",  # Add if available
+                    interview_transcript_text="",  # Add if available
+                    respondent_type=agent_name,
+                    ai_evaluator_agent=None,  # Add if available
+                    processor_llm=processor_llm
+                )
+                logging.info(f"Validation result for agent '{agent_name}' (attempt {attempt+1}): {is_valid}")
+                if is_valid:
+                    validated = True
+                    validated_answer = answer
+                    break
+                else:
+                    attempt += 1
+                    logging.warning(f"Response failed validation for agent '{agent_name}' (attempt {attempt}). Retrying...")
+            except Exception as e:
+                logging.error(f"Error during task execution for agent '{agent_name}' (attempt {attempt+1}): {str(e)}", exc_info=True)
+                attempt += 1
+        # --- End validation and retry loop ---
+
+        if validated_answer:
+            formatted_response = f"**{agent_name}**: {validated_answer}"
+            responses.append(formatted_response)
+            logging.info(f"Validated response from agent '{agent_name}' added to responses")
+        else:
+            fallback_response = f"**PreData Moderator**: Unable to pass validation after {max_attempts} attempts for {agent_name}."
+            responses.append(fallback_response)
+            logging.warning(f"No validated output from agent '{agent_name}' after {max_attempts} attempts. Added fallback response.")
+
     return responses