Update app.py

app.py

@@ -716,13 +716,16 @@ async def analyze_legal_document(
             "subscription_tier": current_user.subscription_tier
         }
         
-        if "detailed_risk_assessment" in SUBSCRIPTION_TIERS[current_user.subscription_tier]["features"]:
-            detailed_risk = get_detailed_risk_info(text)
-            response["detailed_risk_assessment"] = detailed_risk
-
-        if "contract_clause_analysis" in SUBSCRIPTION_TIERS[current_user.subscription_tier]["features"]:
-            clauses = analyze_contract_clauses(text)
-            response["contract_clauses"] = clauses
+        if current_user.subscription_tier == "premium_tier":  
+            # Add detailed risk assessment
+            if "detailed_risk_assessment" in SUBSCRIPTION_TIERS[current_user.subscription_tier]["features"]:
+                detailed_risk = get_detailed_risk_info(text)
+                response["detailed_risk_assessment"] = detailed_risk
+            
+            # Add contract clause analysis
+            if "contract_clause_analysis" in SUBSCRIPTION_TIERS[current_user.subscription_tier]["features"]:
+                clauses = analyze_contract_clauses(text)
+                response["contract_clauses"] = clauses
         
         return response