Update app/routes.py

app/routes.py

@@ -1,65 +1,117 @@
-from flask import Blueprint, jsonify, request, current_app
-import os
-from werkzeug.utils import secure_filename
-from app.utils import OCRModel
-import torch
+from flask import Blueprint, jsonify, request
 import io
+from app.utils import OCRModel
 
 main = Blueprint('main', __name__)
 ocr_model = OCRModel()
 
+# تحديد امتدادات الملفات المسموح بها
 ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}
 
+# قائمة الحساسيات المعروفة (يمكن تخصيصها حسب الحاجة)
+KNOWN_ALLERGENS = {
+    'gluten': ['wheat', 'barley', 'gluten'],
+    'dairy': ['milk', 'yogurt', 'cheese', 'lactose'],
+    'nuts': ['nuts', 'peanuts', 'almonds', 'walnuts'],
+    'eggs': ['eggs'],
+    'soy': ['soy'],
+    'fish': ['fish'],
+    'shellfish': ['oyster', 'shrimp', 'shrimp'],
+}
+
 def allowed_file(filename):
+    """التحقق من صحة امتداد الملف"""
     return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
+def find_allergens(text, user_allergens):
+    """البحث عن الحساسيات في النص"""
+    text = text.lower()
+    found_allergens = set()
+    allergen_details = {}
+
+    for allergen in user_allergens:
+        allergen = allergen.strip().lower()
+        # البحث في القائمة الرئيسية للحساسيات
+        if allergen in KNOWN_ALLERGENS:
+            for variant in KNOWN_ALLERGENS[allergen]:
+                if variant.lower() in text:
+                    found_allergens.add(allergen)
+                    allergen_details[allergen] = variant
+        # البحث المباشر عن النص المدخل
+        elif allergen in text:
+            found_allergens.add(allergen)
+            allergen_details[allergen] = allergen
+
+    return found_allergens, allergen_details
+
 @main.route('/')
 def index():
-    return jsonify({"message": "Welcome to OCR API!"})
+    return jsonify({
+        "message": "Welcome to the Text Recognition and Sensitivity Checking Service",
+        "endpoints": {
+            "/api/ocr": "POST - Image analysis and sensitivity testing",
+        },
+        "supported_formats": list(ALLOWED_EXTENSIONS),
+        "known_allergens": list(KNOWN_ALLERGENS.keys())
+    })
 
 @main.route('/api/ocr', methods=['POST'])
 def process_image():
+    # التحقق من وجود الملف
     if 'file' not in request.files:
-        return jsonify({"error": "No file part"}), 400
+        return jsonify({"error": "No file uploaded"}), 400
     
+    # التحقق من وجود قائمة الحساسيات
     if 'allergens' not in request.form:
-        return jsonify({"error": "No allergens specified"}), 400
+        return jsonify({"error": "Sensitivities not specified"}), 400
     
     file = request.files['file']
     if file.filename == '':
-        return jsonify({"error": "No selected file"}), 400
+        return jsonify({"error": "No file selected"}), 400
     
-    # تحويل الحساسيات إلى قائمة
-    user_allergens = request.form['allergens'].lower().split(',')
-    user_allergens = {allergen.strip() for allergen in user_allergens}
+    # التحقق من نوع الملف
+    if not allowed_file(file.filename):
+        return jsonify({
+            "error": "File type not supported",
+            "supported_formats": list(ALLOWED_EXTENSIONS)
+        }), 400
+
+    # تحضير قائمة الحساسيات
+    user_allergens = request.form['allergens'].split(',')
     
-    if file and allowed_file(file.filename):
-        # قراءة الصورة في الذاكرة بدلاً من حفظها
+    try:
+        # قراءة الصورة
         file_bytes = file.read()
         file_stream = io.BytesIO(file_bytes)
         
-        try:
-            # معالجة الصورة واستخراج النص
-            result = ocr_model.process_image(file_stream)
-            
-            # تحليل النص وفحص الحساسيات
-            found_allergens = set()
-            text_lower = result.lower()
-            
-            for allergen in user_allergens:
-                if allergen in text_lower:
-                    found_allergens.add(allergen)
-            
-            response = {
-                "text": result,
+        # معالجة الصورة
+        extracted_text = ocr_model.process_image(file_stream)
+        
+        # البحث عن الحساسيات
+        found_allergens, allergen_details = find_allergens(extracted_text, user_allergens)
+        
+        # تحضير الرد
+        response = {
+            "success": True,
+            "extracted_text": extracted_text,
+            "analysis": {
                 "found_allergens": list(found_allergens),
+                "allergen_details": allergen_details,
                 "has_allergens": len(found_allergens) > 0,
-                "warning": "تحذير: هذا المنتج يحتوي على مواد قد تسبب لك حساسية!" if found_allergens else "لم يتم العثور على مواد مسببة لل��ساسية"
+                "warning": "Warning: Allergens found!" if found_allergens else "No allergens found"
             }
-            
-            return jsonify(response)
-            
-        except Exception as e:
-            return jsonify({"error": str(e)}), 500
-    
-    return jsonify({"error": "Invalid file type"}), 400
+        }
+        
+        return jsonify(response)
+        
+    except Exception as e:
+        return jsonify({
+            "error": "An error occurred while processing the image.",
+            "details": str(e)
+        }), 500
+
+@main.route('/api/allergens', methods=['GET'])
+def get_known_allergens():
+    return jsonify({
+        "allergens": KNOWN_ALLERGENS
+    })