fix: resolve NameError in processing.py and restore missing question routes

- Fixed NameError 'p_info' in process_crop_v2 by using correct variable name.
- Restored '/get_topic_suggestions' and '/classified/update_single' routes in questions.py that were lost during modularization.

routes/processing.py

@@ -100,7 +100,7 @@ def process_crop_v2():
         max_idx = conn.execute('SELECT MAX(image_index) FROM images WHERE session_id = ?', (session_id,)).fetchone()[0]
         next_idx = (max_idx if max_idx is not None else -1) + 1
         for i, p_box in enumerate(processed_boxes):
-            conn.execute('INSERT INTO images (session_id, image_index, filename, original_name, processed_filename, image_type, box_id) VALUES (?, ?, ?, ?, ?, ?, ?)', (session_id, next_idx + i, p_info['filename'], p_box['original_filename'], p_box['processed_filename'], 'cropped', p_box['box_id']))
+            conn.execute('INSERT INTO images (session_id, image_index, filename, original_name, processed_filename, image_type, box_id) VALUES (?, ?, ?, ?, ?, ?, ?)', (session_id, next_idx + i, p_box['original_filename'], f"Page {page_index + 1} - Q{i + 1}", p_box['processed_filename'], 'cropped', p_box['box_id']))
             img_id = conn.execute('SELECT last_insert_rowid()').fetchone()[0]
             conn.execute("INSERT INTO questions (session_id, image_id, question_number, status, marked_solution, actual_solution) VALUES (?, ?, ?, ?, ?, ?)", (session_id, img_id, p_box.get('question_number'), p_box.get('status', 'unattempted'), p_box.get('marked_solution'), p_box.get('actual_solution')))
         conn.commit(); os.remove(temp_path); return jsonify({'success': True, 'processed_count': len(processed_boxes)})

routes/questions.py

@@ -3,10 +3,77 @@ from flask import current_app, request, jsonify, render_template, flash, redirec
 from .common import main_bp, get_db_connection, login_required, current_user
 from processing import resize_image_if_needed, call_nim_ocr_api, extract_question_number_from_ocr_result
 from strings import ROUTE_SAVE_QUESTIONS, ROUTE_EXTRACT_QUESTION_NUMBER, ROUTE_EXTRACT_ALL_QUESTION_NUMBERS, METHOD_POST
+import requests
+import json
+from nvidia_prompts import BIOLOGY_PROMPT_TEMPLATE, CHEMISTRY_PROMPT_TEMPLATE, PHYSICS_PROMPT_TEMPLATE, MATHEMATICS_PROMPT_TEMPLATE
 
 NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY")
 NVIDIA_NIM_AVAILABLE = bool(NVIDIA_API_KEY)
 
+def get_nvidia_prompt(subject, input_questions):
+    if subject.lower() == 'biology': return BIOLOGY_PROMPT_TEMPLATE.format(input_questions=input_questions)
+    if subject.lower() == 'chemistry': return CHEMISTRY_PROMPT_TEMPLATE.format(input_questions=input_questions)
+    if subject.lower() == 'physics': return PHYSICS_PROMPT_TEMPLATE.format(input_questions=input_questions)
+    if subject.lower() == 'mathematics': return MATHEMATICS_PROMPT_TEMPLATE.format(input_questions=input_questions)
+    return None
+
+@main_bp.route('/get_topic_suggestions', methods=['POST'])
+@login_required
+def get_topic_suggestions():
+    data = request.json
+    question_text, image_id, subject = data.get('question_text'), data.get('image_id'), data.get('subject')
+    if not subject: return jsonify({'error': 'Subject is required'}), 400
+    if not question_text and image_id:
+        try:
+            conn = get_db_connection()
+            row = conn.execute('SELECT question_text, processed_filename, i.session_id FROM questions q JOIN images i ON q.image_id = i.id WHERE i.id = ?', (image_id,)).fetchone()
+            if row:
+                if row['question_text']: question_text = row['question_text']
+                else:
+                    image_path = os.path.join(current_app.config['PROCESSED_FOLDER'], row['processed_filename'])
+                    if os.path.exists(image_path):
+                        question_text = " ".join(item['text_prediction']['text'] for item in call_nim_ocr_api(resize_image_if_needed(image_path))['data'][0]['text_detections'])
+                        conn.execute('UPDATE questions SET question_text = ? WHERE image_id = ?', (question_text, image_id))
+                        conn.commit()
+            conn.close()
+        except Exception as e: return jsonify({'error': f"OCR failed: {str(e)}"}), 500
+    if not question_text: return jsonify({'error': 'Could not obtain question text.'}), 400
+    prompt_content = get_nvidia_prompt(subject, f"1. {question_text}")
+    if not prompt_content: return jsonify({'error': f'Unsupported subject: {subject}'}), 400
+    if not NVIDIA_API_KEY: return jsonify({'error': 'NVIDIA_API_KEY not set'}), 500
+    try:
+        res = requests.post('https://integrate.api.nvidia.com/v1/chat/completions', headers={'Authorization': f'Bearer {NVIDIA_API_KEY}', 'Accept': 'application/json', 'Content-Type': 'application/json'}, json={"model": "nvidia/nemotron-3-nano-30b-a3b", "messages": [{"content": prompt_content, "role": "user"}], "temperature": 0.2, "top_p": 1, "max_tokens": 1024, "stream": False}, timeout=30)
+        res.raise_for_status()
+        content = res.json()['choices'][0]['message']['content']
+        if "```json" in content: content = content.split("```json")[1].split("```")[0].strip()
+        elif "```" in content: content = content.split("```")[1].split("```")[0].strip()
+        data = json.loads(content)
+        suggestions = []
+        if data.get('data'):
+            suggestions.append(data['data'][0].get('chapter_title', 'Unclassified'))
+            if 'other_possible_chapters' in data['data'][0]:
+                others = data['data'][0]['other_possible_chapters']
+                if isinstance(others, list): suggestions.extend(others)
+        return jsonify({'success': True, 'suggestions': suggestions, 'full_response': data})
+    except Exception as e: return jsonify({'error': str(e)}), 500
+
+@main_bp.route('/classified/update_single', methods=['POST'])
+@login_required
+def update_question_classification_single():
+    data = request.json
+    image_id, subject, chapter = data.get('image_id'), data.get('subject'), data.get('chapter')
+    if not image_id: return jsonify({'error': 'Image ID is required'}), 400
+    try:
+        conn = get_db_connection()
+        owner = conn.execute("SELECT s.user_id FROM images i JOIN sessions s ON i.session_id = s.id WHERE i.id = ?", (image_id,)).fetchone()
+        if not owner or owner['user_id'] != current_user.id:
+            conn.close(); return jsonify({'error': 'Unauthorized'}), 403
+        conn.execute('UPDATE questions SET subject = ?, chapter = ? WHERE image_id = ?', (subject, chapter, image_id))
+        conn.commit(); conn.close()
+        return jsonify({'success': True})
+    except Exception as e: return jsonify({'error': str(e)}), 500
+
+
 @main_bp.route('/question_entry_v2/<session_id>')
 @login_required
 def question_entry_v2(session_id):