Update app.py

app.py

@@ -29,7 +29,6 @@ if tokenizer.pad_token is None:
     tokenizer.pad_token_id = tokenizer.convert_tokens_to_ids(tokenizer.pad_token)
 model.config.pad_token_id = tokenizer.pad_token_id
 
-# Prompt template
 PROMPT_TEMPLATE = """You are an AI assistant for construction supervisors. Given the role, project, milestones, and a reflection log, generate:
 
 1. A Daily Checklist with clear and concise tasks based on the role and milestones. 
@@ -98,29 +97,23 @@ def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_pa
             security_token=os.getenv('SF_SECURITY_TOKEN'),
             domain=os.getenv('SF_DOMAIN', 'login')
         )
-
         with open(pdf_path, "rb") as f:
             encoded = base64.b64encode(f.read()).decode()
-
         content = sf.ContentVersion.create({
             'Title': pdf_name,
             'PathOnClient': pdf_name,
             'VersionData': encoded
         })
-
         content_id = content['id']
         download_url = f"https://{sf.sf_instance}/sfc/servlet.shepherd/version/download/{content_id}"
-
         query = sf.query(f"SELECT Id FROM Supervisor__c WHERE Name = '{supervisor_name}' LIMIT 1")
         if query['totalSize'] == 0:
             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:
             return ""
         project_id_sf = project_query['records'][0]['Id']
-
         sf.Supervisor_AI_Coaching__c.create({
             'Project_ID__c': project_id_sf,
             'Supervisor_ID__c': supervisor_id,
@@ -128,10 +121,8 @@ def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_pa
             'Suggested_Tips__c': suggestions,
             'Download_Link__c': download_url
         })
-
         return download_url
-
-    except Exception as e:
+    except:
         return ""
 
 def get_roles_from_salesforce():
@@ -185,7 +176,6 @@ def get_dashboard_data_from_salesforce(supervisor_name, project_id):
             security_token=os.getenv('SF_SECURITY_TOKEN'),
             domain=os.getenv('SF_DOMAIN', 'login')
         )
-
         query = f"""
         SELECT Daily_Checklist__c, Suggested_Tips__c, Download_Link__c, CreatedDate 
         FROM Supervisor_AI_Coaching__c 
@@ -196,7 +186,6 @@ def get_dashboard_data_from_salesforce(supervisor_name, project_id):
         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'),
@@ -210,7 +199,6 @@ def get_dashboard_data_from_salesforce(supervisor_name, project_id):
 def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
     if not all([role, supervisor_name, project_id, milestones, reflection]):
         return "❗ Please fill all fields.", "", None, ""
-
     prompt = PROMPT_TEMPLATE.format(role=role, project_id=project_id, milestones=milestones, reflection=reflection)
     inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True, max_length=512)
     try:
@@ -291,23 +279,34 @@ def create_interface():
             refresh.click(fn=lambda: gr.update(choices=get_roles_from_salesforce()), outputs=role)
 
         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)
-
-            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
+            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, inputs=[dash_supervisor, dash_project],
-                            outputs=[dash_checklist, dash_suggestions, dash_pdf, dash_date])
+            def show_dashboard_html(sup_name, proj_id):
+                checklist, suggestions, link, date = get_dashboard_data_from_salesforce(sup_name, proj_id)
+                link_html = f'<a href="{link}" target="_blank">Download PDF</a>' if link else "No report available"
+                html_card = f"""
+                <div style='border:1px solid #ccc; border-radius:10px; padding:20px; background-color:#f9f9f9'>
+                    <h3>📋 <u>Dashboard Summary</u></h3>
+                    <p><b>Supervisor:</b> {sup_name}</p>
+                    <p><b>Project ID:</b> {proj_id}</p>
+                    <p><b>🗓️ Last Updated:</b> {date}</p>
+                    <hr>
+                    <h4>✅ <u>Daily Checklist</u></h4>
+                    <pre style='background:#eef; padding:10px; border-radius:5px;'>{checklist}</pre>
+                    <h4>💡 <u>Focus Suggestions</u></h4>
+                    <pre style='background:#ffe; padding:10px; border-radius:5px;'>{suggestions}</pre>
+                    <p>{link_html}</p>
+                </div>
+                """
+                return html_card
+
+            load_dash.click(fn=show_dashboard_html, inputs=[dash_supervisor, dash_project], outputs=dash_output)
 
     return demo
 
 if __name__ == "__main__":
     app = create_interface()
-    app.launch()
+    app.launch()