Update app.py

app.py

@@ -8,223 +8,270 @@ import logging
 from datetime import datetime
 from fastapi.responses import HTMLResponse
 
+# Set up logging to capture errors and debug information
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 app = FastAPI()
 
+# Verify API key is set
 API_KEY = os.getenv("HUGGINGFACE_API_KEY")
 if not API_KEY:
     logger.error("HUGGINGFACE_API_KEY environment variable not set")
     raise ValueError("HUGGINGFACE_API_KEY environment variable not set")
 
-# Model to match the incoming Vendor Log data from Salesforce
+# VendorLog model to match Salesforce data
 class VendorLog(BaseModel):
     vendorLogId: str
     vendorId: str
-    workDetails: str  # e.g., "90% completed"
-    qualityReport: str  # e.g., "60% quality"
-    incidentLog: str  # e.g., "Minor"
-    workCompletionDate: str  # e.g., "2025-05-12"
-    actualCompletionDate: str  # e.g., "2025-05-10"
-    vendorLogName: str  # e.g., "abhi"
-    delayDays: int  # e.g., 2
-
-# Store vendor logs for display in the dashboard
+    workDetails: str
+    qualityReport: str
+    incidentLog: str
+    workCompletionDate: str
+    actualCompletionDate: str
+    vendorLogName: str
+    delayDays: int
+    project: str
+
+# Store vendor logs for display
 vendor_logs = []
 
 def calculate_scores(log: VendorLog):
-    # Extract percentages from the string fields
-    work_completion_percentage = float(log.workDetails.replace('% completed', ''))
-    quality_percentage = float(log.qualityReport.replace('% quality', ''))
+    try:
+        # Extract percentages from the string fields
+        work_completion_percentage = float(log.workDetails.replace('% completed', ''))
+        quality_percentage = float(log.qualityReport.replace('% quality', ''))
 
-    # Quality Score: Directly use the quality percentage
-    quality_score = quality_percentage
+        # Quality Score: Directly use the quality percentage
+        quality_score = quality_percentage
 
-    # Timeliness Score: Based on delay days
-    timeliness_score = 100.0 if log.delayDays <= 0 else 80.0 if log.delayDays <= 3 else 60.0 if log.delayDays <= 7 else 40.0
+        # Timeliness Score: Based on delay days
+        timeliness_score = 100.0 if log.delayDays <= 0 else 80.0 if log.delayDays <= 3 else 60.0 if log.delayDays <= 7 else 40.0
 
-    # Safety Score: Based on incident severity
-    severity_map = {'None': 100.0, 'Low': 80.0, 'Minor': 80.0, 'Medium': 50.0, 'High': 20.0}
-    safety_score = severity_map.get(log.incidentLog, 100.0)
+        # Safety Score: Based on incident severity
+        severity_map = {'None': 100.0, 'Low': 80.0, 'Minor': 80.0, 'Medium': 50.0, 'High': 20.0}
+        safety_score = severity_map.get(log.incidentLog, 100.0)
 
-    # Communication Score: Weighted average of other scores
-    communication_score = (quality_score * 0.33 + timeliness_score * 0.33 + safety_score * 0.33)
+        # Communication Score: Weighted average of other scores
+        communication_score = (quality_score * 0.33 + timeliness_score * 0.33 + safety_score * 0.33)
 
-    # Final Score: Average of all scores
-    final_score = (quality_score + timeliness_score + safety_score + communication_score) / 4
+        # Final Score: Average of all scores
+        final_score = (quality_score + timeliness_score + safety_score + communication_score) / 4
 
-    return {
-        'qualityScore': round(quality_score, 2),
-        'timelinessScore': round(timeliness_score, 2),
-        'safetyScore': round(safety_score, 2),
-        'communicationScore': round(communication_score, 2),
-        'finalScore': round(final_score, 2)
-    }
+        return {
+            'qualityScore': round(quality_score, 2),
+            'timelinessScore': round(timeliness_score, 2),
+            'safetyScore': round(safety_score, 2),
+            'communicationScore': round(communication_score, 2),
+            'finalScore': round(final_score, 2)
+        }
+    except Exception as e:
+        logger.error(f"Error calculating scores: {str(e)}")
+        raise
 
 def get_feedback(score: float, metric: str) -> str:
-    if score >= 90:
-        return "Excellent: Maintain this standard"
-    elif score >= 70:
-        return "Good: Keep up the good work"
-    elif score >= 50:
-        if metric == 'Timeliness':
-            return "Needs Improvement: Maintain schedules to complete tasks on time"
-        elif metric == 'Safety':
-            return "Needs Improvement: Implement stricter safety protocols"
-        elif metric == 'Quality':
-            return "Needs Improvement: Focus on improving work quality"
-        else:
-            return "Needs Improvement: Enhance coordination with project teams"
-    else:
-        if metric == 'Timeliness':
-            return "Poor: Significant delays detected"
-        elif metric == 'Safety':
-            return "Poor: Critical safety issues identified"
-        elif metric == 'Quality':
-            return "Poor: Quality standards not met"
+    try:
+        if score >= 90:
+            return "Excellent: Maintain this standard"
+        elif score >= 70:
+            return "Good: Keep up the good work"
+        elif score >= 50:
+            if metric == 'Timeliness':
+                return "Needs Improvement: Maintain schedules to complete tasks on time"
+            elif metric == 'Safety':
+                return "Needs Improvement: Implement stricter safety protocols"
+            elif metric == 'Quality':
+                return "Needs Improvement: Focus on improving work quality"
+            else:
+                return "Needs Improvement: Enhance coordination with project teams"
         else:
-            return "Poor: Communication issues detected"
+            if metric == 'Timeliness':
+                return "Poor: Significant delays detected"
+            elif metric == 'Safety':
+                return "Poor: Critical safety issues identified"
+            elif metric == 'Quality':
+                return "Poor: Quality standards not met"
+            else:
+                return "Poor: Communication issues detected"
+    except Exception as e:
+        logger.error(f"Error generating feedback: {str(e)}")
+        raise
 
 def generate_pdf(vendor_id: str, vendor_log_name: str, scores: dict):
-    filename = f'report_{vendor_id}.pdf'
-    c = canvas.Canvas(filename, pagesize=letter)
-    c.setFont('Helvetica', 12)
-    c.drawString(100, 750, 'Subcontractor Performance Report')
-    c.drawString(100, 730, f'Vendor ID: {vendor_id}')
-    c.drawString(100, 710, f'Vendor Log Name: {vendor_log_name}')
-    c.drawString(100, 690, f'Quality Score: {scores["qualityScore"]}% ({get_feedback(scores["qualityScore"], "Quality")})')
-    c.drawString(100, 670, f'Timeliness Score: {scores["timelinessScore"]}% ({get_feedback(scores["timelinessScore"], "Timeliness")})')
-    c.drawString(100, 650, f'Safety Score: {scores["safetyScore"]}% ({get_feedback(scores["safetyScore"], "Safety")})')
-    c.drawString(100, 630, f'Communication Score: {scores["communicationScore"]}% ({get_feedback(scores["communicationScore"], "Communication")})')
-    c.drawString(100, 610, f'Final Score: {scores["finalScore"]}%')
-    c.save()
-
-    with open(filename, 'rb') as f:
-        pdf_content = f.read()
-    os.remove(filename)
-    return pdf_content
+    try:
+        filename = f'report_{vendor_id}.pdf'
+        c = canvas.Canvas(filename, pagesize=letter)
+        c.setFont('Helvetica', 12)
+        c.drawString(100, 750, 'Subcontractor Performance Report')
+        c.drawString(100, 730, f'Vendor ID: {vendor_id}')
+        c.drawString(100, 710, f'Vendor Log Name: {vendor_log_name}')
+        c.drawString(100, 690, f'Quality Score: {scores["qualityScore"]}% ({get_feedback(scores["qualityScore"], "Quality")})')
+        c.drawString(100, 670, f'Timeliness Score: {scores["timelinessScore"]}% ({get_feedback(scores["timelinessScore"], "Timeliness")})')
+        c.drawString(100, 650, f'Safety Score: {scores["safetyScore"]}% ({get_feedback(scores["safetyScore"], "Safety")})')
+        c.drawString(100, 630, f'Communication Score: {scores["communicationScore"]}% ({get_feedback(scores["communicationScore"], "Communication")})')
+        c.drawString(100, 610, f'Final Score: {scores["finalScore"]}%')
+        c.save()
+
+        with open(filename, 'rb') as f:
+            pdf_content = f.read()
+        os.remove(filename)
+        return pdf_content
+    except Exception as e:
+        logger.error(f"Error generating PDF: {str(e)}")
+        raise
 
 def determine_alert_flag(final_score: float, all_logs: list):
-    if not all_logs:
-        return False
-    # If final score is less than 50, flag an alert
-    if final_score < 50:
-        return True
-    # Otherwise, flag the vendor with the lowest final score
-    lowest_score = min([log['scores']['finalScore'] for log in all_logs])
-    return final_score == lowest_score
+    try:
+        if not all_logs:
+            return False
+        if final_score < 50:
+            return True
+        lowest_score = min([log['scores']['finalScore'] for log in all_logs])
+        return final_score == lowest_score
+    except Exception as e:
+        logger.error(f"Error determining alert flag: {str(e)}")
+        raise
 
 @app.post('/score')
 async def score_vendor(log: VendorLog, authorization: str = Header(...)):
-    if authorization != f'Bearer {API_KEY}':
-        raise HTTPException(status_code=401, detail='Invalid API key')
-
-    # Calculate scores
-    scores = calculate_scores(log)
-    
-    # Generate PDF
-    pdf_content = generate_pdf(log.vendorId, log.vendorLogName, scores)
-    pdf_base64 = base64.b64encode(pdf_content).decode('utf-8')
-
-    # Determine alert flag
-    alert_flag = determine_alert_flag(scores['finalScore'], vendor_logs)
-
-    # Store the log and scores
-    vendor_logs.append({
-        'vendorLogId': log.vendorLogId,
-        'vendorId': log.vendorId,
-        'vendorLogName': log.vendorLogName,
-        'scores': scores,
-        'extracted': True
-    })
-
-    # Return the response to Salesforce
-    return {
-        'vendorLogId': log.vendorLogId,
-        'vendorId': log.vendorId,
-        'vendorLogName': log.vendorLogName,
-        'qualityScore': scores['qualityScore'],
-        'timelinessScore': scores['timelinessScore'],
-        'safetyScore': scores['safetyScore'],
-        'communicationScore': scores['communicationScore'],
-        'finalScore': scores['finalScore'],
-        'pdfContent': pdf_base64,
-        'alert': alert_flag
-    }
+    try:
+        logger.info(f"Received Vendor Log: {log}")
+        if authorization != f'Bearer {API_KEY}':
+            raise HTTPException(status_code=401, detail='Invalid API key')
+
+        # Calculate scores
+        scores = calculate_scores(log)
+        
+        # Generate PDF
+        pdf_content = generate_pdf(log.vendorId, log.vendorLogName, scores)
+        pdf_base64 = base64.b64encode(pdf_content).decode('utf-8')
+
+        # Determine alert flag
+        alert_flag = determine_alert_flag(scores['finalScore'], vendor_logs)
+
+        # Store the log and scores
+        vendor_logs.append({
+            'vendorLogId': log.vendorLogId,
+            'vendorId': log.vendorId,
+            'vendorLogName': log.vendorLogName,
+            'workDetails': log.workDetails,
+            'qualityReport': log.qualityReport,
+            'incidentLog': log.incidentLog,
+            'workCompletionDate': log.workCompletionDate,
+            'actualCompletionDate': log.actualCompletionDate,
+            'delayDays': log.delayDays,
+            'project': log.project,
+            'scores': scores,
+            'extracted': True
+        })
+
+        # Return the response to Salesforce
+        return {
+            'vendorLogId': log.vendorLogId,
+            'vendorId': log.vendorId,
+            'vendorLogName': log.vendorLogName,
+            'qualityScore': scores['qualityScore'],
+            'timelinessScore': scores['timelinessScore'],
+            'safetyScore': scores['safetyScore'],
+            'communicationScore': scores['communicationScore'],
+            'finalScore': scores['finalScore'],
+            'pdfContent': pdf_base64,
+            'alert': alert_flag
+        }
+    except Exception as e:
+        logger.error(f"Error in /score endpoint: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error processing vendor log: {str(e)}")
 
 @app.get('/', response_class=HTMLResponse)
 async def get_dashboard():
-    html_content = """
-    <html>
-        <head>
-            <title>Subcontractor Performance Score App</title>
-            <style>
-                body { font-family: Arial, sans-serif; margin: 20px; }
-                table { width: 100%; border-collapse: collapse; margin-top: 20px; }
-                th, td { border: 1px solid #ddd; padding: 8px; text-align: left; }
-                th { background-color: #f2f2f2; }
-                h1, h2 { text-align: center; }
-            </style>
-        </head>
-        <body>
-            <h1>SUBCONTRACTOR PERFORMANCE SCORE APP GENERATOR</h1>
-            <h2>VENDOR LOGS</h2>
-            <table>
-                <tr>
-                    <th>VENDOR ID</th>
-                    <th>DETAILS EXTRACTED FROM SALESFORCE FROM VENDOR LOG OBJECT</th>
-                    <th>FINAL SCORE CALCULATED BY APP</th>
-                </tr>
-    """
-
-    for log in vendor_logs:
-        html_content += f"""
-                <tr>
-                    <td>{log['vendorId']}</td>
-                    <td>{'YES' if log['extracted'] else 'NO'}</td>
-                    <td>{log['scores']['finalScore'] if log['extracted'] else 'NOT CALCULATED'}</td>
-                </tr>
+    try:
+        html_content = """
+        <html>
+            <head>
+                <title>Subcontractor Performance Score App</title>
+                <style>
+                    body { font-family: Arial, sans-serif; margin: 20px; }
+                    table { width: 100%; border-collapse: collapse; margin-top: 20px; }
+                    th, td { border: 1px solid #ddd; padding: 8px; text-align: left; }
+                    th { background-color: #f2f2f2; }
+                    h1, h2 { text-align: center; }
+                </style>
+            </head>
+            <body>
+                <h1>SUBCONTRACTOR PERFORMANCE SCORE APP GENERATOR</h1>
+                <h2>VENDOR LOGS SUBMISSION</h2>
+                <table>
+                    <tr>
+                        <th>Vendor ID</th>
+                        <th>Vendor Log Name</th>
+                        <th>Project</th>
+                        <th>Work Completion Percentage</th>
+                        <th>Quality Percentage</th>
+                        <th>Incident Severity</th>
+                        <th>Work Completion Date</th>
+                        <th>Actual Completion Date</th>
+                        <th>Delay Days</th>
+                    </tr>
         """
 
-    html_content += """
-            </table>
-            <h2>SUBCONTRACTOR PERFORMANCE SCORES</h2>
-            <table>
-                <tr>
-                    <th>Vendor ID</th>
-                    <th>Vendor Log Name</th>
-                    <th>Quality Score</th>
-                    <th>Timeliness Score</th>
-                    <th>Safety Score</th>
-                    <th>Communication Score</th>
-                    <th>Final Score</th>
-                    <th>Alert Flag</th>
-                </tr>
-    """
-
-    for log in vendor_logs:
-        scores = log['scores']
-        alert_flag = determine_alert_flag(scores['finalScore'], vendor_logs)
-        html_content += f"""
-                <tr>
-                    <td>{log['vendorId']}</td>
-                    <td>{log['vendorLogName']}</td>
-                    <td>{scores['qualityScore']}%</td>
-                    <td>{scores['timelinessScore']}%</td>
-                    <td>{scores['safetyScore']}%</td>
-                    <td>{scores['communicationScore']}%</td>
-                    <td>{scores['finalScore']}%</td>
-                    <td>{'Checked' if alert_flag else 'Unchecked'}</td>
-                </tr>
+        for log in vendor_logs:
+            html_content += f"""
+                    <tr>
+                        <td>{log['vendorId']}</td>
+                        <td>{log['vendorLogName']}</td>
+                        <td>{log['project']}</td>
+                        <td>{log['workDetails']}</td>
+                        <td>{log['qualityReport']}</td>
+                        <td>{log['incidentLog']}</td>
+                        <td>{log['workCompletionDate']}</td>
+                        <td>{log['actualCompletionDate']}</td>
+                        <td>{log['delayDays']}</td>
+                    </tr>
+            """
+
+        html_content += """
+                </table>
+                <h2>SUBCONTRACTOR PERFORMANCE SCORES</h2>
+                <table>
+                    <tr>
+                        <th>Vendor ID</th>
+                        <th>Vendor Log Name</th>
+                        <th>Project</th>
+                        <th>Quality Score</th>
+                        <th>Timeliness Score</th>
+                        <th>Safety Score</th>
+                        <th>Communication Score</th>
+                        <th>Final Score</th>
+                        <th>Alert Flag</th>
+                    </tr>
         """
 
-    html_content += """
-            </table>
-        </body>
-    </html>
-    """
-    return HTMLResponse(content=html_content)
+        for log in vendor_logs:
+            scores = log['scores']
+            alert_flag = determine_alert_flag(scores['finalScore'], vendor_logs)
+            html_content += f"""
+                    <tr>
+                        <td>{log['vendorId']}</td>
+                        <td>{log['vendorLogName']}</td>
+                        <td>{log['project']}</td>
+                        <td>{scores['qualityScore']}%</td>
+                        <td>{scores['timelinessScore']}%</td>
+                        <td>{scores['safetyScore']}%</td>
+                        <td>{scores['communicationScore']}%</td>
+                        <td>{scores['finalScore']}%</td>
+                        <td>{'Checked' if alert_flag else 'Unchecked'}</td>
+                    </tr>
+            """
+
+        html_content += """
+                </table>
+            </body>
+        </html>
+        """
+        return HTMLResponse(content=html_content)
+    except Exception as e:
+        logger.error(f"Error in / endpoint: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error generating dashboard: {str(e)}")
 
 if __name__ == "__main__":
     import uvicorn