Update app.py

app.py

@@ -21,7 +21,7 @@ DB_PATH = DATA_DIR / "usage_limits.db"
 
 DAILY_LIMIT_STANDARD = 75
 DAILY_LIMIT_PRO = 50
-EXEMPTED_USERS = ["multimodalart"]
+EXEMPTED_USERS = []
 db_lock = Lock()
 
 def init_db():
@@ -29,16 +29,53 @@ def init_db():
     print(f"Initializing database at: {DB_PATH}")
     try:
         with sqlite3.connect(DB_PATH) as conn:
-            conn.execute('''
-                CREATE TABLE IF NOT EXISTS usage (
-                    username TEXT PRIMARY KEY,
-                    date TEXT NOT NULL,
-                    count_standard INTEGER NOT NULL DEFAULT 0,
-                    count_pro INTEGER NOT NULL DEFAULT 0
-                )
-            ''')
-            conn.commit()
-        print("Database initialized successfully")
+            cursor = conn.cursor()
+            
+            # Check if table exists and what columns it has
+            cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='usage'")
+            table_exists = cursor.fetchone()
+            
+            if table_exists:
+                # Check current schema
+                cursor.execute("PRAGMA table_info(usage)")
+                columns = [col[1] for col in cursor.fetchall()]
+                
+                # Migrate if old schema (only has 'count' column)
+                if 'count' in columns and 'count_standard' not in columns:
+                    print("Migrating database from old schema to new schema...")
+                    # Rename old count to count_standard, add count_pro
+                    cursor.execute("ALTER TABLE usage RENAME COLUMN count TO count_standard")
+                    cursor.execute("ALTER TABLE usage ADD COLUMN count_pro INTEGER NOT NULL DEFAULT 0")
+                    conn.commit()
+                    print("Database migration completed successfully")
+                elif 'count_standard' not in columns:
+                    # Table exists but doesn't have the right columns - recreate it
+                    print("Recreating table with new schema...")
+                    cursor.execute("DROP TABLE usage")
+                    cursor.execute('''
+                        CREATE TABLE usage (
+                            username TEXT PRIMARY KEY,
+                            date TEXT NOT NULL,
+                            count_standard INTEGER NOT NULL DEFAULT 0,
+                            count_pro INTEGER NOT NULL DEFAULT 0
+                        )
+                    ''')
+                    conn.commit()
+                    print("Database recreated successfully")
+                else:
+                    print("Database schema is already up to date")
+            else:
+                # Create new table with updated schema
+                cursor.execute('''
+                    CREATE TABLE IF NOT EXISTS usage (
+                        username TEXT PRIMARY KEY,
+                        date TEXT NOT NULL,
+                        count_standard INTEGER NOT NULL DEFAULT 0,
+                        count_pro INTEGER NOT NULL DEFAULT 0
+                    )
+                ''')
+                conn.commit()
+                print("Database initialized successfully")
     except Exception as e:
         print(f"Error initializing database: {e}")
         import traceback
@@ -457,6 +494,8 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
                 model_radio = gr.Radio(
                     choices=["Nano Banana", "Nano Banana PRO"],
                     value="Nano Banana PRO",
+                    label="Model Selection",
+                    info="PRO: Higher quality, slower generation (50 credits/day) | Standard: Faster generation (75/day)"
                 )
                 
                 with gr.Row():
@@ -468,13 +507,14 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
                     )
                     
                     resolution_dropdown = gr.Dropdown(
-                        label="Resolution",
+                        label="Resolution (PRO only)",
                         choices=["1K", "2K", "4K"],
                         value="1K",
                         interactive=True,
+                        visible=True
                     )
                     
-                    
+                gr.Markdown("💡 **Credit costs:** 1K = 1 credit, 2K = 2 credits, 4K = 4 credits")
                 
                 generate_button = gr.Button("Generate", variant="primary")
             with gr.Column(scale=1):