ol

app.py

@@ -284,7 +284,8 @@ def classify_document():
         D, I = index.search(np.array([embedding]), k=k)
         
         if len(labels) > 0 and I[0][0] < len(labels):
-            similarity = 1 - D[0][0]
+            # Convert numpy float32 to Python float for JSON serialization
+            similarity = float(1 - D[0][0])
             confidence_threshold = 0.35
             
             best_match = labels[I[0][0]]
@@ -292,7 +293,8 @@ def classify_document():
             
             for i in range(min(k, len(D[0]))):
                 if I[0][i] < len(labels):
-                    sim = 1 - D[0][i]
+                    # Convert numpy float32 to Python float
+                    sim = float(1 - D[0][i])
                     matches.append({"category": labels[I[0][i]], "similarity": round(sim, 3)})
             
             # Save classified document to SQLite
@@ -332,6 +334,7 @@ def classify_document():
         
         return jsonify({"error": "Document not recognized"}), 400
     except Exception as e:
+        print(f"Classification error: {e}")
         return jsonify({"error": str(e)}), 500
 
 @app.route("/api/categories", methods=["GET"])

static/script.js

@@ -592,11 +592,22 @@ document.getElementById('classifyForm').addEventListener('submit', async (e) =>
             fileInput.value = '';
             document.querySelector('#classifyUpload p').textContent = 'Click to select or drag & drop files here';
             loadStats();
+            loadCategories();
         } else {
-            showResult(resultDiv, result.detail, 'error');
+            // Handle different error types
+            let errorMessage = result.error || result.detail || 'Classification failed';
+            if (errorMessage.includes('No categories')) {
+                errorMessage = '⚠️ Please add some document categories first before classifying documents.';
+            } else if (errorMessage.includes('Failed to process')) {
+                errorMessage = '❌ Could not process the uploaded file. Please ensure it\'s a valid image or PDF.';
+            } else if (errorMessage.includes('JSON serializable')) {
+                errorMessage = '🔧 Processing error occurred. Please try again.';
+            }
+            showResult(resultDiv, errorMessage, 'error');
         }
     } catch (error) {
-        showResult(resultDiv, 'Classification failed: ' + error.message, 'error');
+        console.error('Classification error:', error);
+        showResult(resultDiv, '❌ Network error: Please check your connection and try again.', 'error');
     }
 });