Update app.py

app.py

@@ -9,7 +9,7 @@ from datetime import datetime
 from fastapi.responses import HTMLResponse
 from simple_salesforce import Salesforce
 from dotenv import load_dotenv
-from datasets import load_dataset  # For Hugging Face
+from datasets import load_dataset
 
 # Load environment variables
 load_dotenv()
@@ -103,7 +103,7 @@ def fetch_huggingface_records(dataset_name: str = "imdb"):
         dataset = load_dataset(dataset_name)
         logger.info(f"Successfully fetched dataset: {dataset_name}")
         records = [record for record in dataset['train']]  # Assuming 'train' split
-        return records[:10]  # Limit to 10 records for demonstration
+        return records[:10]  # Limit to 10 records
     except Exception as e:
         logger.error(f"Error fetching Hugging Face dataset {dataset_name}: {str(e)}")
         return []
@@ -238,7 +238,7 @@ def store_scores_in_salesforce(log: VendorLog, scores: dict, pdf_content: bytes,
             'Alert_Flag__c': alert_flag
         })
         score_record_id = score_record['id']
-        logger.info(f"Successfully created Subcontractor_Performance_Score__c record with ID: {score_record_id}")
+        logger.info(f"Created Subcontractor_Performance_Score__c record with ID: {score_record_id}")
 
         pdf_base64 = base64.b64encode(pdf_content).decode('utf-8')
         content_version = sf.ContentVersion.create({
@@ -256,7 +256,7 @@ def store_scores_in_salesforce(log: VendorLog, scores: dict, pdf_content: bytes,
         
         pdf_url = f"https://{sf.sf_instance}/sfc/servlet.shepherd/document/download/{content_document_id}"
         sf.Subcontractor_Performance_Score__c.update(score_record_id, {'PDF_Link__c': pdf_url})
-        logger.info(f"Successfully updated Subcontractor_Performance_Score__c record with PDF URL: {pdf_url}")
+        logger.info(f"Updated Subcontractor_Performance_Score__c record with PDF URL: {pdf_url}")
     except Exception as e:
         logger.error(f"Error storing scores in Salesforce: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Error storing scores: {str(e)}")
@@ -506,4 +506,23 @@ async def get_huggingface_records():
 
 @app.get('/debug')
 async def debug_info():
-    """Return
+    """Return debug information about Salesforce and Hugging Face."""
+    try:
+        log_count = sf.query("SELECT COUNT() FROM Vendor_Log__c")['totalSize']
+        fields = [f['name'] for f in sf.Vendor_Log__c.describe()['fields']]
+        score_fields = [f['name'] for f in sf.Subcontractor_Performance_Score__c.describe()['fields']]
+        hf_records = fetch_huggingface_records()
+        return {
+            "salesforce_connected": True,
+            "vendor_log_count": log_count,
+            "vendor_log_fields": fields,
+            "score_fields": score_fields,
+            "huggingface_records_sample": hf_records
+        }
+    except Exception as e:
+        logger.error(f"Debug error: {str(e)}")
+        return {"salesforce_connected": False, "error": str(e)}
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)