Update backup_features.py

backup_features.py

@@ -1,30 +1,43 @@
-def extract_entities(text):
-    return {
-        "parties": ["Party A", "Party B"],
-        "dates": ["January 1, 2024"],
-        "amounts": ["$10,000"]
-    }
+import re
 
 def extract_clauses(text):
-    return ["Confidentiality", "Term", "IP Ownership"]
+    return [
+        {
+            "title": "Confidentiality",
+            "eli5": "Keep shared secrets safe.",
+            "simple": "You must protect confidential information.",
+            "pro": "Recipient shall maintain strict confidentiality over protected disclosures."
+        },
+        {
+            "title": "Non-Disclosure",
+            "eli5": "Don't tell anyone.",
+            "simple": "You agree not to disclose protected information.",
+            "pro": "Recipient shall not disseminate any protected information to unauthorized parties."
+        }
+    ]
 
-def risk_analysis(text):
+def get_risks(text):
     return [
-        "Very broad confidentiality scope.",
-        "No defined liability cap.",
-        "Unilateral termination clause.",
-        "Unlimited data retention.",
-        "Missing dispute resolution mechanism."
+        "Broad confidentiality scope",
+        "Indefinite protection period",
+        "Unilateral termination clause",
+        "Unclear data-handling terms",
+        "No liability limitation"
     ]
 
 def fairness_score(text):
-    user = 35
-    company = 65
-    return (50, user, company)
+    return {"user": 40, "company": 60}
+
+def extract_entities(text):
+    return {
+        "parties": re.findall(r"[A-Z][a-z]+ [A-Z][a-z]+", text),
+        "dates": re.findall(r"\b\d{4}\b", text),
+        "money": re.findall(r"\$\d+", text)
+    }
 
-def alternative_clauses():
+def alternative_clauses(text):
     return [
-        "A mutual confidentiality version is commonly used.",
-        "Liability caps between both parties are standard.",
-        "Shorter NDA duration (1–2 years) is common."
+        "Confidentiality limited to 3 years.",
+        "Mutual NDA instead of one-way.",
+        "Liability capped at actual damages."
     ]