Update app.py

app.py

@@ -146,9 +146,11 @@ def get_roles_from_salesforce():
             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']))
+        result = sf.query("SELECT DISTINCT Role__c FROM Supervisor__c WHERE Role__c != NULL")
+        roles = [record['Role__c'] for record in result['records']]
+        return roles
     except Exception as e:
+        print(f"Error fetching roles: {e}")
         return []
 
 def get_supervisor_name_by_role(role):
@@ -162,6 +164,7 @@ def get_supervisor_name_by_role(role):
         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:
+        print(f"Error fetching supervisor names: {e}")
         return []
 
 def get_projects_for_supervisor(supervisor_name):
@@ -179,6 +182,7 @@ def get_projects_for_supervisor(supervisor_name):
         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:
+        print(f"Error fetching projects: {e}")
         return ""
 
 def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
@@ -284,11 +288,12 @@ def show_dashboard_html(sup_name, proj_id):
 
 # Supervisor Dashboard Tab
 def create_interface():
+    roles = get_roles_from_salesforce()  # Fetch roles from Salesforce
     with gr.Blocks(theme="soft", css=".footer { display: none; }") as demo:
         gr.Markdown("## 🧠 AI-Powered Supervisor Assistant")
 
         with gr.Row():
-            role = gr.Dropdown(choices=[], label="Role")
+            role = gr.Dropdown(choices=roles, label="Role")
             supervisor_name = gr.Dropdown(choices=[], label="Supervisor Name")
             project_id = gr.Textbox(label="Project ID", interactive=False)
 
@@ -322,15 +327,6 @@ def create_interface():
 
         refresh.click(fn=lambda: gr.update(choices=get_roles_from_salesforce()), outputs=role)
 
-        # Supervisor Dashboard Tab
-        with gr.Tab("📊 Supervisor Dashboard"):
-            dash_supervisor = gr.Textbox(label="Supervisor Name", placeholder="e.g., SUP-056")
-            dash_project = gr.Textbox(label="Project ID", placeholder="e.g., PROJ-078")
-            load_dash = gr.Button("📥 Load Dashboard")
-            dash_output = gr.HTML()
-
-            load_dash.click(fn=show_dashboard_html, inputs=[dash_supervisor, dash_project], outputs=dash_output)
-
     return demo
 
 if __name__ == "__main__":