Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 import tempfile
 from tqdm import tqdm
 import re
-from ai_mapping import extract_key_values_with_layoutlm, run_ai_mapping_with_layoutlm
+from ai_mapping import extract_key_values_with_layoutlm, run_ai_mapping_with_layoutlm, extract_clauses
 from ocr_utils import extract_text_from_pdf_with_tesseract_or_layoutlm
 from salesforce_utils import get_token, create_or_update_record
 
@@ -28,8 +28,8 @@ def save_temp_file(pdf_bytes):
 def detect_risks(data):
     """Detect risks (e.g., missing dates, large amounts)."""
     risks = []
-    if not data.get("Date"):
-        risks.append("No expiration date detected - potential obligation risk.")
+    if not data.get("Agreement Start Date") and not data.get("Agreement End Date"):
+        risks.append("No agreement dates detected - potential obligation risk.")
     if data.get("Amount") and float(data.get("Amount", "0").replace('$', '').replace(',', '')) > 1000000:
         risks.append("Large amount detected - review for financial risk.")
     return risks
@@ -58,6 +58,10 @@ def process_contract(pdf_bytes, object_type):
         print(f"Extraction failed: {key_data.get('error', 'Unknown error')}")
         return f"❌ Extraction failed: {key_data.get('error', 'Unknown error')}", {}, [], "0/1"
 
+    print("Extracting clauses")
+    clauses = extract_clauses(page_data)
+    print(f"Extracted clauses: {clauses}")
+
     print("Detecting risks")
     risks = detect_risks(key_data)
     print(f"Detected risks: {risks}")
@@ -66,6 +70,8 @@ def process_contract(pdf_bytes, object_type):
     # Mock CLM fields with Salesforce-ready structure
     clm_fields = {"Name": f"Contract_{len(contract_data) + 1}", "Type__c": object_type, "Status__c": status}
     clm_fields.update({k: v for k, v in key_data.items() if k not in ["status", "error", "key_values"]})
+    for clause_name, clause_text in clauses.items():
+        clm_fields[f"{clause_name}_Text__c"] = clause_text
 
     # Optional Salesforce sync
     try:
@@ -81,6 +87,7 @@ def process_contract(pdf_bytes, object_type):
     contract_id = f"Contract_{len(contract_data) + 1}"
     contract_data[contract_id] = {
         "data": key_data,
+        "clauses": clauses,
         "risks": risks,
         "clm_fields": clm_fields,
         "status": status
@@ -108,26 +115,30 @@ with gr.Blocks(title="Contract Intelligence App") as demo:
     process_button = gr.Button("Process Contracts")
     status_output = gr.Textbox(label="Status", interactive=False)
     extracted_data_output = gr.JSON(label="Extracted Data")
+    clauses_output = gr.JSON(label="Extracted Clauses")
     risks_output = gr.Textbox(label="Detected Risks", interactive=False)
 
     def process_and_display(files, obj_type):
         if not files:
-            return "❌ No files uploaded.", {}, "No risks detected", gr.update(value="0/0")
+            return "❌ No files uploaded.", {}, {}, "No risks detected", gr.update(value="0/0")
         results = []
         all_data = {}
+        all_clauses = {}
         all_risks = []
         for i, file in enumerate(files):
             status, data, risks, _ = process_contract(file, obj_type)
+            clauses = contract_data[f"Contract_{len(contract_data)}"]["clauses"]  # Get clauses from latest contract
             results.append(f"{status} - File: File_{i}")
             all_data.update({f"File_{i}": data})
+            all_clauses.update({f"File_{i}": clauses})
             all_risks.extend(risks)
         progress = f"{len(files)}/{len(files)}"
-        return "\n".join(results), all_data, "\n".join(all_risks) if all_risks else "No risks detected", gr.update(value=progress)
+        return "\n".join(results), all_data, all_clauses, "\n".join(all_risks) if all_risks else "No risks detected", gr.update(value=progress)
 
     process_button.click(
         fn=process_and_display,
         inputs=[file_input, object_type],
-        outputs=[status_output, extracted_data_output, risks_output, upload_progress]
+        outputs=[status_output, extracted_data_output, clauses_output, risks_output, upload_progress]
     )
 
     with gr.Tab("Contract Repository"):