Update app.py

app.py

@@ -1,317 +1,82 @@
-from fastapi import FastAPI, Request
-from fastapi.responses import HTMLResponse
+import gradio as gr
 from transformers import pipeline
-import uvicorn
-from typing import Optional
+import requests
+import json
+import pandas as pd  # Import pandas
 
 # Initialize sentiment analysis pipeline
 sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
 
-# FastAPI app
-app = FastAPI()
-
-# Function to calculate scores (moved outside the route for reusability)
-def calculate_scores_from_logs(logs: list) -> dict:
+# Function to calculate scores via API call
+def calculate_scores_from_logs(log_type_1, quality_score_1, delay_percentage_1, safety_compliance_1, feedback_1,
+                             log_type_2, quality_score_2, delay_percentage_2, safety_compliance_2, feedback_2, vendor_id, month):
     """
-    Calculates performance scores from a list of vendor logs.
+    Calculates performance scores by calling the Salesforce API.
 
     Args:
-        logs (list): A list of vendor log dictionaries.
+        log_type_1, log_type_2 (str): Type of log (Quality, Delay, Incident, Communication).
+        quality_score_1, quality_score_2 (float): Quality score.
+        delay_percentage_1, delay_percentage_2 (float): Delay percentage.
+        safety_compliance_1, safety_compliance_2 (str): Safety compliance (True/False).
+        feedback_1, feedback_2 (str): Feedback text.
+        vendor_id (str):  Vendor ID.
+        month (str): The month for which to calculate scores (YYYY-MM-DD).
 
     Returns:
-        dict: A dictionary containing the calculated scores and alert flag.
+        dict: A dictionary containing the calculated scores and alert flag, or an error message.
     """
-    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
+    logs = [
+        {
+            "log_type": log_type_1,
+            "quality_score": float(quality_score_1) if log_type_1 == "Quality" and quality_score_1 else None,
+            "delay_percentage": float(delay_percentage_1) if log_type_1 == "Delay" and delay_percentage_1 else None,
+            "safety_compliance": safety_compliance_1 == "True" if log_type_1 == "Incident" else None,
+            "feedback": feedback_1 if log_type_1 == "Communication" else "",
+        },
+        {
+            "log_type": log_type_2,
+            "quality_score": float(quality_score_2) if log_type_2 == "Quality" and quality_score_2 else None,
+            "delay_percentage": float(delay_percentage_2) if log_type_2 == "Delay" and delay_percentage_2 else None,
+            "safety_compliance": safety_compliance_2 == "True" if log_type_2 == "Incident" else None,
+            "feedback": feedback_2 if log_type_2 == "Communication" else "",
+        },
+    ]
+    payload = json.dumps(logs)
+    headers = {'Content-Type': 'application/json'}
+    #  Replace with your Salesforce API endpoint
+    salesforce_api_url = f"https://your-salesforce-domain.com/services/apexrest/VendorScoreCalculator?vendorId={vendor_id}&month={month}"  
 
-    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_endpoint(request: Request):
-    """
-    Endpoint to receive vendor logs from Salesforce and return calculated scores.
-    """
     try:
-        logs = await request.json()
-        scores = calculate_scores_from_logs(logs)  # Use the function
-        return scores
-    except Exception as e:
-        return {"error": str(e)}, 400
-
-@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"], select, 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>
-"""
-
-
-@app.get("/")
-async def main():
-    """
-    Root endpoint to display a user-friendly HTML form for testing.
-    """
-    return HTMLResponse(content=html_content)
-
+        response = requests.post(salesforce_api_url, headers=headers, data=payload)
+        response.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
+        return response.json()  # Return the JSON response from Salesforce
+    except requests.exceptions.RequestException as e:
+        error_message = f"Error calling Salesforce API: {e}"
+        print(error_message)
+        return {"error": error_message}  # Return a user-friendly error message
+
+# Gradio Interface
+iface = gr.Interface(
+    fn=calculate_scores_from_logs,
+    inputs=[
+        gr.Dropdown(["Quality", "Delay", "Incident", "Communication"], label="Log Type 1"),
+        gr.Number(label="Quality Score 1 (for Quality)"),
+        gr.Number(label="Delay Percentage 1 (for Delay)"),
+        gr.Radio(["True", "False"], label="Safety Compliance 1 (for Incident)"),
+        gr.Textbox(label="Feedback 1 (for Communication)"),
+        gr.Dropdown(["Quality", "Delay", "Incident", "Communication"], label="Log Type 2"),
+        gr.Number(label="Quality Score 2 (for Quality)"),
+        gr.Number(label="Delay Percentage 2 (for Delay)"),
+        gr.Radio(["True", "False"], label="Safety Compliance 2 (for Incident)"),
+        gr.Textbox(label="Feedback 2 (for Communication)"),
+        gr.Textbox(label="Vendor ID", placeholder="Enter Vendor ID"),  # Added Vendor ID
+        gr.Textbox(label="Month (YYYY-MM-DD)", placeholder="Enter Month (YYYY-MM-DD)"),  # Added Month
+    ],
+    outputs=gr.Label(label="Calculated Scores"),
+    title="Vendor Performance Score Calculator",
+    description="Calculate vendor performance scores based on log data and Vendor ID/Month."
+)
+
+# Run the Gradio interface
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    iface.launch(server_name="0.0.0.0", server_port=7860)