Update app.py

app.py

@@ -94,8 +94,8 @@ def save_report_as_pdf(role, supervisor_name, project_id, checklist, suggestions
 
     return temp_pdf_path, filename
 
-# Function to upload PDF to Salesforce and update the Download_Link__c field
-def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_path, pdf_name, checklist, suggestions):
+# Upload to Salesforce and update record with the generated URL
+def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_path, pdf_name):
     try:
         sf = Salesforce(
             username=os.getenv('SF_USERNAME'),
@@ -103,7 +103,7 @@ def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_pa
             security_token=os.getenv('SF_SECURITY_TOKEN'),
             domain=os.getenv('SF_DOMAIN', 'login')
         )
-
+        
         # Read and encode the file as base64
         with open(pdf_path, "rb") as f:
             encoded = base64.b64encode(f.read()).decode()
@@ -123,10 +123,9 @@ def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_pa
 
         # Query Salesforce to find the specific Supervisor_AI_Coaching__c record
         query = sf.query(f"""
-            SELECT Id, Daily_Checklist__c, Suggested_Tips__c, Download_Link__c 
-            FROM Supervisor_AI_Coaching__c 
+            SELECT Id FROM Supervisor_AI_Coaching__c 
             WHERE Project_ID__c = '{project_id}' 
-            AND Supervisor_Name__c = '{supervisor_name}'
+            AND Name = '{supervisor_name}'
             LIMIT 1
         """)
 
@@ -134,21 +133,10 @@ def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_pa
             # Get the ID of the Supervisor_AI_Coaching__c record
             coaching_id = query['records'][0]['Id']
             
-            # Update the Supervisor_AI_Coaching__c record with the generated checklist, suggestions, and download URL
+            # Update the Supervisor_AI_Coaching__c record with the download URL
             sf.Supervisor_AI_Coaching__c.update(coaching_id, {
-                'Daily_Checklist__c': checklist,  # Update Daily_Checklist__c with the generated checklist
-                'Suggested_Tips__c': suggestions,  # Update Suggested_Tips__c with the generated suggestions
                 'Download_Link__c': download_url  # Update the Download_Link__c field with the URL
             })
-        else:
-            # If record doesn't exist, create a new one
-            sf.Supervisor_AI_Coaching__c.create({
-                'Supervisor_Name__c': supervisor_name,
-                'Project_ID__c': project_id,
-                'Daily_Checklist__c': checklist,  # Set the Daily_Checklist__c field
-                'Suggested_Tips__c': suggestions,  # Set the Suggested_Tips__c field
-                'Download_Link__c': download_url  # Set the Download_Link__c field
-            })
 
         return download_url
 
@@ -156,7 +144,62 @@ def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_pa
         print(f"⚠️ Upload error: {e}")
         return ""
 
-# Function to generate outputs and update Salesforce record
+# Salesforce helpers
+def get_roles_from_salesforce():
+    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')
+        )
+        result = sf.query("SELECT Role__c FROM Supervisor__c WHERE Role__c != NULL")
+        return list(set(record['Role__c'] for record in result['records']))
+    except Exception as e:
+        print(f"⚠️ Error fetching roles: {e}")
+        return []
+
+def get_supervisor_name_by_role(role):
+    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')
+        )
+        result = sf.query(f"SELECT Name FROM Supervisor__c WHERE Role__c = '{role}'")
+        return [record['Name'] for record in result['records']]
+    except Exception as e:
+        print(f"⚠️ Error fetching names: {e}")
+        return []
+
+def get_projects_for_supervisor(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')
+        )
+        result = sf.query(f"SELECT Id FROM Supervisor__c WHERE Name = '{supervisor_name}' LIMIT 1")
+        if result['totalSize'] == 0:
+            return ""
+        supervisor_id = result['records'][0]['Id']
+        project_result = sf.query(f"SELECT Name FROM Project__c WHERE Supervisor_ID__c = '{supervisor_id}' LIMIT 1")
+        return project_result['records'][0]['Name'] if project_result['totalSize'] > 0 else ""
+    except Exception as e:
+        print(f"⚠️ Error fetching project: {e}")
+        return ""
+
+# Generate Salesforce dashboard URL
+def generate_salesforce_dashboard_url(supervisor_name, project_id):
+    return f"https://aicoachforsitesupervisors-dev-ed--c.develop.vf.force.com/apex/DashboardPage?supervisorName={supervisor_name}&projectId={project_id}"
+
+def open_dashboard(role, supervisor_name, project_id):
+    url = generate_salesforce_dashboard_url(supervisor_name, project_id)
+    return f'<a href="{url}" target="_blank">Open Salesforce Dashboard</a>'
+
+# Generate AI output and return the correct file for download
 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.", "", None, ""
@@ -193,7 +236,7 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
         suggestions = "- Monitor team coordination\n- Review safety protocols with the team"
 
     pdf_path, pdf_name = save_report_as_pdf(role, supervisor_name, project_id, checklist, suggestions)
-    pdf_url = upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_path, pdf_name, checklist, suggestions)
+    pdf_url = upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_path, pdf_name)
 
     if pdf_url:
         suggestions += f"\n\n🔗 [Download PDF Report]({pdf_url})"
@@ -201,21 +244,6 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
     # Return the generated checklist, suggestions, and the path to the PDF
     return checklist, suggestions, pdf_path, pdf_name
 
-# Salesforce helper function to get roles from Salesforce
-def get_roles_from_salesforce():
-    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')
-        )
-        result = sf.query("SELECT Role__c FROM Supervisor__c WHERE Role__c != NULL")
-        return list(set(record['Role__c'] for record in result['records']))
-    except Exception as e:
-        print(f"⚠️ Error fetching roles: {e}")
-        return []
-
 # Gradio Interface
 def create_interface():
     roles = get_roles_from_salesforce()
@@ -268,3 +296,5 @@ def create_interface():
 if __name__ == "__main__":
     app = create_interface()
     app.launch()
+
+