Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import cv2
 import numpy as np
+import base64
 from flask import Flask, request, render_template, jsonify
 from werkzeug.utils import secure_filename
 
@@ -128,5 +129,35 @@ def predict():
                 os.remove(filepath)
             return jsonify({'error': str(e)}), 500
 
+@app.route('/test-example', methods=['POST'])
+def test_example():
+    """Test with example image"""
+    try:
+        example_path = 'image/20251012_13h56m44s_grim.png'
+        
+        if not os.path.exists(example_path):
+            return jsonify({'error': 'Example image not found. Please add 20251012_13h56m44s_grim.png to image/ folder'}), 404
+        
+        print(f"Testing with example file: {example_path}")
+        
+        # Make prediction
+        result = predict_image(example_path)
+        
+        # Read and encode image as base64
+        with open(example_path, 'rb') as f:
+            img_data = f.read()
+        img_base64 = base64.b64encode(img_data).decode('utf-8')
+        result['image'] = f"data:image/png;base64,{img_base64}"
+        
+        print(f"✓ Example prediction completed: {result['class']}")
+        
+        return jsonify(result)
+        
+    except Exception as e:
+        print(f"Error during example prediction: {e}")
+        import traceback
+        traceback.print_exc()
+        return jsonify({'error': str(e)}), 500
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=False)