Update app.py

app.py

@@ -7,10 +7,10 @@ from flask import Flask, jsonify, request, render_template, redirect, url_for
 
 # Configure logging with more detail
 logging.basicConfig(
-    level=logging.DEBUG,  # Set to DEBUG for detailed logs
+    level=logging.DEBUG,
     format='%(asctime)s - %(levelname)s - %(message)s',
     handlers=[
-        logging.StreamHandler()  # Output logs to console
+        logging.StreamHandler()
     ]
 )
 logger = logging.getLogger(__name__)
@@ -52,43 +52,37 @@ def test_salesforce_connection():
             domain=SALESFORCE_DOMAIN
         )
         logger.info("Successfully connected to Salesforce. Session ID: %s", sf.session_id)
-        # Test a simple query to verify object access
         sf.query("SELECT Id FROM Account LIMIT 1")
         logger.info("Successfully queried Account object. Connection is working.")
-        return True, None
+        return True, None, sf
     except SalesforceAuthenticationFailed as e:
         logger.error("Salesforce authentication failed: %s", e)
-        return False, f"Salesforce authentication failed: {str(e)}"
+        return False, f"Salesforce authentication failed: {str(e)}", None
     except Exception as e:
         logger.error("Salesforce connection error: %s", e)
-        return False, f"Salesforce connection error: {str(e)}"
+        return False, f"Salesforce connection error: {str(e)}", None
 
-def fetch_salesforce_input(supervisor_id, project_id):
+def fetch_project_data(project_id):
     """
-    Fetch input data from Salesforce for a given supervisor and project.
+    Fetch data from the Project__c object in Salesforce.
     """
-    try:
-        logger.debug("Connecting to Salesforce to fetch input data")
-        sf = Salesforce(
-            username=SALESFORCE_USERNAME,
-            password=SALESFORCE_PASSWORD,
-            security_token=SALESFORCE_SECURITY_TOKEN,
-            domain=SALESFORCE_DOMAIN
-        )
-        logger.info("Connected to Salesforce for input fetch")
+    success, error, sf = test_salesforce_connection()
+    if not success:
+        return None, error
 
+    try:
+        logger.debug("Fetching Project__c record with Id: %s", project_id)
         query = f"""
             SELECT Supervisor_ID__c, Role__c, Project_ID__c, Weather__c, Milestones__c, Reflection_Log__c
-            FROM Supervisor_AI_Coaching__c
-            WHERE Supervisor_ID__c = '{supervisor_id}' AND Project_ID__c = '{project_id}'
-            ORDER BY Generated_Date__c DESC
+            FROM Project__c
+            WHERE Id = '{project_id}'
             LIMIT 1
         """
         logger.debug("Executing Salesforce query: %s", query)
         result = sf.query(query)
         if result['totalSize'] > 0:
             record = result['records'][0]
-            logger.info("Fetched record from Salesforce: %s", record)
+            logger.info("Fetched Project__c record: %s", record)
             return {
                 'supervisor_id': record['Supervisor_ID__c'] or '',
                 'role': record['Role__c'] or '',
@@ -98,8 +92,8 @@ def fetch_salesforce_input(supervisor_id, project_id):
                 'reflection': record['Reflection_Log__c'] or ''
             }, None
         else:
-            logger.info("No input data found for supervisor %s and project %s", supervisor_id, project_id)
-            return None, "No input data found in Salesforce."
+            logger.info("No Project__c record found for Id: %s", project_id)
+            return None, "No Project__c record found for the given Id."
 
     except Exception as e:
         logger.error("Salesforce fetch error: %s", e)
@@ -137,7 +131,7 @@ Format the response as JSON:
             "max_length": 200,
             "temperature": 0.7,
             "top_p": 0.9,
-            "debug": True  # Debug mode is already enabled
+            "debug": True
         }
     }
 
@@ -170,24 +164,20 @@ Format the response as JSON:
         logger.error("Unexpected error in Hugging Face API call: %s", e)
         return None, f"Unexpected error in Hugging Face API call: {str(e)}"
 
-def save_to_salesforce(output, supervisor_id, project_id):
+def save_to_coaching_log(output, supervisor_id, project_id):
     """
-    Save coaching output to Salesforce Supervisor_AI_Coaching__c object.
+    Save coaching output to the Coaching_Log__c object in Salesforce.
     """
     if not output:
         logger.error("No coaching output to save")
         return False, "No coaching output to save."
 
-    try:
-        logger.debug("Connecting to Salesforce to save output")
-        sf = Salesforce(
-            username=SALESFORCE_USERNAME,
-            password=SALESFORCE_PASSWORD,
-            security_token=SALESFORCE_SECURITY_TOKEN,
-            domain=SALESFORCE_DOMAIN
-        )
-        logger.info("Connected to Salesforce for saving output")
+    success, error, sf = test_salesforce_connection()
+    if not success:
+        return False, error
 
+    try:
+        logger.debug("Saving to Coaching_Log__c")
         coaching_record = {
             "Supervisor_ID__c": supervisor_id,
             "Project_ID__c": project_id,
@@ -196,85 +186,116 @@ def save_to_salesforce(output, supervisor_id, project_id):
             "Quote__c": output["quote"],
             "Generated_Date__c": datetime.now().strftime("%Y-%m-%d")
         }
-        logger.debug("Saving record to Salesforce: %s", coaching_record)
+        logger.debug("Saving record to Coaching_Log__c: %s", coaching_record)
 
-        sf.Supervisor_AI_Coaching__c.upsert(
-            f"Supervisor_ID__c/{supervisor_id}_{datetime.now().strftime('%Y-%m-%d')}",
-            coaching_record
-        )
-        logger.info("Successfully saved coaching record to Salesforce for supervisor %s", supervisor_id)
+        sf.Coaching_Log__c.create(coaching_record)
+        logger.info("Successfully saved coaching record to Coaching_Log__c for supervisor %s", supervisor_id)
         return True, None
 
     except Exception as e:
         logger.error("Salesforce save error: %s", e)
         return False, f"Salesforce save error: {str(e)}"
 
-@app.route('/', methods=['GET'])
-def redirect_to_ui():
+@app.route('/process_new_project', methods=['POST'])
+def process_new_project():
     """
-    Redirect root URL to the UI.
+    Endpoint called by Salesforce when a new Project__c record is created.
     """
-    return redirect(url_for('ui'))
+    data = request.get_json()
+    if not data or 'projectId' not in data:
+        logger.error("Invalid request: projectId not provided")
+        return jsonify({"status": "error", "message": "projectId not provided"}), 400
+
+    project_id = data['projectId']
+    logger.info("Received new Project__c record with Id: %s", project_id)
+
+    # Fetch project data
+    project_data, fetch_error = fetch_project_data(project_id)
+    if fetch_error:
+        logger.error("Failed to fetch project data: %s", fetch_error)
+        return jsonify({"status": "error", "message": fetch_error}), 500
+
+    if not project_data:
+        logger.error("No project data found for Id: %s", project_id)
+        return jsonify({"status": "error", "message": "No project data found"}), 404
+
+    # Prepare data for Hugging Face
+    data = {
+        'supervisor_id': project_data['supervisor_id'],
+        'role': project_data['role'],
+        'project_id': project_data['project_id'],
+        'milestones': [m.strip() for m in project_data['milestones'].split(',') if m.strip()],
+        'reflection_log': project_data['reflection'],
+        'weather': project_data['weather']
+    }
+
+    # Generate coaching output
+    coaching_output, gen_error = generate_coaching_output(data)
+    if gen_error:
+        logger.error("Failed to generate coaching output: %s", gen_error)
+        return jsonify({"status": "error", "message": gen_error}), 500
+
+    if not coaching_output:
+        logger.error("No coaching output generated")
+        return jsonify({"status": "error", "message": "No coaching output generated"}), 500
+
+    # Save to Coaching_Log__c
+    success, save_error = save_to_coaching_log(coaching_output, project_data['supervisor_id'], project_data['project_id'])
+    if not success:
+        logger.error("Failed to save to Coaching_Log__c: %s", save_error)
+        return jsonify({"status": "error", "message": save_error}), 500
+
+    return jsonify({"status": "success", "message": "Coaching log generated and saved successfully"}), 200
 
 @app.route('/ui', methods=['GET'])
 def ui():
     """
-    Serve the HTML user interface, fetch input from Salesforce, generate output, and save to Salesforce.
+    Serve the HTML user interface to display the latest processed record.
     """
     form_data = {}
     output = {}
     error = ""
 
-    # Test Salesforce connection first
-    success, conn_error = test_salesforce_connection()
+    # Fetch the latest Coaching_Log__c record for display
+    success, error, sf = test_salesforce_connection()
     if not success:
-        logger.error("Salesforce connection test failed: %s", conn_error)
-        return render_template('index.html', form_data={}, output={}, error=conn_error)
-
-    # Hardcoded supervisor_id and project_id for demonstration
-    supervisor_id = "SUP-001"
-    project_id = "PROJ-123"
+        logger.error("Salesforce connection test failed: %s", error)
+        return render_template('index.html', form_data={}, output={}, error=error)
 
-    # Fetch input data from Salesforce
-    logger.info("Fetching input data from Salesforce for supervisor %s and project %s", supervisor_id, project_id)
-    sf_input, fetch_error = fetch_salesforce_input(supervisor_id, project_id)
-    if fetch_error:
-        logger.error("Failed to fetch input data: %s", fetch_error)
-        return render_template('index.html', form_data={}, output={}, error=fetch_error)
-
-    if sf_input:
-        form_data = sf_input
-        # Prepare data for generating output
-        data = {
-            'supervisor_id': form_data['supervisor_id'],
-            'role': form_data['role'],
-            'project_id': form_data['project_id'],
-            'milestones': [m.strip() for m in form_data['milestones'].split(',') if m.strip()],
-            'reflection_log': form_data['reflection'],
-            'weather': form_data['weather']
-        }
-        # Generate output automatically
-        logger.info("Generating output for fetched Salesforce data")
-        coaching_output, gen_error = generate_coaching_output(data)
-        if gen_error:
-            logger.error("Failed to generate coaching output: %s", gen_error)
-            return render_template('index.html', form_data=form_data, output={}, error=gen_error)
-
-        if coaching_output:
-            # Save the generated output to Salesforce
-            logger.info("Saving generated output to Salesforce")
-            success, save_error = save_to_salesforce(coaching_output, supervisor_id, project_id)
-            if success:
-                output = coaching_output
-            else:
-                logger.error("Failed to save generated output to Salesforce: %s", save_error)
-                return render_template('index.html', form_data=form_data, output={}, error=save_error)
-    else:
-        logger.warning("No input data found in Salesforce. Displaying empty fields.")
-        error = "No input data found in Salesforce for the given Supervisor ID and Project ID."
+    try:
+        query = """
+            SELECT Supervisor_ID__c, Project_ID__c, Daily_Checklist__c, Suggested_Tips__c, Quote__c
+            FROM Coaching_Log__c
+            ORDER BY Generated_Date__c DESC
+            LIMIT 1
+        """
+        result = sf.query(query)
+        if result['totalSize'] > 0:
+            record = result['records'][0]
+            form_data = {
+                'supervisor_id': record['Supervisor_ID__c'] or '',
+                'project_id': record['Project_ID__c'] or ''
+            }
+            output = {
+                'checklist': record['Daily_Checklist__c'].split('\n') if record['Daily_Checklist__c'] else [],
+                'tips': record['Suggested_Tips__c'].split('\n') if record['Suggested_Tips__c'] else [],
+                'quote': record['Quote__c'] or ''
+            }
+        else:
+            error = "No coaching log records found."
+    except Exception as e:
+        logger.error("Error fetching Coaching_Log__c: %s", e)
+        error = f"Error fetching Coaching_Log__c: {str(e)}"
 
     return render_template('index.html', form_data=form_data, output=output, error=error)
 
+@app.route('/', methods=['GET'])
+def redirect_to_ui():
+    """
+    Redirect root URL to the UI.
+    """
+    return redirect(url_for('ui'))
+
 @app.route('/health', methods=['GET'])
 def health_check():
     """