Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import logging
+import requests
 from datetime import datetime
 from flask import Flask, render_template, request, redirect, url_for, session, jsonify, send_file
 from simple_salesforce import Salesforce
@@ -22,18 +23,80 @@ app = Flask(__name__)
 app.secret_key = os.getenv('FLASK_SECRET_KEY', 'default-secret-key')
 
 # Salesforce connection
+sf = None
 try:
     sf = Salesforce(
         username=os.getenv('SALESFORCE_USERNAME'),
         password=os.getenv('SALESFORCE_PASSWORD'),
         security_token=os.getenv('SALESFORCE_SECURITY_TOKEN'),
-        domain=os.getenv('SALESFORCE_DOMAIN', 'test')
+        domain=os.getenv('SALESFORCE_DOMAIN', 'login')
     )
-    logging.info("Successfully connected to Salesforce")
+    # Verify connection by querying Salesforce metadata
+    user_info = sf.query("SELECT Id, Name FROM User WHERE Username = '{}'".format(os.getenv('SALESFORCE_USERNAME')))
+    if user_info['totalSize'] > 0:
+        logging.info(f"Successfully connected to Salesforce org. User: {user_info['records'][0]['Name']}")
+    else:
+        logging.error("Salesforce connection established, but user not found in org")
 except Exception as e:
     logging.error(f"Failed to connect to Salesforce: {str(e)}")
     sf = None
 
+# Hugging Face API configuration
+HUGGING_FACE_API_TOKEN = os.getenv('HUGGING_FACE_API_TOKEN')
+HUGGING_FACE_API_URL = os.getenv('HUGGING_FACE_API_URL')
+HUGGING_FACE_HEADERS = {
+    'Authorization': f'Bearer {HUGGING_FACE_API_TOKEN}',
+    'Content-Type': 'application/json'
+}
+
+def generate_coaching_output_with_hf(data):
+    try:
+        prompt = (
+            f"Generate a daily checklist and focus tips for a supervisor with the following details:\n"
+            f"Project ID: {data.get('project_id', 'N/A')}\n"
+            f"Engagement Score: {data.get('engagement_score', 0)}%\n"
+            f"KPI Flag: {'Active' if data.get('kpi_flag', False) else 'Inactive'}\n"
+            f"Reflection Log: {data.get('reflection_log', 'No reflections yet.')}\n\n"
+            "Provide a checklist (3 items) and focus tips (3 items) in a concise format."
+        )
+        payload = {
+            "inputs": prompt,
+            "parameters": {"max_length": 150, "num_return_sequences": 1}
+        }
+        response = requests.post(HUGGING_FACE_API_URL, headers=HUGGING_FACE_HEADERS, json=payload)
+        response.raise_for_status()
+        result = response.json()
+        generated_text = result[0]['generated_text'].split('\n')
+
+        checklist = []
+        tips = []
+        is_checklist = True
+        for line in generated_text:
+            line = line.strip()
+            if not line:
+                continue
+            if "focus tips" in line.lower():
+                is_checklist = False
+                continue
+            if is_checklist and len(checklist) < 3:
+                checklist.append(line.replace('- ', '').strip())
+            elif not is_checklist and len(tips) < 3:
+                tips.append(line.replace('- ', '').strip())
+
+        # Fallback if model output is incomplete
+        while len(checklist) < 3:
+            checklist.append(f"Placeholder Task {len(checklist) + 1}")
+        while len(tips) < 3:
+            tips.append(f"Placeholder Tip {len(tips) + 1}")
+
+        return {'checklist': checklist, 'tips': tips}
+    except Exception as e:
+        logging.error(f"Error generating coaching output with Hugging Face API: {str(e)}")
+        return {
+            'checklist': ['Review project milestones', 'Conduct team check-in', 'Update resource allocation'],
+            'tips': ['Set clear deadlines', 'Encourage team feedback', 'Monitor progress daily']
+        }
+
 @app.route('/')
 def home():
     if 'supervisor_name' not in session:
@@ -56,7 +119,7 @@ def login():
 
         if not sf:
             logging.error("Salesforce connection not available during login")
-            return jsonify({'status': 'error', 'message': 'Salesforce connection unavailable'}), 500
+            return render_template('login.html', error="Salesforce connection unavailable")
 
         try:
             query = f"SELECT Id, Name, Password__c FROM Supervisor_AI_Coaching__c WHERE Name = '{username}'"
@@ -93,7 +156,7 @@ def signup():
 
         if not sf:
             logging.error("Salesforce connection not available during signup")
-            return jsonify({'status': 'error', 'message': 'Salesforce connection unavailable'}), 500
+            return render_template('signup.html', error="Salesforce connection unavailable")
 
         try:
             existing_user = sf.query(f"SELECT Id FROM Supervisor_AI_Coaching__c WHERE Name = '{username}'")
@@ -222,19 +285,9 @@ def generate_coaching():
         return jsonify({'status': 'error', 'message': 'Salesforce connection unavailable'}), 500
 
     try:
-        # Simulated coaching logic (replace with actual AI model in production)
-        output = {
-            'checklist': [
-                'Review project milestones',
-                'Conduct team check-in',
-                'Update resource allocation'
-            ],
-            'tips': [
-                'Set clear deadlines',
-                'Encourage team feedback',
-                'Monitor progress daily'
-            ]
-        }
+        # Fetch supervisor data to pass to Hugging Face API
+        data = request.json
+        output = generate_coaching_output_with_hf(data)
 
         if supervisor_name != 'GUEST':
             query = f"SELECT Id FROM Supervisor_AI_Coaching__c WHERE Name = '{supervisor_name}'"
@@ -325,7 +378,6 @@ def download_pdf():
         buffer.seek(0)
 
         if supervisor_name != 'GUEST':
-            # Update the download link (placeholder for now)
             record_id = result['records'][0]['Id']
             sf.Supervisor_AI_Coaching__c.update(record_id, {
                 'Download_Link__c': 'https://example.com/report.pdf'