Update app.py

app.py

@@ -17,7 +17,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer.pad_token = tokenizer.eos_token
 model.config.pad_token_id = tokenizer.eos_token_id
 
-# Define a more explicit prompt template with few-shot examples
+# Define a prompt template (no examples, just structured prompt)
 PROMPT_TEMPLATE = """You are an AI coach for construction supervisors. Based on the following inputs, generate a daily checklist, focus suggestions, and a motivational quote. Format your response with clear labels as follows:
 
 Checklist:
@@ -29,38 +29,6 @@ Suggestions:
 Quote:
 - Your motivational quote here
 
-Here are examples of expected outputs:
-
-Example 1:
-Inputs:
-Role: Supervisor
-Project: PROJ-01
-Milestones: Lay foundation, Check materials
-Reflection: Team is on track, but weather might delay work.
-Checklist:
-- Confirm foundation laying schedule by 8 AM.
-- Inspect materials for quality at 10 AM.
-Suggestions:
-- Monitor weather updates and plan contingencies.
-- Ensure team has protective gear for rain.
-Quote:
-- Stay steady—every foundation leads to greater heights!
-
-Example 2:
-Inputs:
-Role: Foreman
-Project: PROJ-02
-Milestones: Install beams, Safety check
-Reflection: Minor delay due to late delivery.
-Checklist:
-- Oversee beam installation starting at 9 AM.
-- Conduct safety check at 1 PM.
-Suggestions:
-- Follow up with supplier to prevent future delays.
-- Review safety protocols with the team.
-Quote:
-- Safety first, progress always—keep the momentum!
-
 Now, generate the checklist, suggestions, and quote for the following inputs:
 
 Inputs:
@@ -68,8 +36,7 @@ Role: {role}
 Project: {project_id}
 Milestones: {milestones}
 Reflection: {reflection}
-
-Generate the response in the same format as the examples above."""
+"""
 
 # Cache reset check
 last_reset = datetime.date.today()
@@ -213,4 +180,4 @@ def create_interface():
 
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch()
+    demo.launch()