Update app/routes.py

app/routes.py

@@ -30,22 +30,6 @@ def allowed_file(filename):
     """Validate file extension"""
     return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
-@main.route('/')
-def index():
-    return jsonify({
-        "message": "Welcome to the Advanced OCR and Allergy Detection Service",
-        "endpoints": {
-            "/api/ocr": "POST - Image analysis and allergy detection",
-            "/api/allergens": "GET - List of known allergens from dataset"
-        },
-        "supported_formats": list(ALLOWED_EXTENSIONS),
-        "features": [
-            "OCR text extraction",
-            "Allergy detection from database",
-            "AI-powered allergy risk assessment"
-        ]
-    })
-
 @main.route('/api/ocr', methods=['POST'])
 def process_image():
     global allergy_analyzer
@@ -66,6 +50,14 @@ def process_image():
                 "supported_formats": list(ALLOWED_EXTENSIONS)
             }), 400
 
+        # الحصول على حساسيات المستخدم من الطلب
+        user_allergies = request.form.get('user_allergies', '').split(',')
+        user_allergies = [a.strip().lower() for a in user_allergies if a.strip()]
+        
+        if not user_allergies:
+            logger.warning("No user allergies provided")
+            return jsonify({"error": "User allergies not provided"}), 400
+
         # تأكد من تهيئة محلل الحساسيات
         if allergy_analyzer is None:
             init_allergy_analyzer(current_app._get_current_object())
@@ -75,15 +67,17 @@ def process_image():
         file_stream = io.BytesIO(file_bytes)
         image = Image.open(file_stream)
         
-        # تحليل الصورة
+        # تحليل الصورة مع مراعاة حساسيات المستخدم
         analysis_results = allergy_analyzer.analyze_image(
             image,
-            current_app.config['CLAUDE_API_KEY']
+            current_app.config['CLAUDE_API_KEY'],
+            user_allergies=user_allergies
         )
         
         # بناء الاستجابة
         response = {
             "success": True,
+            "user_allergies": user_allergies,
             "extracted_text": analysis_results.get("extracted_text", ""),
             "analysis": {
                 "detected_allergens": analysis_results.get("detected_allergens", []),
@@ -93,7 +87,7 @@ def process_image():
             },
             "warnings": {
                 "has_allergens": len(analysis_results.get("detected_allergens", [])) > 0,
-                "message": "⚠️ Warning: Allergens found!" if analysis_results.get("detected_allergens") else "✅ No allergens found",
+                "message": "⚠️ Warning: Allergens found that match your allergies!" if analysis_results.get("detected_allergens") else "✅ No allergens found that match your allergies",
                 "severity": "high" if analysis_results.get("detected_allergens") else "none"
             }
         }
@@ -106,16 +100,4 @@ def process_image():
         return jsonify({
             "error": "An error occurred while processing the image.",
             "details": str(e)
-        }), 500
-
-@main.route('/api/allergens', methods=['GET'])
-def get_known_allergens():
-    global allergy_analyzer
-    if allergy_analyzer is None:
-        init_allergy_analyzer(current_app._get_current_object())
-    allergens = allergy_analyzer.get_allergen_list()
-    return jsonify({
-        "allergens": allergens,
-        "total_count": len(allergens),
-        "last_updated": "2024-03-24"
-    })
+        }), 500