Update app.py

app.py

@@ -20,6 +20,7 @@ DATA_DIR.mkdir(exist_ok=True)
 DB_PATH = DATA_DIR / "usage_limits.db"
 
 DAILY_LIMIT = 50
+EXEMPTED_USERS = ["multimodalart"]
 db_lock = Lock()
 
 def init_db():
@@ -46,6 +47,11 @@ def check_and_update_usage(username: str) -> bool:
     Check if user has reached daily limit and update usage.
     Returns True if user can generate, False if limit reached.
     """
+    # Exempted users bypass all checks
+    if username in EXEMPTED_USERS:
+        print(f"User {username} is exempted from rate limits")
+        return True
+    
     with db_lock:
         try:
             with sqlite3.connect(DB_PATH) as conn:
@@ -96,6 +102,10 @@ def check_and_update_usage(username: str) -> bool:
 
 def get_remaining_generations(username: str) -> int:
     """Get the number of remaining generations for today."""
+    # Exempted users have unlimited generations
+    if username in EXEMPTED_USERS:
+        return 999999  # Return a large number to indicate unlimited
+    
     with db_lock:
         try:
             with sqlite3.connect(DB_PATH) as conn: