Update app.py

app.py

@@ -716,17 +716,13 @@ async def analyze_legal_document(
             "subscription_tier": current_user.subscription_tier
         }
         
-        # Add premium features if user has access
-        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
+        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
         
         return response