Update app.py

app.py

@@ -7,11 +7,12 @@ import sys
 import io
 from ultralytics import YOLO
 import time
+import easyocr
 
 
 # Set the default encoding to utf-8
 sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
-
+reader = easyocr.Reader(['en'])  
 app = Flask(__name__)
 
 # Load your pre-trained vegetable classification model
@@ -88,5 +89,32 @@ def detect():
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
+@app.route('/ocr', methods=['POST'])
+def ocr():
+    try:
+        # Extract and decode image data
+        data = request.json
+        image_data = data['image'].split(',')[1]  # Remove data:image/jpeg;base64,
+
+        # Decode the base64 string into a NumPy array
+        nparr = np.frombuffer(base64.b64decode(image_data), np.uint8)
+
+        # Convert the NumPy array into an OpenCV image (BGR format)
+        image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
+
+        # Convert the image to RGB (as easyocr expects RGB format)
+        image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+
+        # Perform OCR using easyocr
+        results = reader.readtext(image_rgb)
+
+        # Check if any text was detected
+        text_detected = bool(results)  # True if text is detected, False otherwise
+
+        return jsonify({'text_detected': text_detected})
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)