mashaelalbu commited on
Commit
f726507
·
verified ·
1 Parent(s): e6a5b93

Update app/utils.py

Browse files
Files changed (1) hide show
  1. app/utils.py +15 -7
app/utils.py CHANGED
@@ -170,14 +170,22 @@ class AllergyAnalyzer:
170
  if db_results:
171
  for allergy in db_results:
172
  detected_allergens.add(allergy)
173
- database_matches[allergy] = database_matches.get(allergy, []) + [token]
174
- else:
 
 
175
  # إذا لم توجد في قاعدة البيانات، نستخدم Claude API
176
- if claude_api_key:
177
- for allergy in user_allergens:
178
- if self.check_claude_allergens(token, allergy, claude_api_key, image):
179
- detected_allergens.add(allergy)
180
- claude_matches[allergy] = claude_matches.get(allergy, []) + [token]
 
 
 
 
 
 
181
 
182
  return {
183
  "detected_allergens": list(detected_allergens),
 
170
  if db_results:
171
  for allergy in db_results:
172
  detected_allergens.add(allergy)
173
+ if allergy not in database_matches:
174
+ database_matches[allergy] = []
175
+ database_matches[allergy].append(token)
176
+ elif claude_api_key:
177
  # إذا لم توجد في قاعدة البيانات، نستخدم Claude API
178
+ claude_results = []
179
+ for allergy in user_allergens:
180
+ if self.check_claude_allergens(token, allergy, claude_api_key, image):
181
+ claude_results.append(allergy)
182
+
183
+ if claude_results:
184
+ for allergy in claude_results:
185
+ detected_allergens.add(allergy)
186
+ if allergy not in claude_matches:
187
+ claude_matches[allergy] = []
188
+ claude_matches[allergy].append(token)
189
 
190
  return {
191
  "detected_allergens": list(detected_allergens),