Update app.py

app.py

@@ -32,6 +32,7 @@ VALID_API_KEYS = {
     "irfan_test_key_456": "Irfan (self)",
     "sistem_gelap": "Gelap"
 }
+    
 # --- 1. NEW HELPER FUNCTIONS TO FIX 'TypeError' ---
 def hydrate_history(raw_history_list: list) -> list:
     """Converts a list of dicts from session back into LangChain Message objects."""
@@ -380,11 +381,30 @@ def reset_metrics(domain):
         return jsonify({"error": str(e)}), 500
         
 # Helper function to check API key
+API_USAGE = {}
+
 def check_api_key(request_data):
     api_key = request_data.get("api_key")
+    
+    # 1. Check if key exists
     if not api_key or api_key not in VALID_API_KEYS:
-        return False, {"error": "Invalid or missing API key"}, 401
-    logger.info(f"API request authenticated for user: {VALID_API_KEYS[api_key]}")
+        return False, {"error": "Invalid API key"}, 401
+    
+    # 2. Initialize counter for this key if new
+    if api_key not in API_USAGE:
+        API_USAGE[api_key] = 0
+        
+    # 3. Check Quota (The "Selling" Logic)
+    # Example: "public_demo_key_789" only gets 5 tries
+    if api_key in ["anak_mail_123", "sistem_gelap"] and API_USAGE[api_key] >= 5:
+    return False, {
+        "error": "Quota exceeded! Please buy the Premium Plan."
+    }, 429
+        
+    # 4. Increment Counter
+    API_USAGE[api_key] += 1
+    logger.info(f"User {VALID_API_KEYS[api_key]} used {API_USAGE[api_key]} requests.")
+    
     return True, None, None
 
 # Helper function to save and process uploaded files (Base64)