Update app.py

app.py

@@ -111,75 +111,17 @@ def field_exists(sf, object_name, field_name):
         print(f"⚠️ Error checking field {field_name}: {e}")
         return False
 
-def save_to_salesforce(role, project_id, milestones, reflection, checklist, suggestions, supervisor_name):
-    try:
-        sf = Salesforce(
-            username=os.getenv('SF_USERNAME'),
-            password=os.getenv('SF_PASSWORD'),
-            security_token=os.getenv('SF_SECURITY_TOKEN'),
-            domain=os.getenv('SF_DOMAIN', 'login')
-        )
-        supervisor_result = sf.query(f"SELECT Id FROM Supervisor__c WHERE Name = '{supervisor_name}' LIMIT 1")
-        if supervisor_result['totalSize'] == 0:
-            print("❌ No supervisor found.")
-            return
-        supervisor_id = supervisor_result['records'][0]['Id']
-        project_result = sf.query(f"SELECT Id FROM Project__c WHERE Name = '{project_id}' LIMIT 1")
-        if project_result['totalSize'] == 0:
-            print("❌ No project found.")
-            return
-        project_record_id = project_result['records'][0]['Id']
-
-        data = {
-            'Supervisor_ID__c': supervisor_id,
-            'Project_ID__c': project_record_id,
-            'Daily_Checklist__c': checklist[:255],
-            'Suggested_Tips__c': suggestions[:255],
-            'Reflection_Log__c': reflection[:255],
-            'KPI_Flag__c': KPI_FLAG_DEFAULT,
-            'Engagement_Score__c': ENGAGEMENT_SCORE_DEFAULT
-        }
-
-        if field_exists(sf, 'Supervisor_AI_Coaching__c', 'Milestones_KPIs__c'):
-            data['Milestones_KPIs__c'] = milestones[:255]
-
-        sf.Supervisor_AI_Coaching__c.create(data)
-        print("✅ Record created in Salesforce.")
-    except Exception as e:
-        print(f"❌ Error saving to Salesforce: {e}")
-
-# Fallback generation for checklist and suggestions
-def generate_fallback_checklist(role, milestones):
-    checklist_items = []
-    if role.lower() == "safety officer":
-        checklist_items.append("- Conduct morning safety briefing")
-        checklist_items.append("- Review incident reports from previous day")
-    
-    if milestones and milestones.strip():
-        kpis = [kpi.strip() for kpi in milestones.split(",")]
-        for kpi in kpis:
-            checklist_items.append(f"- Ensure progress on {kpi}")
-    else:
-        checklist_items.append("- Perform daily safety inspection")
-    
-    return "\n".join(checklist_items)
-
-def generate_fallback_suggestions(reflection):
-    suggestions_items = []
-    reflection_lower = reflection.lower()
-    if "student" in reflection_lower or "learning" in reflection_lower:
-        suggestions_items.append("- Ensure students are logging incidents consistently")
-        suggestions_items.append("- Provide guidance on timely incident recording")
-    if "incident" in reflection_lower:
-        suggestions_items.append("- Follow up on reported incidents with corrective actions")
-    
-    if not suggestions_items:
-        suggestions_items.append("- Monitor team coordination")
-        suggestions_items.append("- Review safety protocols with the team")
-    
-    return "\n".join(suggestions_items)
-
-# Generate Function
+# New function to generate Salesforce dashboard URL
+def generate_salesforce_dashboard_url(supervisor_name, project_id):
+    return f"https://your-salesforce-instance.lightning.force.com/lightning/r/Project__c/{project_id}/view?supervisorName={supervisor_name}"
+
+# Dashboard button function
+def open_dashboard(role, supervisor_name, project_id):
+    # Generate dynamic URL based on supervisor and project
+    dashboard_url = generate_salesforce_dashboard_url(supervisor_name, project_id)
+    return f'<a href="{dashboard_url}" target="_blank" rel="noopener noreferrer" style="font-size:16px;">Open Salesforce Dashboard</a>'
+
+# Generate function
 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.", ""
@@ -206,10 +148,7 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
         result = tokenizer.decode(outputs[0], skip_special_tokens=True)
     except Exception as e:
         print(f"⚠️ Generation error: {e}")
-        checklist = generate_fallback_checklist(role, milestones)
-        suggestions = generate_fallback_suggestions(reflection)
-        save_to_salesforce(role, project_id, milestones, reflection, checklist, suggestions, supervisor_name)
-        return checklist, suggestions
+        return "", ""
 
     def extract_between(text, start, end):
         s = text.find(start)
@@ -219,19 +158,8 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
     checklist = extract_between(result, "Checklist:\n", "Suggestions:")
     suggestions = extract_between(result, "Suggestions:\n", None)
 
-    if not checklist.strip():
-        checklist = generate_fallback_checklist(role, milestones)
-    if not suggestions.strip():
-        suggestions = generate_fallback_suggestions(reflection)
-
-    save_to_salesforce(role, project_id, milestones, reflection, checklist, suggestions, supervisor_name)
     return checklist, suggestions
 
-# Dashboard button function
-def open_dashboard():
-    salesforce_dashboard_url = "https://your-salesforce-instance.lightning.force.com/lightning/page/home"
-    return f'<a href="{salesforce_dashboard_url}" target="_blank" rel="noopener noreferrer" style="font-size:16px;">Open Salesforce Dashboard</a>'
-
 # Interface
 def create_interface():
     roles = get_roles_from_salesforce()
@@ -268,7 +196,7 @@ def create_interface():
 
         refresh.click(fn=lambda: gr.update(choices=get_roles_from_salesforce()), outputs=role)
 
-        dashboard_btn.click(fn=open_dashboard, inputs=None, outputs=dashboard_link)
+        dashboard_btn.click(fn=open_dashboard, inputs=[role, supervisor_name, project_id], outputs=dashboard_link)
 
     return demo