Update app.py

app.py

@@ -113,16 +113,12 @@ def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_pa
 
         query = sf.query(f"SELECT Id FROM Supervisor__c WHERE Name = '{supervisor_name}' LIMIT 1")
         if query['totalSize'] == 0:
-            print("⚠️ Supervisor not found!")
             return ""
-
         supervisor_id = query['records'][0]['Id']
 
         project_query = sf.query(f"SELECT Id FROM Project__c WHERE Name = '{project_id}' LIMIT 1")
         if project_query['totalSize'] == 0:
-            print(f"⚠️ Project '{project_id}' not found in Salesforce!")
             return ""
-
         project_id_sf = project_query['records'][0]['Id']
 
         sf.Supervisor_AI_Coaching__c.create({
@@ -136,7 +132,6 @@ def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_pa
         return download_url
 
     except Exception as e:
-        print(f"⚠️ Upload error: {e}")
         return ""
 
 def get_roles_from_salesforce():
@@ -149,8 +144,7 @@ def get_roles_from_salesforce():
         )
         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:
-        print(f"⚠️ Error fetching roles: {e}")
+    except:
         return []
 
 def get_supervisor_name_by_role(role):
@@ -163,8 +157,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 names: {e}")
+    except:
         return []
 
 def get_projects_for_supervisor(supervisor_name):
@@ -181,16 +174,38 @@ def get_projects_for_supervisor(supervisor_name):
         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:
-        print(f"⚠️ Error fetching project: {e}")
+    except:
         return ""
 
-def generate_salesforce_dashboard_url(supervisor_name, project_id):
-    return f"https://aicoachforsitesupervisors-dev-ed--c.develop.vf.force.com/apex/DashboardPage?supervisorName={supervisor_name}&projectId={project_id}"
+def get_dashboard_data_from_salesforce(supervisor_name, project_id):
+    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')
+        )
 
-def open_dashboard(role, supervisor_name, project_id):
-    url = generate_salesforce_dashboard_url(supervisor_name, project_id)
-    return f'<a href="{url}" target="_blank">Open Salesforce Dashboard</a>'
+        query = f"""
+        SELECT Daily_Checklist__c, Suggested_Tips__c, Download_Link__c, CreatedDate 
+        FROM Supervisor_AI_Coaching__c 
+        WHERE Supervisor_ID__r.Name = '{supervisor_name}' AND Project_ID__r.Name = '{project_id}'
+        ORDER BY CreatedDate DESC
+        LIMIT 1
+        """
+        result = sf.query(query)
+        if result['totalSize'] == 0:
+            return "No dashboard data found.", "", "", ""
+
+        record = result['records'][0]
+        return (
+            record.get('Daily_Checklist__c', 'N/A'),
+            record.get('Suggested_Tips__c', 'N/A'),
+            record.get('Download_Link__c', ''),
+            record.get('CreatedDate', 'Unknown')
+        )
+    except Exception as e:
+        return f"Error loading dashboard: {e}", "", "", ""
 
 def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
     if not all([role, supervisor_name, project_id, milestones, reflection]):
@@ -210,8 +225,7 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
                 pad_token_id=tokenizer.pad_token_id
             )
         result = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    except Exception as e:
-        print(f"⚠️ Generation error: {e}")
+    except:
         return "", "", None, ""
 
     def extract_between(text, start, end):
@@ -237,52 +251,63 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
 
 def create_interface():
     roles = get_roles_from_salesforce()
-    with gr.Blocks(theme="soft",css=".footer { display: none; }") as demo:
+    with gr.Blocks(theme="soft", css=".footer { display: none; }") as demo:
         gr.Markdown("## 🧠 AI-Powered Supervisor Assistant")
 
-        with gr.Row():
-            role = gr.Dropdown(choices=roles, label="Role")
-            supervisor_name = gr.Dropdown(choices=[], label="Supervisor Name")
-            project_id = gr.Textbox(label="Project ID", interactive=False)
+        with gr.Tab("✏️ Generate Checklist"):
+            with gr.Row():
+                role = gr.Dropdown(choices=roles, label="Role")
+                supervisor_name = gr.Dropdown(choices=[], label="Supervisor Name")
+                project_id = gr.Textbox(label="Project ID", interactive=False)
+
+            milestones = gr.Textbox(label="Milestones (comma-separated KPIs)")
+            reflection = gr.Textbox(label="Reflection Log", lines=4)
+
+            with gr.Row():
+                generate = gr.Button("Generate")
+                clear = gr.Button("Clear")
+                refresh = gr.Button("🔄 Refresh Roles")
 
-        milestones = gr.Textbox(label="Milestones (comma-separated KPIs)")
-        reflection = gr.Textbox(label="Reflection Log", lines=4)
+            checklist_output = gr.Textbox(label="✅ Daily Checklist")
+            suggestions_output = gr.Textbox(label="💡 Focus Suggestions")
+            download_button = gr.File(label="⬇ Download Report")
+            pdf_link = gr.HTML()
 
-        with gr.Row():
-            generate = gr.Button("Generate")
-            clear = gr.Button("Clear")
-            refresh = gr.Button("🔄 Refresh Roles")
-            dashboard_btn = gr.Button("Dashboard")
+            role.change(fn=lambda r: gr.update(choices=get_supervisor_name_by_role(r)), inputs=role, outputs=supervisor_name)
+            supervisor_name.change(fn=get_projects_for_supervisor, inputs=supervisor_name, outputs=project_id)
 
-        checklist_output = gr.Textbox(label="✅ Daily Checklist")
-        suggestions_output = gr.Textbox(label="💡 Focus Suggestions")
-        download_button = gr.File(label="⬇ Download Report")
-        pdf_link = gr.HTML()
-        dashboard_link = gr.HTML()
+            def handle_generate(role, supervisor_name, project_id, milestones, reflection):
+                checklist, suggestions, pdf_path, pdf_name = generate_outputs(role, supervisor_name, project_id, milestones, reflection)
+                return checklist, suggestions, pdf_path, pdf_name
 
-        role.change(fn=lambda r: gr.update(choices=get_supervisor_name_by_role(r)), inputs=role, outputs=supervisor_name)
-        supervisor_name.change(fn=get_projects_for_supervisor, inputs=supervisor_name, outputs=project_id)
+            generate.click(fn=handle_generate,
+                           inputs=[role, supervisor_name, project_id, milestones, reflection],
+                           outputs=[checklist_output, suggestions_output, download_button, pdf_link])
 
-        def handle_generate(role, supervisor_name, project_id, milestones, reflection):
-            checklist, suggestions, pdf_path, pdf_name = generate_outputs(role, supervisor_name, project_id, milestones, reflection)
-            return checklist, suggestions, pdf_path, pdf_name
+            clear.click(fn=lambda: ("", "", "", "", ""),
+                        inputs=None,
+                        outputs=[role, supervisor_name, project_id, milestones, reflection])
 
-        generate.click(fn=handle_generate,
-                       inputs=[role, supervisor_name, project_id, milestones, reflection],
-                       outputs=[checklist_output, suggestions_output, download_button, pdf_link])
+            refresh.click(fn=lambda: gr.update(choices=get_roles_from_salesforce()), outputs=role)
 
-        clear.click(fn=lambda: ("", "", "", "", ""),
-                    inputs=None,
-                    outputs=[role, supervisor_name, project_id, milestones, reflection])
+        with gr.Tab("📊 Supervisor Dashboard"):
+            dash_supervisor = gr.Textbox(label="Supervisor Name")
+            dash_project = gr.Textbox(label="Project ID")
+            load_dash = gr.Button("Load Dashboard")
+            dash_checklist = gr.Textbox(label="✅ Last Checklist", lines=6)
+            dash_suggestions = gr.Textbox(label="💡 Suggestions", lines=4)
+            dash_pdf = gr.HTML(label="📄 PDF Link")
+            dash_date = gr.Textbox(label="🕒 Last Updated", interactive=False)
 
-        refresh.click(fn=lambda: gr.update(choices=get_roles_from_salesforce()), outputs=role)
+            def show_dashboard(sup_name, proj_id):
+                checklist, suggestions, link, date = get_dashboard_data_from_salesforce(sup_name, proj_id)
+                return checklist, suggestions, f'<a href="{link}" target="_blank">Download PDF</a>' if link else "No link", date
 
-        dashboard_btn.click(fn=open_dashboard,
-                            inputs=[role, supervisor_name, project_id],
-                            outputs=dashboard_link)
+            load_dash.click(fn=show_dashboard, inputs=[dash_supervisor, dash_project],
+                            outputs=[dash_checklist, dash_suggestions, dash_pdf, dash_date])
 
     return demo
 
 if __name__ == "__main__":
     app = create_interface()
-    app.launch()
+    app.launch()