Update app.py

app.py

@@ -7,6 +7,7 @@ import os
 import logging
 from datetime import datetime
 from fastapi.responses import HTMLResponse
+from simple_salesforce import Salesforce
 
 # Set up logging to capture errors and debug information
 logging.basicConfig(level=logging.INFO)
@@ -14,11 +15,24 @@ 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")
+# Salesforce credentials
+SF_USERNAME = "scores@app.com"
+SF_PASSWORD = "Internal@1"
+SF_SECURITY_TOKEN = "NbUKcTx45azba5HEdntE9YAh"
+SF_DOMAIN = "login"
+
+# Initialize Salesforce connection
+try:
+    sf = Salesforce(
+        username=SF_USERNAME,
+        password=SF_PASSWORD,
+        security_token=SF_SECURITY_TOKEN,
+        domain=SF_DOMAIN
+    )
+    logger.info("Successfully connected to Salesforce")
+except Exception as e:
+    logger.error(f"Failed to connect to Salesforce: {str(e)}")
+    raise ValueError("Failed to connect to Salesforce")
 
 # VendorLog model to match Salesforce data
 class VendorLog(BaseModel):
@@ -136,8 +150,9 @@ def determine_alert_flag(final_score: float, all_logs: list):
 async def score_vendor(log: VendorLog, authorization: str = Header(...)):
     try:
         logger.info(f"Received Vendor Log: {log}")
-        if authorization != f'Bearer {API_KEY}':
-            raise HTTPException(status_code=401, detail='Invalid API key')
+        # Validate Salesforce session or token (optional, depending on your setup)
+        if not sf.session_id:
+            raise HTTPException(status_code=401, detail='Salesforce session invalid')
 
         # Calculate scores
         scores = calculate_scores(log)
@@ -165,7 +180,26 @@ async def score_vendor(log: VendorLog, authorization: str = Header(...)):
             'extracted': True
         })
 
-        # Return the response to Salesforce
+        # Optionally, save scores back to Salesforce (example mapping to a custom object)
+        try:
+            sf.Vendor_Log__c.create({
+                'Vendor_Log_Id__c': log.vendorLogId,
+                'Vendor_Id__c': log.vendorId,
+                'Vendor_Log_Name__c': log.vendorLogName,
+                'Quality_Score__c': scores['qualityScore'],
+                'Timeliness_Score__c': scores['timelinessScore'],
+                'Safety_Score__c': scores['safetyScore'],
+                'Communication_Score__c': scores['communicationScore'],
+                'Final_Score__c': scores['finalScore'],
+                'Alert_Flag__c': alert_flag,
+                'PDF_Content__c': pdf_base64  # Assuming a text field for base64 PDF
+            })
+            logger.info(f"Successfully saved scores to Salesforce for Vendor Log: {log.vendorLogId}")
+        except Exception as e:
+            logger.error(f"Error saving to Salesforce: {str(e)}")
+            # Continue even if Salesforce save fails to ensure response is returned
+
+        # Return the response to the caller
         return {
             'vendorLogId': log.vendorLogId,
             'vendorId': log.vendorId,