Update app.py

app.py

@@ -28,10 +28,11 @@ if tokenizer.pad_token is None:
     tokenizer.pad_token_id = tokenizer.convert_tokens_to_ids(tokenizer.pad_token)
 model.config.pad_token_id = tokenizer.pad_token_id
 
-# Refined Prompt
+# Refined Prompt to generate day-by-day tasks based on milestones
 PROMPT_TEMPLATE = """You are an AI assistant for construction supervisors. Given the role, project, milestones, and a reflection log, generate:
 
-1. A Daily Checklist with at least 4 items based on the role and milestones.
+1. A Daily Checklist with clear and concise tasks based on the role and milestones. 
+   Split the checklist into day-by-day tasks for a specified time period (e.g., one week).
 2. Focus Suggestions based on concerns or keywords in the reflection log. Provide at least 2 suggestions.
 
 Inputs:
@@ -41,9 +42,14 @@ Milestones: {milestones}
 Reflection Log: {reflection}
 
 Output Format:
-Checklist:
-- 
-
+Checklist (Day-by-Day):
+- Day 1: 
+   - Task 1
+   - Task 2
+- Day 2: 
+   - Task 1
+   - Task 2
+...
 Suggestions:
 - 
 """
@@ -190,7 +196,7 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
         with torch.no_grad():
             outputs = model.generate(
                 inputs['input_ids'],
-                max_new_tokens=100,  # Limit output length to avoid delays
+                max_new_tokens=150,  # Increased max tokens to capture more content
                 no_repeat_ngram_size=2,
                 do_sample=True,
                 top_p=0.9,
@@ -269,4 +275,3 @@ def create_interface():
 if __name__ == "__main__":
     app = create_interface()
     app.launch()
-