Update app.py

app.py

@@ -33,6 +33,43 @@ if not all([SALESFORCE_USERNAME, SALESFORCE_PASSWORD, SALESFORCE_SECURITY_TOKEN]
 # Initialize Flask app
 app = Flask(__name__)
 
+def fetch_input_from_salesforce(supervisor_id, project_id):
+    """
+    Fetch input data from Salesforce for a given supervisor and project.
+    """
+    try:
+        sf = Salesforce(
+            username=SALESFORCE_USERNAME,
+            password=SALESFORCE_PASSWORD,
+            security_token=SALESFORCE_SECURITY_TOKEN,
+            domain=SALESFORCE_DOMAIN
+        )
+        logger.info("Connected to Salesforce for input fetch")
+
+        query = f"""
+            SELECT Role__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
+            LIMIT 1
+        """
+        result = sf.query(query)
+        if result['totalSize'] > 0:
+            record = result['records'][0]
+            return {
+                'role': record['Role__c'] or '',
+                'weather': record['Weather__c'] or '',
+                'milestones': record['Milestones__c'] or '',
+                'reflection': record['Reflection_Log__c'] or ''
+            }
+        else:
+            logger.info("No input data found for supervisor %s and project %s", supervisor_id, project_id)
+            return None
+
+    except Exception as e:
+        logger.error("Salesforce fetch error: %s", e)
+        return None
+
 def generate_coaching_output(data):
     """
     Generate daily checklist and tips using Hugging Face LLM.
@@ -95,9 +132,9 @@ Format the response as JSON:
         logger.error("Unexpected error in Hugging Face API call: %s", e)
         return None
 
-def save_to_salesforce(output, supervisor_id, project_id):
+def save_to_salesforce(data, output):
     """
-    Save coaching output to Salesforce Supervisor_AI_Coaching__c object.
+    Save input data and coaching output to Salesforce Supervisor_AI_Coaching__c object.
     """
     if not output:
         logger.error("No coaching output to save")
@@ -113,8 +150,12 @@ def save_to_salesforce(output, supervisor_id, project_id):
         logger.info("Connected to Salesforce")
 
         coaching_record = {
-            "Supervisor_ID__c": supervisor_id,
-            "Project_ID__c": project_id,
+            "Supervisor_ID__c": data['supervisor_id'],
+            "Project_ID__c": data['project_id'],
+            "Role__c": data['role'],
+            "Weather__c": data['weather'],
+            "Milestones__c": ", ".join(data['milestones']),
+            "Reflection_Log__c": data['reflection_log'],
             "Daily_Checklist__c": "\n".join(output["checklist"]),
             "Suggested_Tips__c": "\n".join(output["tips"]),
             "Quote__c": output["quote"],
@@ -122,10 +163,10 @@ def save_to_salesforce(output, supervisor_id, project_id):
         }
 
         sf.Supervisor_AI_Coaching__c.upsert(
-            f"Supervisor_ID__c/{supervisor_id}_{datetime.now().strftime('%Y-%m-%d')}",
+            f"Supervisor_ID__c/{data['supervisor_id']}_{datetime.now().strftime('%Y-%m-%d')}",
             coaching_record
         )
-        logger.info("Successfully saved coaching record to Salesforce for supervisor %s", supervisor_id)
+        logger.info("Successfully saved coaching record to Salesforce for supervisor %s", data['supervisor_id'])
         return True
 
     except Exception as e:
@@ -139,35 +180,66 @@ def redirect_to_ui():
     """
     return redirect(url_for('ui'))
 
-@app.route('/ui', methods=['GET'])
+@app.route('/ui', methods=['GET', 'POST'])
 def ui():
     """
-    Serve the HTML user interface.
+    Serve the HTML user interface and handle form submissions.
     """
-    return render_template('index.html')
+    form_data = {}
+    output = {}
+    error = ""
+
+    if request.method == 'POST':
+        action = request.form.get('action')
+        form_data = {
+            'supervisor_id': request.form.get('supervisor_id', ''),
+            'project_id': request.form.get('project_id', ''),
+            'role': '',
+            'weather': '',
+            'milestones': '',
+            'reflection': ''
+        }
 
-@app.route('/generate', methods=['POST'])
-def generate_endpoint():
-    """
-    Endpoint to generate coaching output based on supervisor data.
-    """
-    try:
-        data = request.get_json()
-        if not data or not all(key in data for key in ['supervisor_id', 'role', 'project_id', 'milestones', 'reflection_log', 'weather']):
-            return jsonify({"status": "error", "message": "Invalid or missing supervisor data"}), 400
-
-        coaching_output = generate_coaching_output(data)
-        if coaching_output:
-            success = save_to_salesforce(coaching_output, data["supervisor_id"], data["project_id"])
-            if success:
-                return jsonify({"status": "success", "output": coaching_output}), 200
-            else:
-                return jsonify({"status": "error", "message": "Failed to save to Salesforce"}), 500
+        if not all([form_data['supervisor_id'], form_data['project_id']]):
+            error = "Error: Supervisor ID and Project ID are required."
         else:
-            return jsonify({"status": "error", "message": "Failed to generate coaching output"}), 500
-    except Exception as e:
-        logger.error("Error in generate endpoint: %s", e)
-        return jsonify({"status": "error", "message": str(e)}), 500
+            if action == 'fetch':
+                # Fetch input data from Salesforce
+                input_data = fetch_input_from_salesforce(form_data['supervisor_id'], form_data['project_id'])
+                if input_data:
+                    form_data.update(input_data)
+                else:
+                    error = "No input data found in Salesforce for the given Supervisor ID and Project ID."
+
+            elif action == 'generate':
+                # Fetch input data from Salesforce
+                input_data = fetch_input_from_salesforce(form_data['supervisor_id'], form_data['project_id'])
+                if not input_data:
+                    error = "No input data found in Salesforce for the given Supervisor ID and Project ID."
+                else:
+                    form_data.update(input_data)
+                    # Prepare data for generation
+                    data = {
+                        'supervisor_id': form_data['supervisor_id'],
+                        'project_id': form_data['project_id'],
+                        'role': form_data['role'],
+                        'milestones': [m.strip() for m in form_data['milestones'].split(',') if m.strip()] if form_data['milestones'] else [],
+                        'reflection_log': form_data['reflection'],
+                        'weather': form_data['weather']
+                    }
+                    # Generate output
+                    coaching_output = generate_coaching_output(data)
+                    if coaching_output:
+                        # Save both input and output to Salesforce
+                        success = save_to_salesforce(data, coaching_output)
+                        if success:
+                            output = coaching_output
+                        else:
+                            error = "Error: Failed to save to Salesforce."
+                    else:
+                        error = "Error: Failed to generate coaching output."
+
+    return render_template('index.html', form_data=form_data, output=output, error=error)
 
 @app.route('/health', methods=['GET'])
 def health_check():