Update app.py

app.py

@@ -19,7 +19,7 @@ KPI_FLAG_DEFAULT = os.getenv('KPI_FLAG', 'True') == 'True'
 ENGAGEMENT_SCORE_DEFAULT = float(os.getenv('ENGAGEMENT_SCORE', '85.0'))
 
 # Load model and tokenizer
-model_name = "distilgpt2"  # You can use a larger model here if needed
+model_name = "distilgpt2"
 tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
 model = AutoModelForCausalLM.from_pretrained(model_name, low_cpu_mem_usage=True)
 
@@ -145,7 +145,7 @@ def save_to_salesforce(role, project_id, milestones, reflection, checklist, sugg
     except Exception as e:
         print(f"❌ Error saving to Salesforce: {e}")
 
-# Generate Function
+# Generate Function (Updated)
 def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
     if not all([role, supervisor_name, project_id, milestones, reflection]):
         return "❗ Please fill all fields.", "", ""
@@ -162,35 +162,48 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
         with torch.no_grad():
             outputs = model.generate(
                 inputs['input_ids'],
-                max_new_tokens=150,
+                max_new_tokens=100,  # Reduced for faster generation
                 no_repeat_ngram_size=2,
                 do_sample=True,
                 top_p=0.9,
-                temperature=0.8,
+                temperature=0.7,  # Slightly lowered for more focused output
                 pad_token_id=tokenizer.pad_token_id
             )
         result = tokenizer.decode(outputs[0], skip_special_tokens=True)
     except Exception as e:
         print(f"⚠️ Generation error: {e}")
-        return "Error during text generation", "", ""
+        # Fallback values if generation fails
+        checklist = "- Conduct safety briefing\n- Inspect equipment functionality"
+        suggestions = "- Monitor team coordination\n- Review safety protocols"
+        quote = "Success is the sum of small efforts, repeated day in and day out."
+        save_to_salesforce(role, project_id, milestones, reflection, checklist, suggestions, quote, supervisor_name)
+        return checklist, suggestions, quote
 
     def extract_between(text, start, end):
         s = text.find(start)
         e = text.find(end, s) if end else len(text)
-        return text[s + len(start):e].strip() if s != -1 else "Not found"
+        return text[s + len(start):e].strip() if s != -1 else ""
 
     checklist = extract_between(result, "Checklist:\n", "Suggestions:")
     suggestions = extract_between(result, "Suggestions:\n", "Quote:")
     quote = extract_between(result, "Quote:\n", None)
 
+    # Ensure checklist and suggestions are never blank
+    if not checklist.strip():
+        checklist = "- Conduct safety briefing based on role\n- Verify milestone progress"
+    if not suggestions.strip():
+        suggestions = "- Address concerns from reflection log\n- Focus on team communication"
+
+    # Fallback quote if empty
+    if not quote.strip():
+        quote = "Success is the sum of small efforts, repeated day in and day out."
+
     save_to_salesforce(role, project_id, milestones, reflection, checklist, suggestions, quote, supervisor_name)
     return checklist, suggestions, quote
 
 # Dashboard button function
 def open_dashboard():
-    # Replace this with your actual Salesforce dashboard URL
     salesforce_dashboard_url = "https://your-salesforce-instance.lightning.force.com/lightning/page/home"
-    # Return HTML link that opens in a new tab
     return f'<a href="{salesforce_dashboard_url}" target="_blank" rel="noopener noreferrer" style="font-size:16px;">Open Salesforce Dashboard</a>'
 
 # Interface
@@ -211,12 +224,12 @@ def create_interface():
             generate = gr.Button("Generate")
             clear = gr.Button("Clear")
             refresh = gr.Button("🔄 Refresh Roles")
-            dashboard_btn = gr.Button("Dashboard")  # Dashboard button
+            dashboard_btn = gr.Button("Dashboard")
 
         checklist_output = gr.Textbox(label="✅ Daily Checklist")
         suggestions_output = gr.Textbox(label="💡 Focus Suggestions")
         quote_output = gr.Textbox(label="✨ Motivational Quote")
-        dashboard_link = gr.HTML("")  # For showing the dashboard link
+        dashboard_link = gr.HTML("")
 
         role.change(fn=lambda r: gr.update(choices=get_supervisor_name_by_role(r)), inputs=role, outputs=supervisor_name)
         supervisor_name.change(fn=get_projects_for_supervisor, inputs=supervisor_name, outputs=project_id)
@@ -236,4 +249,4 @@ def create_interface():
 
 if __name__ == "__main__":
     app = create_interface()
-    app.launch()
+    app.launch()