Update auth.py

auth.py

@@ -68,8 +68,8 @@ SUBSCRIPTION_TIERS = {
             "documents_per_month": 3,
             "video_size_mb": 0,
             "audio_size_mb": 0,
-            "daily_api_calls": 10,              # <-- Add this
-            "max_document_size_mb": 5           # <-- Add this
+            "daily_api_calls": 10,
+            "max_document_size_mb": 5
         }
     },
     "standard_tier": {
@@ -81,21 +81,24 @@ SUBSCRIPTION_TIERS = {
             "documents_per_month": 20,
             "video_size_mb": 100,
             "audio_size_mb": 50,
-            "daily_api_calls": 100,             # <-- Add this
-            "max_document_size_mb": 20          # <-- Add this
+            "daily_api_calls": 100,
+            "max_document_size_mb": 20
         }
     },
     "premium_tier": {
         "price": 1499,
         "currency": "INR",
-        "features": ["basic_document_analysis", "basic_risk_assessment", "video_analysis", "audio_analysis", "chatbot", "detailed_risk_assessment", "contract_clause_analysis"],
+        "features": [
+            "basic_document_analysis", "basic_risk_assessment", "video_analysis",
+            "audio_analysis", "chatbot", "detailed_risk_assessment", "contract_clause_analysis"
+        ],
         "limits": {
             "document_size_mb": 50,
             "documents_per_month": 999999,
             "video_size_mb": 500,
             "audio_size_mb": 200,
-            "daily_api_calls": 1000,            # <-- Add this
-            "max_document_size_mb": 50          # <-- Add this
+            "daily_api_calls": 1000,
+            "max_document_size_mb": 50
         }
     }
 }
@@ -530,7 +533,7 @@ def check_subscription_access(user: User, feature: str, file_size_mb: Optional[f
     if user.subscription_tier != "free_tier" and user.subscription_expiry and user.subscription_expiry < datetime.now():
         # Downgrade to free tier if subscription expired
         user.subscription_tier = "free_tier"
-        user.api_calls_remaining = SUBSCRIPTION_TIERS["free_tier"]["daily_api_calls"]
+        user.api_calls_remaining = SUBSCRIPTION_TIERS["free_tier"]["limits"]["daily_api_calls"]
         with get_db_connection() as conn:
             c = conn.cursor()
             c.execute("""
@@ -539,7 +542,7 @@ def check_subscription_access(user: User, feature: str, file_size_mb: Optional[f
                 WHERE id = ?
             """, (user.subscription_tier, user.api_calls_remaining, user.id))
             conn.commit()
-        
+    
     # Reset API calls if needed
     user = reset_api_calls_if_needed(user)
     
@@ -560,7 +563,7 @@ def check_subscription_access(user: User, feature: str, file_size_mb: Optional[f
     
     # Check file size limit if applicable
     if file_size_mb:
-        max_size = SUBSCRIPTION_TIERS[user.subscription_tier]["max_document_size_mb"]
+        max_size = SUBSCRIPTION_TIERS[user.subscription_tier]["limits"]["max_document_size_mb"]
         if file_size_mb > max_size:
             raise HTTPException(
                 status_code=413,
@@ -579,12 +582,12 @@ def check_subscription_access(user: User, feature: str, file_size_mb: Optional[f
         conn.commit()
     
     return True
-
+    
 def reset_api_calls_if_needed(user: User):
     """Reset API call counter if it's a new day"""
     today = datetime.now().date()
     if user.last_reset_date is None or user.last_reset_date.date() < today:
-        tier_limits = SUBSCRIPTION_TIERS[user.subscription_tier]
+        tier_limits = SUBSCRIPTION_TIERS[user.subscription_tier]["limits"]
         user.api_calls_remaining = tier_limits["daily_api_calls"]
         user.last_reset_date = datetime.now()
         # Update the user in the database
@@ -596,7 +599,6 @@ def reset_api_calls_if_needed(user: User):
                 WHERE id = ?
             """, (user.api_calls_remaining, user.last_reset_date, user.id))
             conn.commit()
-            
     return user
 
 def login_user(email, password):