Update app.py

app.py

@@ -1,3 +1,4 @@
+```python
 import gradio as gr
 import logging
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
@@ -37,7 +38,6 @@ MODEL_PATH = "your_actual_username/fine_tuned_bart_construction"  # Replace with
 FALLBACK_MODEL = "facebook/bart-large"  # Fallback public model
 
 try:
-    # Try loading fine-tuned model
     logging.info(f"Attempting to load model from {MODEL_PATH}")
     model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_PATH, use_auth_token=HF_TOKEN)
     tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH, use_auth_token=HF_TOKEN)
@@ -59,7 +59,7 @@ class ChecklistInput(BaseModel):
     project_id: str = "Unknown"
     project_name: str = "Unknown Project"
     milestones: str = "No milestones provided"
-    record_id: str = None  # Salesforce record ID
+    record_id: str = None  # Supervisor_AI_Coaching__c record ID
     supervisor_id: str = None  # Supervisor_ID__c
     project_id_sf: str = None  # Project_ID__c
     coaching_id: str = None  # Coaching_ID__c
@@ -70,31 +70,31 @@ app = FastAPI()
 @app.post("/generate")
 async def generate_checklist(data: ChecklistInput):
     try:
-        # Prepare input for the model
+        # Generate checklist and tips
         inputs = f"Role: {data.role} Project: {data.project_id} ({data.project_name}) Milestones: {data.milestones}"
         logging.info(f"Generating checklist for inputs: {inputs[:100]}...")
-
-        # Tokenize and generate
         input_ids = tokenizer(inputs, return_tensors="pt", max_length=128, truncation=True).input_ids
         outputs = model.generate(input_ids, max_length=128, num_beams=4, early_stopping=True)
         checklist = tokenizer.decode(outputs[0], skip_special_tokens=True)
-
-        # Static tips (replace with model-generated tips after fine-tuning)
         tips = "1. Prioritize safety checks\n2. Review milestones\n3. Log progress"
-
-        # Basic KPI logic: Flag if milestones contain "delay" or similar
         kpi_flag = "delay" in data.milestones.lower() or "behind" in data.milestones.lower()
 
-        # Update Salesforce record if record_id is provided
+        # Update Salesforce if record_id is provided
         if data.record_id:
             try:
                 sf = get_salesforce_connection()
-                # Fetch existing record to preserve fields
-                existing_record = sf.Supervisor_AI_Coaching__c.get(data.record_id, default=None)
+                existing_record = sf.Supervisor_AI_Coaching__c.get(data.record_id, default={
+                    'Supervisor_ID__c': None,
+                    'Project_ID__c': None,
+                    'Coaching_ID__c': None,
+                    'Reflection_Log__c': '',
+                    'Download_Link__c': '',
+                    'Engagement_Score__c': 0
+                })
                 update_data = {
                     'Daily_Checklist__c': checklist,
                     'Suggested_Tips__c': tips,
-                    'Engagement_Score__c': 10 if not existing_record.get('Engagement_Score__c') else existing_record.get('Engagement_Score__c') + 10,
+                    'Engagement_Score__c': existing_record.get('Engagement_Score__c', 0) + 10,
                     'KPI_Flag__c': kpi_flag,
                     'Supervisor_ID__c': data.supervisor_id if data.supervisor_id else existing_record.get('Supervisor_ID__c'),
                     'Project_ID__c': data.project_id_sf if data.project_id_sf else existing_record.get('Project_ID__c'),
@@ -103,7 +103,7 @@ async def generate_checklist(data: ChecklistInput):
                     'Download_Link__c': existing_record.get('Download_Link__c', '')
                 }
                 sf.Supervisor_AI_Coaching__c.update(data.record_id, update_data)
-                logging.info(f"Updated Salesforce record {data.record_id}")
+                logging.info(f"Updated Salesforce record {data.record_id} with fields: {update_data}")
             except Exception as sf_e:
                 logging.error(f"Failed to update Salesforce: {sf_e}")
 
@@ -127,15 +127,21 @@ def gradio_generate_checklist(role, project_id, project_name, milestones, record
         tips = "1. Prioritize safety checks\n2. Review milestones\n3. Log progress"
         kpi_flag = "delay" in milestones.lower() or "behind" in milestones.lower()
 
-        # Update Salesforce if record_id is provided
         status = "No Salesforce update (record_id not provided)"
         if record_id:
             sf = get_salesforce_connection()
-            existing_record = sf.Supervisor_AI_Coaching__c.get(record_id, default=None)
+            existing_record = sf.Supervisor_AI_Coaching__c.get(record_id, default={
+                'Supervisor_ID__c': None,
+                'Project_ID__c': None,
+                'Coaching_ID__c': None,
+                'Reflection_Log__c': '',
+                'Download_Link__c': '',
+                'Engagement_Score__c': 0
+            })
             update_data = {
                 'Daily_Checklist__c': checklist,
                 'Suggested_Tips__c': tips,
-                'Engagement_Score__c': 10 if not existing_record.get('Engagement_Score__c') else existing_record.get('Engagement_Score__c') + 10,
+                'Engagement_Score__c': existing_record.get('Engagement_Score__c', 0) + 10,
                 'KPI_Flag__c': kpi_flag,
                 'Supervisor_ID__c': supervisor_id if supervisor_id else existing_record.get('Supervisor_ID__c'),
                 'Project_ID__c': project_id_sf if project_id_sf else existing_record.get('Project_ID__c'),
@@ -144,7 +150,7 @@ def gradio_generate_checklist(role, project_id, project_name, milestones, record
                 'Download_Link__c': existing_record.get('Download_Link__c', '')
             }
             sf.Supervisor_AI_Coaching__c.update(record_id, update_data)
-            status = f"Updated Salesforce record {record_id}"
+            status = f"Updated Salesforce record {record_id} with fields: {update_data}"
 
         return checklist, tips, kpi_flag, status
     except Exception as e:
@@ -159,8 +165,8 @@ iface = gr.Interface(
         gr.Textbox(label="Project Name", value="Building A"),
         gr.Textbox(label="Milestones", value="Complete foundation by 5/15"),
         gr.Textbox(label="Salesforce Record ID (optional)", value=""),
-        gr.Textbox(label="Supervisor ID (optional)", value=""),
-        gr.Textbox(label="Project ID (Salesforce, optional)", value=""),
+        gr.Textbox(label="Supervisor ID (Salesforce User ID, optional)", value=""),
+        gr.Textbox(label="Project ID (Salesforce Project__c ID, optional)", value=""),
         gr.Textbox(label="Coaching ID (optional)", value="")
     ],
     outputs=[
@@ -182,4 +188,5 @@ if __name__ == "__main__":
         logging.info("Gradio interface launched successfully")
     except Exception as e:
         logging.error(f"Failed to launch Gradio interface: {e}")
-        raise e
+        raise e
+```