increas estimated time

backend/integrations/ai_integration/mock_ai_generator.py

@@ -416,17 +416,40 @@ class MockAIGenerator(AIGeneratorInterface):
         Returns:
             String response from the AI containing the estimated duration
         """
-        # For the mock implementation, we'll return a simple response with a number
-        # based on the length of the prompt and keywords
+        # For the mock implementation, we'll return a response with a number
+        # based on the length of the prompt and keywords, following the same logic as the AI service
         import random
-        
+        import re
+
+        # Extract information from the prompt to determine duration
         # Count the number of questions mentioned in the prompt
         question_count = prompt.count("Question ")  # Count occurrences of "Question "
         
-        # Estimate duration based on question count (3 minutes per question, with some randomness)
-        estimated_minutes = max(5, question_count * 3 + random.randint(-2, 5))
+        # Extract seniority level from the prompt
+        seniority_match = re.search(r'- Seniority: (\w+)', prompt)
+        seniority = seniority_match.group(1).lower() if seniority_match else 'junior'
+        
+        # Count text-based questions mentioned in the prompt
+        text_questions = prompt.count("(Text-based question requiring written response)")
+        
+        # Calculate base duration (at least 2 minutes per question)
+        base_duration = question_count * 2
+        
+        # Adjust based on job seniority
+        if seniority in ['senior', 'lead']:
+            base_duration = int(base_duration * 1.5)  # 50% more time for senior roles
+        elif seniority in ['mid', 'intermediate']:
+            base_duration = int(base_duration * 1.2)  # 20% more time for mid-level roles
+        # Junior/intern roles get the base time (2 min per question)
         
+        # Adjust based on question complexity (text-based questions take more time)
+        if text_questions > 0:
+            base_duration += text_questions * 2  # Additional 2 minutes per text question
+        
+        # Add some randomness to simulate realistic variations
+        estimated_minutes = max(5, base_duration + random.randint(-1, 3))
+
         # Ensure it's within reasonable bounds
         estimated_minutes = min(180, max(5, estimated_minutes))
-        
+
         return f"{estimated_minutes} minutes"

backend/services/ai_service.py

@@ -106,15 +106,15 @@ def estimate_assessment_duration(title: str, job_info: dict, questions: List[Ass
     prompt = f"""
     Based on the following assessment details, estimate how many minutes a candidate would need to complete this assessment.
     Consider the complexity of the questions and the job requirements.
-    
+
     Assessment Title: {title}
-    
+
     Job Information:
     - Title: {job_info.get('title', 'N/A')}
     - Seniority: {job_info.get('seniority', 'N/A')}
     - Description: {job_info.get('description', 'N/A')}
     - Skill Categories: {', '.join(job_info.get('skill_categories', []))}
-    
+
     Questions Count: {len(questions)}
     """
 
@@ -129,7 +129,11 @@ def estimate_assessment_duration(title: str, job_info: dict, questions: List[Ass
     if additional_note:
         prompt += f"\nAdditional Notes: {additional_note}"
 
-    prompt += "\n\nPlease provide only a number representing the estimated duration in minutes."
+    prompt += """
+\n\nPlease provide only a number representing the estimated duration in minutes.
+Consider that each question should take at least 2 minutes to answer, with additional time for complex questions,
+especially for senior roles or text-based questions requiring detailed responses.
+"""
 
     # Get the AI's estimation
     duration_estimate = ai_generator.estimate_duration(prompt)
@@ -138,12 +142,51 @@ def estimate_assessment_duration(title: str, job_info: dict, questions: List[Ass
     duration_match = re.search(r'\d+', duration_estimate)
     if duration_match:
         duration_minutes = int(duration_match.group())
-        # Ensure the duration is within reasonable bounds (1-180 minutes)
-        duration_minutes = max(1, min(180, duration_minutes))
-        logger.info(f"Estimated duration for assessment '{title}': {duration_minutes} minutes")
+        
+        # Apply our own logic to ensure minimum duration per question and adjust based on job seniority
+        # Calculate base duration (at least 2 minutes per question)
+        base_duration = len(questions) * 2
+        
+        # Adjust based on job seniority
+        seniority = job_info.get('seniority', '').lower()
+        if seniority in ['senior', 'lead']:
+            base_duration = int(base_duration * 1.5)  # 50% more time for senior roles
+        elif seniority in ['mid', 'intermediate']:
+            base_duration = int(base_duration * 1.2)  # 20% more time for mid-level roles
+        # Junior/intern roles get the base time (2 min per question)
+        
+        # Adjust based on question complexity (text-based questions take more time)
+        text_questions = sum(1 for q in questions if q.type == 'text_based')
+        if text_questions > 0:
+            # Add extra time for text-based questions (they typically take longer)
+            base_duration += text_questions * 2  # Additional 2 minutes per text question
+        
+        # Take the maximum of AI estimation and our calculated minimum
+        duration_minutes = max(duration_minutes, base_duration)
+        
+        # Ensure the duration is within reasonable bounds (5-180 minutes)
+        duration_minutes = max(5, min(180, duration_minutes))
+        logger.info(f"Estimated duration for assessment '{title}': {duration_minutes} minutes (AI: {int(duration_match.group())}, calculated min: {base_duration})")
         return duration_minutes
     else:
         # If no number is found in the response, return a default duration based on question count
-        default_duration = min(60, max(5, len(questions) * 3))  # 3 minutes per question, capped at 60
-        logger.warning(f"No duration found in AI response. Using default: {default_duration} minutes")
+        # Calculate base duration (at least 2 minutes per question)
+        base_duration = len(questions) * 2
+        
+        # Adjust based on job seniority
+        seniority = job_info.get('seniority', '').lower()
+        if seniority in ['senior', 'lead']:
+            base_duration = int(base_duration * 1.5)  # 50% more time for senior roles
+        elif seniority in ['mid', 'intermediate']:
+            base_duration = int(base_duration * 1.2)  # 20% more time for mid-level roles
+        
+        # Adjust based on question complexity (text-based questions take more time)
+        text_questions = sum(1 for q in questions if q.type == 'text_based')
+        if text_questions > 0:
+            base_duration += text_questions * 2  # Additional 2 minutes per text question
+        
+        # Ensure minimum duration is at least 5 minutes
+        default_duration = max(5, base_duration)
+        
+        logger.warning(f"No duration found in AI response. Using calculated default: {default_duration} minutes")
         return default_duration