Update app.py

app.py

@@ -15,12 +15,11 @@ SF_PASSWORD = os.getenv("SF_PASSWORD")
 SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
 SF_LOGIN_URL = os.getenv("SF_LOGIN_URL", "https://login.salesforce.com")
 SF_OBJECT_NAME = os.getenv("SF_OBJECT_NAME", "Agent_Prospect__c")
-SF_SCORE_FIELD = os.getenv("SF_SCORE_FIELD", "Suitability_Srode__c")
+SF_SCORE_FIELD = os.getenv("SF_SCORE_FIELD", "Suitability_Score__c")
 SF_LINK_FIELD = os.getenv("SF_RESUME_FIELD_LINK", "Resume_File_Link__c")
-SF_RECORD_ID = os.getenv("SF_RECORD_ID")
 
 # Validate required credentials
-required = ["SF_USERNAME", "SF_PASSWORD", "SF_SECURITY_TOKEN", "SF_RECORD_ID"]
+required = ["SF_USERNAME", "SF_PASSWORD", "SF_SECURITY_TOKEN"]
 missing = [var for var in required if not os.getenv(var)]
 if missing:
     raise ValueError(f"Missing required .env variables: {', '.join(missing)}")
@@ -46,9 +45,6 @@ classifier = pipeline("text-classification", model="nlptown/bert-base-multilingu
 
 def process_resume(file):
     try:
-        record_id = SF_RECORD_ID
-        print(f"Processing resume for record ID: {record_id}")
-
         # Extract text from PDF
         with pdfplumber.open(file.name) as pdf:
             extracted_text = "\n".join([page.extract_text() or "" for page in pdf.pages])
@@ -88,7 +84,19 @@ def process_resume(file):
         content_doc_id = query_result["records"][0]["ContentDocumentId"]
         print(f"ContentDocumentId: {content_doc_id}")
 
-        # Link file to Salesforce record
+        # Create a new Agent_Prospect__c record
+        try:
+            record_result = sf.__getattr__(SF_OBJECT_NAME).create({
+                SF_SCORE_FIELD: score,
+                SF_LINK_FIELD: ""  # Placeholder; will update with download link
+            })
+            record_id = record_result.get("id")
+            print(f"New Agent_Prospect__c record created: {record_id}")
+        except Exception as e:
+            print(f"Error creating Agent_Prospect__c record: {str(e)}")
+            raise
+
+        # Link file to the new Salesforce record
         try:
             sf.ContentDocumentLink.create({
                 "ContentDocumentId": content_doc_id,
@@ -105,14 +113,17 @@ def process_resume(file):
         download_link = f"https://{sf.sf_instance}/sfc/servlet.shepherd/document/download/{content_doc_id}"
         print(f"Download link: {download_link}")
 
-        # Update record with score and link
-        sf.__getattr__(SF_OBJECT_NAME).update(record_id, {
-            SF_SCORE_FIELD: score,
-            SF_LINK_FIELD: download_link
-        })
-        print("Salesforce record updated successfully.")
+        # Update the record with the download link
+        try:
+            sf.__getattr__(SF_OBJECT_NAME).update(record_id, {
+                SF_LINK_FIELD: download_link
+            })
+            print("Salesforce record updated with download link.")
+        except Exception as e:
+            print(f"Error updating record with download link: {str(e)}")
+            raise
 
-        return f"{summary}\n\n✅ Score and resume uploaded to Salesforce.\n📎 [Download Resume]({download_link})"
+        return f"{summary}\n\n✅ New record created and resume uploaded to Salesforce.\n📎 [Download Resume]({download_link})\nRecord ID: {record_id}"
 
     except Exception as e:
         return f"❌ Error: {str(e)}"
@@ -123,5 +134,5 @@ gr.Interface(
     inputs=gr.File(label="Upload Resume (PDF)", file_types=[".pdf"]),
     outputs="text",
     title="LIC Resume AI Scorer",
-    description="Upload a resume PDF. It will be scored and stored in Salesforce automatically."
+    description="Upload a resume PDF. A new record will be created in Salesforce with the score and resume."
 ).launch(share=False)