Update app.py

app.py

@@ -13,21 +13,32 @@ def process_contract(pdf_file, object_name):
     try:
         token, instance_url = get_token()
     except Exception as e:
-        return f"❌ {str(e)}", None
+        return f"❌ Salesforce authentication failed: {str(e)}", None
 
     # Step 1: OCR
-    text_data = extract_text_from_pdf(pdf_file.name)  # ✅ FIXED: use .name to get file path
+    try:
+        text_data = extract_text_from_pdf(pdf_file.name)  # Use .name to get file path
+        if not text_data:
+            return "⚠️ No text extracted from PDF. It may be empty or require OCR.", None
+    except Exception as e:
+        return f"❌ Failed to extract text from PDF: {str(e)}", None
 
     # Step 2: AI Mapping
-    mappings = run_ai_mapping(text_data)
+    try:
+        mappings = run_ai_mapping(text_data)
+    except Exception as e:
+        return f"❌ AI mapping failed: {str(e)}", None
 
     # Step 3: Create Salesforce record
-    record_response = create_record(object_name, mappings, token, instance_url)
-    if 'id' in record_response:
-        attach_pdf(record_response['id'], pdf_file.name, token, instance_url)
-        return f"✅ Record created: {record_response['id']}", mappings
-    else:
-        return f"❌ Failed to create record: {record_response}", mappings
+    try:
+        record_response = create_record(object_name, mappings, token, instance_url)
+        if 'id' in record_response:
+            attach_pdf(record_response['id'], pdf_file.name, token, instance_url)
+            return f"✅ Record created: {record_response['id']}", mappings
+        else:
+            return f"❌ Failed to create record: {record_response}", mappings
+    except Exception as e:
+        return f"❌ Failed to create Salesforce record: {str(e)}", mappings
 
 def get_salesforce_object_names():
     try:
@@ -35,7 +46,7 @@ def get_salesforce_object_names():
         objects = get_salesforce_objects(token, instance_url)
         return [obj['name'] for obj in objects if obj.get('createable')]
     except Exception as e:
-        return [f"❌ {str(e)}"]
+        return [f"❌ Failed to fetch Salesforce objects: {str(e)}"]
 
 with gr.Blocks() as demo:
     gr.Markdown("## 📄 Smart Contract Migrator (Gradio)")