feat: add user setting to disable AI topic suggestions

- Added use_ai_suggestions column to users table with migration.
- Added toggle in settings page to enable/disable AI suggestions.
- Updated manual classification wizard to respect the setting (hides AI buttons and disables automated fetches).
- Updated keyboard shortcuts to skip AI calls when disabled.

database.py

@@ -395,11 +395,17 @@ def setup_database():
         cursor.execute("SELECT group_name FROM sessions LIMIT 1")
     except sqlite3.OperationalError:
         cursor.execute("ALTER TABLE sessions ADD COLUMN group_name TEXT")
+try:
+    cursor.execute("SELECT pdf_viewer FROM users LIMIT 1")
+except sqlite3.OperationalError:
+    cursor.execute("ALTER TABLE users ADD COLUMN pdf_viewer TEXT DEFAULT 'native'")
 
-    try:
-        cursor.execute("SELECT pdf_viewer FROM users LIMIT 1")
-    except sqlite3.OperationalError:
-        cursor.execute("ALTER TABLE users ADD COLUMN pdf_viewer TEXT DEFAULT 'adobe'")
+try:
+    cursor.execute("SELECT use_ai_suggestions FROM users LIMIT 1")
+except sqlite3.OperationalError:
+    cursor.execute("ALTER TABLE users ADD COLUMN use_ai_suggestions INTEGER DEFAULT 1")
+
+conn.commit()
 
     # Migration to normalize subject names (e.g., "CHEMISTRY" to "Chemistry", "PHYSICS" to "Physics", etc.)
     try:

settings_routes.py

@@ -35,6 +35,9 @@ def settings():
         # --- Handle Two-Page Crop Toggle ---
         two_page_crop = 1 if request.form.get('two_page_crop') else 0
 
+        # --- Handle AI Suggestions Toggle ---
+        use_ai_suggestions = 1 if request.form.get('use_ai_suggestions') else 0
+
         # --- Handle Classifier Model Setting ---
         classifier_model = request.form.get('classifier_model', 'gemini')
         if classifier_model not in ['gemini', 'nova', 'gemma']:
@@ -75,8 +78,8 @@ def settings():
 
         # --- Update Database ---
         conn = get_db_connection()
-        conn.execute('UPDATE users SET neetprep_enabled = ?, v2_default = ?, magnifier_enabled = ?, two_page_crop = ?, dpi = ?, color_rm_dpi = ?, classifier_model = ?, pdf_viewer = ? WHERE id = ?',
-                     (neetprep_enabled, v2_default, magnifier_enabled, two_page_crop, dpi, color_rm_dpi, classifier_model, pdf_viewer, current_user.id))
+        conn.execute('UPDATE users SET neetprep_enabled = ?, v2_default = ?, magnifier_enabled = ?, two_page_crop = ?, use_ai_suggestions = ?, dpi = ?, color_rm_dpi = ?, classifier_model = ?, pdf_viewer = ? WHERE id = ?',
+                     (neetprep_enabled, v2_default, magnifier_enabled, two_page_crop, use_ai_suggestions, dpi, color_rm_dpi, classifier_model, pdf_viewer, current_user.id))
         conn.commit()
         conn.close()
 
@@ -85,6 +88,7 @@ def settings():
         current_user.v2_default = v2_default
         current_user.magnifier_enabled = magnifier_enabled
         current_user.two_page_crop = two_page_crop
+        current_user.use_ai_suggestions = use_ai_suggestions
         current_user.dpi = dpi
         current_user.color_rm_dpi = color_rm_dpi
         current_user.classifier_model = classifier_model

templates/question_entry_v2.html

@@ -540,20 +540,24 @@
                 <div class="mb-2">
                     <label class="form-label small text-muted mb-2">Chapter / Topic</label>
                     <div class="input-group input-group-lg">
-                        <input type="text" id="topic_input" class="form-control bg-dark text-white border-secondary" placeholder="Start typing or use AI...">
+                        <input type="text" id="topic_input" class="form-control bg-dark text-white border-secondary" placeholder="Start typing...">
                         <button class="btn btn-outline-secondary" type="button" onclick="document.getElementById('topic_input').value=''; document.getElementById('topic_input').focus();">
                             <i class="bi bi-x-lg"></i>
                         </button>
+                        {% if current_user.use_ai_suggestions %}
                         <button class="btn btn-info" type="button" id="btn_get_suggestion" onclick="getAiSuggestion()">
                             <span class="spinner-border spinner-border-sm d-none" role="status"></span>
                             <i class="bi bi-stars"></i>
                         </button>
+                        {% endif %}
                     </div>
                 </div>
 
                 <!-- Suggestions -->
                 <div id="ai_suggestion_container" class="d-none">
+                    {% if current_user.use_ai_suggestions %}
                     <label class="form-label d-block small text-info mb-2 mt-1"><i class="bi bi-robot me-1"></i>AI Suggested:</label>
+                    {% endif %}
                     <div id="ai_suggestion_chips" class="d-flex flex-wrap gap-2"></div>
                 </div>
                 <div id="ai_error_msg" class="alert alert-danger d-none mt-2 py-2"></div>
@@ -1903,8 +1907,8 @@
             input.select();
         }, 100);
 
-        // Auto-load suggestions if topic is empty
-        if (!document.getElementById('topic_input').value) {
+        // Auto-load suggestions if topic is empty and setting enabled
+        if (!document.getElementById('topic_input').value && {% if current_user.use_ai_suggestions %}true{% else %}false{% endif %}) {
             getAiSuggestion();
         }
     }
@@ -2138,14 +2142,18 @@
                     // Clear and fetch fresh suggestions for the new subject
                     input.value = '';
                     input.focus();
-                    getAiSuggestion();
+                    if ({% if current_user.use_ai_suggestions %}true{% else %}false{% endif %}) {
+                        getAiSuggestion();
+                    }
                 }
             }
 
             // Alt + S to trigger AI suggestion
             if (e.altKey && (e.key === 's' || e.key === 'S')) {
-                e.preventDefault();
-                getAiSuggestion();
+                if ({% if current_user.use_ai_suggestions %}true{% else %}false{% endif %}) {
+                    e.preventDefault();
+                    getAiSuggestion();
+                }
             }
         });
     });

templates/settings.html

@@ -98,6 +98,15 @@
                                     Choose the AI model for automatic question classification. Gemini uses Google's Gemini API, while Nova uses Amazon's Nova model via OpenRouter. Requires relevant API keys to be set.
                                 </div>
                             </div>
+                            <div class="mb-3 form-check form-switch">
+                                <input type="checkbox" class="form-check-input" id="use_ai_suggestions" name="use_ai_suggestions" {% if current_user.use_ai_suggestions %}checked{% endif %}>
+                                <label class="form-check-label" for="use_ai_suggestions">
+                                    Enable AI Topic Suggestions
+                                </label>
+                                <div class="form-text">
+                                    Show AI-suggested topics and chapters in the manual classification wizard. If disabled, you can still type manually.
+                                </div>
+                            </div>
                         </fieldset>
                         <hr>
                         <fieldset>

user_auth.py

@@ -5,7 +5,7 @@ from utils import get_db_connection
 
 class User(UserMixin):
     """User model for Flask-Login."""
-    def __init__(self, id, username, email, password_hash, neetprep_enabled, dpi, color_rm_dpi, v2_default=0, magnifier_enabled=1, two_page_crop=0, google_token=None, classifier_model='gemini', pdf_viewer='adobe'):
+    def __init__(self, id, username, email, password_hash, neetprep_enabled, dpi, color_rm_dpi, v2_default=0, magnifier_enabled=1, two_page_crop=0, google_token=None, classifier_model='gemini', pdf_viewer='adobe', use_ai_suggestions=1):
         self.id = id
         self.username = username
         self.email = email
@@ -19,6 +19,7 @@ class User(UserMixin):
         self.google_token = google_token
         self.classifier_model = classifier_model
         self.pdf_viewer = pdf_viewer  # 'adobe', 'browser', or 'legacy'
+        self.use_ai_suggestions = use_ai_suggestions
 
     @staticmethod
     def get(user_id):
@@ -40,7 +41,8 @@ class User(UserMixin):
                 user_data.get('two_page_crop', 0),
                 user_data.get('google_token'),
                 user_data.get('classifier_model', 'gemini'),
-                user_data.get('pdf_viewer', 'adobe')
+                user_data.get('pdf_viewer', 'adobe'),
+                user_data.get('use_ai_suggestions', 1)
             )
         return None
 
@@ -64,7 +66,8 @@ class User(UserMixin):
                 user_data.get('two_page_crop', 0),
                 user_data.get('google_token'),
                 user_data.get('classifier_model', 'gemini'),
-                user_data.get('pdf_viewer', 'adobe')
+                user_data.get('pdf_viewer', 'adobe'),
+                user_data.get('use_ai_suggestions', 1)
             )
         return None