Update model.py

model.py

@@ -33,7 +33,7 @@ def call_ai_model_for_insights(input_data: Dict, delay_risk: float) -> List[str]
     """
     model_name = "t5-small"
     max_retries = 3
-    retry_delay = 15  # seconds
+    retry_delay = 20  # Increased for network stability
     
     # Log system resources if psutil is available
     if psutil:
@@ -56,28 +56,25 @@ def call_ai_model_for_insights(input_data: Dict, delay_risk: float) -> List[str]
             
             logger.info("Model loaded successfully. Generating insights...")
             prompt = f"""
-            Summarize the following project delay risk data into 2-4 concise insights or mitigation strategies as a list:
-            Project: {input_data.get('project_name', 'Unnamed Project')}
+            Summarize delay risk data into 2-3 concise insights:
+            Project: {input_data.get('project_name', 'Unnamed')}
             Phase: {input_data.get('phase', '')}
             Task: {input_data.get('task', '')}
-            Expected Duration: {input_data.get('task_expected_duration', 0)} days
-            Actual Duration: {input_data.get('task_actual_duration', 0)} days
-            Current Progress: {input_data.get('current_progress', 0)}%
+            Progress: {input_data.get('current_progress', 0)}%
             Workforce Gap: {input_data.get('workforce_gap', 0)}%
-            Workforce Skill Level: {input_data.get('workforce_skill_level', '').lower()}
-            Shift Hours: {input_data.get('workforce_shift_hours', 0)} hours
-            Weather Impact Score: {input_data.get('weather_impact_score', 0)} (Condition: {get_weather_condition(input_data.get('weather_impact_score', 0))})
-            Calculated Delay Risk: {delay_risk:.1f}%
+            Skill: {input_data.get('workforce_skill_level', '').lower()}
+            Weather Score: {input_data.get('weather_impact_score', 0)}
+            Risk: {delay_risk:.1f}%
 
-            Format the response as a list, e.g., ["Insight 1", "Insight 2"].
+            Format as a list, e.g., ["Insight 1", "Insight 2"].
             """
             
             with torch.no_grad():
-                inputs = tokenizer(prompt, return_tensors="pt", max_length=64, truncation=True).to("cpu")
+                inputs = tokenizer(prompt, return_tensors="pt", max_length=32, truncation=True).to("cpu")
                 outputs = model.generate(
                     **inputs,
-                    max_new_tokens=20,
-                    num_beams=2,
+                    max_new_tokens=15,
+                    num_beams=1,
                     temperature=0.7,
                     do_sample=True
                 )
@@ -85,7 +82,7 @@ def call_ai_model_for_insights(input_data: Dict, delay_risk: float) -> List[str]
             
             insights = [line.strip() for line in response.split("\n") if line.strip() and line.strip() not in [prompt]]
             logger.info(f"Generated insights: {insights}")
-            return insights[:4] or ["No insights generated; review input data."]
+            return insights[:3] or ["No insights generated; review input data."]
         except Exception as e:
             logger.error(f"Attempt {attempt + 1}/{max_retries} - Model inference failed: {str(e)}")
             if attempt < max_retries - 1:
@@ -95,11 +92,11 @@ def call_ai_model_for_insights(input_data: Dict, delay_risk: float) -> List[str]
                 logger.error("Max retries reached. Using fallback insights.")
                 fallback_insights = []
                 if delay_risk > 75:
-                    fallback_insights.append("High risk detected; allocate additional resources urgently.")
+                    fallback_insights.append("High risk; allocate resources urgently.")
                 elif delay_risk > 50:
-                    fallback_insights.append("Moderate risk; consider extending shift hours or hiring staff.")
+                    fallback_insights.append("Moderate risk; extend shift hours.")
                 if input_data.get('workforce_gap', 0) > 20:
-                    fallback_insights.append("Significant workforce gap; recruit additional workers.")
+                    fallback_insights.append("Workforce gap; recruit workers.")
                 if input_data.get('weather_impact_score', 0) > 50:
                     fallback_insights.append("Adverse weather; prioritize indoor tasks.")
                 return fallback_insights or ["AI model failed to generate insights; check system resources."]