Update app.py

app.py

@@ -2,6 +2,7 @@ import logging
 import os
 import uvicorn
 from fastapi import FastAPI, HTTPException
+from fastapi.responses import FileResponse
 from pydantic import BaseModel
 from simple_salesforce import Salesforce
 from contextlib import asynccontextmanager
@@ -97,6 +98,45 @@ def generate_coaching_data(project: dict, role: str) -> dict:
         "engagementScore": engagement_score
     }
 
+@app.get("/api/latest-project")
+async def get_latest_project():
+    """Fetch the latest Project__c record based on CreatedDate."""
+    if sf is None:
+        logger.error("Salesforce connection not initialized")
+        raise HTTPException(status_code=500, detail="Salesforce connection not initialized")
+    
+    try:
+        query = """
+            SELECT Id, Name, Project_Name__c, Milestones__c, Weather_Logs__c, Safety_Logs__c, Location__c, Supervisor_ID__c
+            FROM Project__c
+            ORDER BY CreatedDate DESC
+            LIMIT 1
+        """
+        logger.info(f"Executing SOQL query for latest project: {query}")
+        result = sf.query(query)
+        
+        if result['totalSize'] == 0:
+            logger.warning("No Project__c records found")
+            raise HTTPException(status_code=404, detail="No projects found")
+        
+        project = result['records'][0]
+        logger.info(f"Fetched latest project: {project['Name']}")
+        
+        # Prepare response with extracted fields
+        response = {
+            "projectId": project['Name'],
+            "projectName": project['Project_Name__c'],
+            "milestones": project.get('Milestones__c', ''),
+            "weatherLogs": project.get('Weather_Logs__c', ''),
+            "safetyLogs": project.get('Safety_Logs__c', ''),
+            "role": "Site Manager"  # Default role; can be updated based on Supervisor_ID__c if needed
+        }
+        
+        return response
+    except Exception as e:
+        logger.error(f"Error fetching latest project: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
+
 @app.post("/api/generate")
 async def generate_coaching(request: ProjectRequest):
     if sf is None:
@@ -104,7 +144,7 @@ async def generate_coaching(request: ProjectRequest):
         raise HTTPException(status_code=500, detail="Salesforce connection not initialized")
     
     try:
-        # Query Project__c by projectId
+        # Query Project__c by projectId to get the full record
         query = f"SELECT Id, Name, Project_Name__c, Milestones__c, Weather_Logs__c, Safety_Logs__c, Location__c, Supervisor_ID__c FROM Project__c WHERE Name = '{request.projectId}' LIMIT 1"
         logger.info(f"Executing SOQL query: {query}")
         result = sf.query(query)
@@ -136,6 +176,25 @@ async def generate_coaching(request: ProjectRequest):
             "engagementScore": coaching_data['engagementScore']
         }
         
+        # Insert into Supervisor_AI_Coaching__c
+        try:
+            coaching_record = {
+                'Project__c': project['Id'],
+                'Checklist__c': coaching_data['checklist'],
+                'Tips__c': coaching_data['tips'],
+                'Enguner_score__c': coaching_data['engagementScore'],
+                'Role__c': request.role,
+                'Project_Name__c': project['Project_Name__c'],
+                'Milestones__c': project.get('Milestones__c', ''),
+                'Weather_Logs__c': project.get('Weather_Logs__c', ''),
+                'Safety_Logs__c': project.get('Safety_Logs__c', '')
+            }
+            sf.Supervisor_AI_Coaching__c.create(coaching_record)
+            logger.info(f"Inserted coaching data into Supervisor_AI_Coaching__c for project {project['Name']}")
+        except Exception as e:
+            logger.error(f"Failed to insert into Supervisor_AI_Coaching__c: {str(e)}")
+            raise HTTPException(status_code=500, detail=f"Failed to save coaching data: {str(e)}")
+        
         logger.info(f"Generated coaching response: {response}")
         return response
     except Exception as e:
@@ -144,8 +203,13 @@ async def generate_coaching(request: ProjectRequest):
 
 @app.get("/")
 async def root():
-    logger.info("Root endpoint accessed")
-    return {"message": "Welcome to AI Coach API. Use POST /api/generate"}
+    logger.info("Serving index.html from root directory")
+    return FileResponse("index.html")
+
+@app.get("/styles.css")
+async def serve_styles():
+    logger.info("Serving styles.css from root directory")
+    return FileResponse("styles.css", media_type="text/css")
 
 # Start Uvicorn server for Hugging Face Spaces
 logger.info("Starting Uvicorn server for Hugging Face Spaces")