Update app.py

app.py

@@ -121,9 +121,55 @@ def save_report_as_pdf(role, supervisor_name, project_id, checklist, suggestions
     shutil.copy(file_path, temp_pdf_path)
     return temp_pdf_path, filename
 
+# 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:
+        return []
+
+# Function to get supervisor names based on role
+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:
+        return []
+
+# Function to get the project name for a supervisor
+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:
+        return ""
+
 # Gradio interface
 def create_interface():
-    roles = ['Supervisor A', 'Supervisor B']  # Add roles dynamically from Salesforce if needed
+    roles = get_roles_from_salesforce()  # Get roles from Salesforce dynamically
     with gr.Blocks(theme="soft", css=".footer { display: none; }") as demo:
         gr.Markdown("## 🧠 AI-Powered Supervisor Assistant")