Update app.py

app.py

@@ -717,7 +717,7 @@ async def analyze_legal_document(
         }
         
         # Add premium features if user has access
-        if current_user.subscription_tier == "free_tier":   # was premium
+        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)
@@ -1415,12 +1415,8 @@ async def paypal_webhook(request: Request):
         # Return 200 even on error to acknowledge receipt to PayPal
         return {"status": "error", "message": str(e)}
 @app.get("/download_risk_chart")
-async def download_risk_chart(current_user: User = Depends(get_current_active_user)):
-    if "detailed_risk_assessment" not in SUBSCRIPTION_TIERS[current_user.subscription_tier]["features"]:
-        raise HTTPException(
-            status_code=403,
-            detail=f"The risk visualization feature is not available in your {current_user.subscription_tier} subscription."
-        )
+async def download_risk_chart():
+    """Generate and return a risk assessment chart as an image file."""
     try:
         risk_scores = {
             "Liability": 11,
@@ -1547,4 +1543,4 @@ async def startup_event():
         print(f"Error during startup: {str(e)}")
 
 if __name__ == "__main__":
-    uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=True)
+    uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=True)