Update app.py

app.py

@@ -186,6 +186,54 @@ def generate_checklist_and_suggestions(role, project_id, milestones, reflection)
     
     return checklist, suggestions
 
+# Function to upload the report and create the Supervisor AI Coaching record in Salesforce
+def upload_report_and_create_supervisor_ai_coaching(supervisor_name, project_id, checklist, suggestions, pdf_path, pdf_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')
+        )
+
+        # Upload the PDF file to Salesforce as Content Version
+        with open(pdf_path, "rb") as f:
+            encoded = base64.b64encode(f.read()).decode()
+
+        content = sf.ContentVersion.create({
+            'Title': pdf_name,
+            'PathOnClient': pdf_name,
+            'VersionData': encoded
+        })
+
+        content_id = content['id']
+        download_url = f"https://{sf.sf_instance}/sfc/servlet.shepherd/version/download/{content_id}"
+
+        # Create a Supervisor AI Coaching record
+        query = sf.query(f"SELECT Id FROM Supervisor__c WHERE Name = '{supervisor_name}' LIMIT 1")
+        supervisor_id = query['records'][0]['Id'] if query['totalSize'] > 0 else None
+        if not supervisor_id:
+            return "Supervisor not found."
+
+        project_query = sf.query(f"SELECT Id FROM Project__c WHERE Name = '{project_id}' LIMIT 1")
+        project_id_sf = project_query['records'][0]['Id'] if project_query['totalSize'] > 0 else None
+        if not project_id_sf:
+            return "Project not found."
+
+        # Create Supervisor AI Coaching record with all necessary fields
+        sf.Supervisor_AI_Coaching__c.create({
+            'Project_ID__c': project_id_sf,
+            'Supervisor_ID__c': supervisor_id,
+            'Daily_Checklist__c': checklist,
+            'Suggested_Tips__c': suggestions,
+            'Download_Link__c': download_url
+        })
+
+        return "Supervisor AI Coaching record created and report uploaded successfully."
+
+    except Exception as e:
+        return f"Error: {str(e)}"
+
 # Gradio interface
 def create_interface():
     roles = get_roles_from_salesforce()  # Get roles from Salesforce dynamically
@@ -216,11 +264,12 @@ def create_interface():
         def handle_generate(role, supervisor_name, project_id, milestones, reflection):
             checklist, suggestions = generate_checklist_and_suggestions(role, project_id, milestones, reflection)
             pdf_path, pdf_name = save_report_as_pdf(role, supervisor_name, project_id, checklist, suggestions)
-            return checklist, suggestions, pdf_path, pdf_name
+            supervisor_ai_coaching_response = upload_report_and_create_supervisor_ai_coaching(supervisor_name, project_id, checklist, suggestions, pdf_path, pdf_name)
+            return checklist, suggestions, pdf_path, pdf_name, supervisor_ai_coaching_response
 
         generate.click(fn=handle_generate,
                        inputs=[role, supervisor_name, project_id, milestones, reflection],
-                       outputs=[checklist_output, suggestions_output, download_button, pdf_link])
+                       outputs=[checklist_output, suggestions_output, download_button, pdf_link, gr.HTML()])
 
         clear.click(fn=lambda: ("", "", "", "", ""),
                     inputs=None,