Update app/routes.py

app/routes.py

@@ -1,4 +1,4 @@
-# routes.py
+# routes.py (معدل)
 from flask import Blueprint, jsonify, request, current_app
 import io
 import pandas as pd
@@ -66,14 +66,14 @@ def process_image():
                 "supported_formats": list(ALLOWED_EXTENSIONS)
             }), 400
 
-        # القائمة الثابتة للحساسيات (بدلاً من إدخال المستخدم)
-        FIXED_ALLERGENS = [
-            "dairy", "eggs", "peanuts", "soy", 
-            "tree nuts", "wheat", "fish", 
-            "shellfish", "sesame"
-        ]
+        # الحصول على قائمة الحساسيات من المستخدم أو استخدام القائمة الثابتة
+        user_allergens = request.form.get('allergens')
+        if user_allergens:
+            user_allergens = [a.strip().lower() for a in user_allergens.split(',')]
+        else:
+            user_allergens = list(allergy_analyzer.allergy_dict.keys())
         
-        logger.info(f"Processing image for fixed allergens: {FIXED_ALLERGENS}")
+        logger.info(f"Processing image for allergens: {user_allergens}")
         
         # تأكد من تهيئة محلل الحساسيات
         if allergy_analyzer is None:
@@ -84,16 +84,27 @@ def process_image():
         file_stream = io.BytesIO(file_bytes)
         image = Image.open(file_stream)
         
-        # تحليل الصورة باستخدام القائمة الثابتة
-        analysis_results = allergy_analyzer.analyze_image(image, FIXED_ALLERGENS)
+        # تحليل الصورة
+        analysis_results = allergy_analyzer.analyze_image(
+            image, 
+            user_allergens,
+            current_app.config['CLAUDE_API_KEY']
+        )
         
+        # بناء الاستجابة
         response = {
             "success": True,
-            "analysis": analysis_results,
+            "extracted_text": analysis_results.get("extracted_text", ""),
+            "analysis": {
+                "detected_allergens": analysis_results.get("detected_allergens", []),
+                "database_matches": analysis_results.get("database_matches", {}),
+                "claude_matches": analysis_results.get("claude_matches", {}),
+                "analyzed_tokens": analysis_results.get("analyzed_tokens", [])
+            },
             "warnings": {
-                "has_allergens": len(analysis_results['detected_allergens']) > 0,
-                "message": "⚠️ Warning: Allergens found!" if analysis_results['detected_allergens'] else "✅ No allergens found",
-                "severity": "high" if analysis_results['detected_allergens'] else "none"
+                "has_allergens": len(analysis_results.get("detected_allergens", [])) > 0,
+                "message": "⚠️ Warning: Allergens found!" if analysis_results.get("detected_allergens") else "✅ No allergens found",
+                "severity": "high" if analysis_results.get("detected_allergens") else "none"
             }
         }