Update app.py

app.py

@@ -1,6 +1,8 @@
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, Request, Form
+from fastapi.responses import HTMLResponse
 from transformers import pipeline
 import uvicorn
+from typing import Optional
 
 # Initialize sentiment analysis pipeline
 sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
@@ -8,61 +10,298 @@ sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncas
 # FastAPI app
 app = FastAPI()
 
+# Function to calculate scores (moved outside the route for reusability)
+def calculate_scores_from_logs(logs: list) -> dict:
+    """
+    Calculates performance scores from a list of vendor logs.
+
+    Args:
+        logs (list): A list of vendor log dictionaries.
+
+    Returns:
+        dict: A dictionary containing the calculated scores and alert flag.
+    """
+    quality_score = 0
+    timeliness_score = 0
+    safety_score = 0
+    communication_score = 0
+    quality_count = 0
+    delay_count = 0
+    incident_count = 0
+    feedback_count = 0
+
+    for log in logs:
+        log_type = log.get("log_type")
+        if log_type == "Quality" and "quality_score" in log and log["quality_score"] is not None:
+            try:
+                quality_score += float(log["quality_score"])
+                quality_count += 1
+            except (ValueError, TypeError):
+                pass
+        if log_type == "Delay" and "delay_percentage" in log and log["delay_percentage"] is not None:
+            try:
+                timeliness_score += (100 - float(log["delay_percentage"]))
+                delay_count += 1
+            except (ValueError, TypeError):
+                pass
+        if log_type == "Incident":
+            safety_score += (100 if log.get("safety_compliance", False) else 0)
+            incident_count += 1
+        if "feedback" in log and log["feedback"]:
+            feedback = log["feedback"]
+            sentiment = sentiment_analyzer(feedback)[0]
+            sentiment_score = sentiment["score"] * 100 if sentiment["label"] == "POSITIVE" else 50
+            base_score = 90 if len(feedback) > 50 else 70
+            communication_score += base_score * (sentiment_score / 100)
+            feedback_count += 1
+
+    quality_score = quality_score / quality_count if quality_count > 0 else 0
+    timeliness_score = timeliness_score / delay_count if delay_count > 0 else 0
+    safety_score = safety_score / incident_count if incident_count > 0 else 0
+    communication_score = communication_score / feedback_count if feedback_count > 0 else 0
+    alert_flag = quality_score < 50 or timeliness_score < 50
+
+    return {
+        "quality_score": round(quality_score, 2),
+        "timeliness_score": round(timeliness_score, 2),
+        "safety_score": round(safety_score, 2),
+        "communication_score": round(communication_score, 2),
+        "alert_flag": alert_flag,
+    }
+
 @app.post("/calculate_scores")
-async def calculate_scores(request: Request):
+async def calculate_scores_endpoint(request: Request):
+    """
+    Endpoint to receive vendor logs from Salesforce and return calculated scores.
+    """
     try:
         logs = await request.json()
-        quality_score = 0
-        timeliness_score = 0
-        safety_score = 0
-        communication_score = 0
-        quality_count = 0
-        delay_count = 0
-        incident_count = 0
-        feedback_count = 0
-
-        for log in logs:
-            log_type = log.get("log_type")
-            if log_type == "Quality" and "quality_score" in log and log["quality_score"] is not None:
-                try:
-                    quality_score += float(log["quality_score"])
-                    quality_count += 1
-                except (ValueError, TypeError):
-                    pass
-            if log_type == "Delay" and "delay_percentage" in log and log["delay_percentage"] is not None:
-                try:
-                    timeliness_score += (100 - float(log["delay_percentage"]))
-                    delay_count += 1
-                except (ValueError, TypeError):
-                    pass
-            if log_type == "Incident":
-                safety_score += (100 if log.get("safety_compliance", False) else 0)
-                incident_count += 1
-            if "feedback" in log and log["feedback"]:
-                feedback = log["feedback"]
-                sentiment = sentiment_analyzer(feedback)[0]
-                sentiment_score = sentiment["score"] * 100 if sentiment["label"] == "POSITIVE" else 50
-                base_score = 90 if len(feedback) > 50 else 70
-                communication_score += base_score * (sentiment_score / 100)
-                feedback_count += 1
-
-        quality_score = quality_score / quality_count if quality_count > 0 else 0
-        timeliness_score = timeliness_score / delay_count if delay_count > 0 else 0
-        safety_score = safety_score / incident_count if incident_count > 0 else 0
-        communication_score = communication_score / feedback_count if feedback_count > 0 else 0
-
-        alert_flag = quality_score < 50 or timeliness_score < 50
-
-        return {
-            "quality_score": round(quality_score, 2),
-            "timeliness_score": round(timeliness_score, 2),
-            "safety_score": round(safety_score, 2),
-            "communication_score": round(communication_score, 2),
-            "alert_flag": alert_flag
-        }
+        scores = calculate_scores_from_logs(logs)  # Use the function
+        return scores
     except Exception as e:
         return {"error": str(e)}, 400
 
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+@app.get("/")
+async def main():
+    """
+    Root endpoint to display a user-friendly HTML form for testing.
+    """
+    html_content = """
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Vendor Score Calculator</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            margin: 0;
+            padding: 0;
+            background-color: #f4f4f4;
+            display: flex;
+            justify-content: center;
+            align-items: center;
+            min-height: 100vh;
+        }
+
+        .container {
+            background-color: #fff;
+            padding: 20px;
+            border-radius: 8px;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+            width: 80%;
+            max-width: 600px;
+        }
+
+        h1 {
+            text-align: center;
+            color: #333;
+        }
+
+        form {
+            margin-top: 20px;
+        }
+
+        label {
+            display: block;
+            margin-bottom: 5px;
+            color: #555;
+        }
+
+        input[type="text"], textarea {
+            width: 100%;
+            padding: 8px 10px;
+            margin-bottom: 15px;
+            border: 1px solid #ddd;
+            border-radius: 4px;
+            box-sizing: border-box;
+        }
+        textarea {
+            resize: none;
+            height: 80px;
+        }
+
+        button[type="submit"] {
+            background-color: #4caf50;
+            color: white;
+            padding: 10px 15px;
+            border: none;
+            border-radius: 4px;
+            cursor: pointer;
+            width: 100%;
+            transition: background-color 0.3s ease;
+            font-size: 16px;
+            display: block;
+            margin: 20px auto;
+
+        }
+
+        button[type="submit"]:hover {
+            background-color: #45a049;
+        }
+
+        #results {
+            margin-top: 30px;
+            padding: 15px;
+            background-color: #e3e3e3;
+            border-radius: 4px;
+            text-align: center;
+            display: none;
+        }
+
+        #results h2{
+            color: #333;
+        }
+
+        #results p {
+            margin: 10px 0;
+            font-size: 18px;
+            color: #333;
+        }
+        #results p strong{
+            color: #2196f3;
+        }
+
+        @media screen and (max-width: 768px) {
+            .container {
+                width: 95%;
+            }
+        }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h1>Vendor Performance Score Calculator</h1>
+        <form id="scoreForm">
+            <label for="log_type_1">Log Type 1:</label>
+            <select id="log_type_1" name="log_type_1">
+                <option value="Quality">Quality</option>
+                <option value="Delay">Delay</option>
+                <option value="Incident">Incident</option>
+                <option value="Communication">Communication</option>
+            </select>
+            <input type="text" id="quality_score_1" name="quality_score_1" placeholder="Quality Score (for Quality)">
+            <input type="text" id="delay_percentage_1" name="delay_percentage_1" placeholder="Delay Percentage (for Delay)">
+            <label for="safety_compliance_1">Safety Compliance 1:</label>
+            <select id="safety_compliance_1" name="safety_compliance_1">
+                <option value="true">Compliant</option>
+                <option value="false">Non-Compliant</option>
+            </select>
+            <textarea id="feedback_1" name="feedback_1" placeholder="Feedback (for Communication)"></textarea>
+
+            <label for="log_type_2">Log Type 2:</label>
+             <select id="log_type_2" name="log_type_2">
+                <option value="Quality">Quality</option>
+                <option value="Delay">Delay</option>
+                <option value="Incident">Incident</option>
+                 <option value="Communication">Communication</option>
+            </select>
+            <input type="text" id="quality_score_2" name="quality_score_2" placeholder="Quality Score (for Quality)">
+            <input type="text" id="delay_percentage_2" name="delay_percentage_2" placeholder="Delay Percentage (for Delay)">
+             <label for="safety_compliance_2">Safety Compliance 2:</label>
+            <select id="safety_compliance_2" name="safety_compliance_2">
+                <option value="true">Compliant</option>
+                <option value="false">Non-Compliant</option>
+            </select>
+            <textarea id="feedback_2" name="feedback_2" placeholder="Feedback (for Communication)"></textarea>
+
+            <button type="submit">Calculate Scores</button>
+        </form>
+
+        <div id="results">
+            <h2>Performance Scores</h2>
+            <p>Quality Score: <strong id="display_quality_score"></strong></p>
+            <p>Timeliness Score: <strong id="display_timeliness_score"></strong></p>
+            <p>Safety Score: <strong id="display_safety_score"></strong></p>
+            <p>Communication Score: <strong id="display_communication_score"></strong></p>
+            <p>Alert Flag: <strong id="display_alert_flag"></strong></p>
+        </div>
+    </div>
+
+    <script>
+        document.getElementById('scoreForm').addEventListener('submit', async (event) => {
+            event.preventDefault();
+
+            const log1_type = document.getElementById('log_type_1').value;
+            const quality_score_1 = document.getElementById('quality_score_1').value;
+            const delay_percentage_1 = document.getElementById('delay_percentage_1').value;
+            const safety_compliance_1 = document.getElementById('safety_compliance_1').value;
+            const feedback_1 = document.getElementById('feedback_1').value;
+
+             const log2_type = document.getElementById('log_type_2').value;
+            const quality_score_2 = document.getElementById('quality_score_2').value;
+            const delay_percentage_2 = document.getElementById('delay_percentage_2').value;
+            const safety_compliance_2 = document.getElementById('safety_compliance_2').value;
+            const feedback_2 = document.getElementById('feedback_2').value;
+
+
+            const logs = [
+                {
+                    log_type: log1_type,
+                    quality_score: log1_type === 'Quality' ? quality_score_1 : null,
+                    delay_percentage: log1_type === 'Delay' ? delay_percentage_1 : null,
+                    safety_compliance: log1_type === 'Incident' ? safety_compliance_1 === 'true' : null,
+                    feedback: log1_type === 'Communication' ? feedback_1 : '',
+                },
+                {
+                    log_type: log2_type,
+                    quality_score: log2_type === 'Quality' ? quality_score_2 : null,
+                    delay_percentage: log2_type === 'Delay' ? delay_percentage_2 : null,
+                    safety_compliance: log2_type === 'Incident' ? safety_compliance_2 === 'true' : null,
+                    feedback: log2_type === 'Communication' ? feedback_2 : '',
+                },
+            ];
+
+            try {
+                const response = await fetch('/calculate_scores', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json',
+                    },
+                    body: JSON.stringify(logs),
+                });
+
+                if (!response.ok) {
+                    throw new Error(`HTTP error! status: ${response.status}`);
+                }
+
+                const data = await response.json();
+
+                document.getElementById('display_quality_score').textContent = data.quality_score;
+                document.getElementById('display_timeliness_score').textContent = data.timeliness_score;
+                document.getElementById('display_safety_score').textContent = data.safety_score;
+                document.getElementById('display_communication_score').textContent = data.communication_score;
+                document.getElementById('display_alert_flag').textContent = data.alert_flag;
+                document.getElementById('results').style.display = 'block';
+
+            } catch (error) {
+                alert('Error calculating scores. Check console');
+                console.error('Error:', error);
+            }
+        });
+    </script>
+</body>
+</html>
+