Update app.py

app.py

@@ -62,37 +62,43 @@ def compare_skills(skill_1, skill_2):
     
     return score, common_skill
 
-def extract_ner_info(text, nlp):
+def extract_ner_info(text, nlp, fixed_loc):
     ner_results = nlp(text)
     full_name = ''
     loc = ''
     age = None
-    # جمع‌آوری قطعات B-pers و I-pers برای تشکیل نام کامل
+    # جمع‌آوری قطعات B-pers و I-pers برای تشکیل نام کامل با شرط کاهش ۱۰ درصدی
     i = 0
     while i < len(ner_results):
-        if ner_results[i]['entity'] == 'B-pers':
+        if ner_results[i]['entity'] == 'B-pers' and ner_results[i]['score'] >= 0.80:
             if full_name:  # اگر قبلاً نامی پیدا کرده‌ایم، فاصله اضافه کن
                 full_name += ' '
             full_name += ner_results[i]['word']
+            current_score = ner_results[i]['score']
+            stop_adding = False  # پرچم برای توقف بعد از اولین نام کامل
             for j in range(i + 1, len(ner_results)):
-                if ner_results[j]['entity'].startswith('I-pers'):
-                    full_name += ner_results[j]['word'].replace('##', '')
-                    i = j  # به موقعیت جدید برو
+                if ner_results[j]['entity'] == 'I-pers' and ner_results[j]['score'] >= 0.80:
+                    if ner_results[j]['score'] >= current_score * 0.90:  # شرط کاهش ۱۰ درصدی
+                        full_name += ner_results[j]['word'].replace('##', '')
+                        current_score = ner_results[j]['score']
+                        i = j  # به موقعیت جدید برو
+                    else:
+                        stop_adding = True  # توقف بعد از اولین نام کامل
+                        break
                 else:
+                    stop_adding = True  # توقف بعد از اولین نام کامل
                     break
+            if stop_adding:
+                break  # توقف بعد از اولین نام کامل
         i += 1
     # جمع‌آوری مکان‌ها با امتیاز بالای ۸۰
     for entity in ner_results:
-        if entity['entity'] == 'B-loc' and entity['score'] >= 0.80:
+        if entity['entity'] in ['B-loc', 'I-loc'] and entity['score'] >= 0.80:
             if loc:
                 loc += ' '
             loc += entity['word']
     # استخراج سن از متن
     age_match = re.search(r'سن\s*:\s*(\d+)', text)
-    if age_match:
-        age = int(age_match.group(1))
-    # استخراج سن از متن
-    age_match = re.search(r'سن\s*:\s*(\d+)', text)
     if age_match:
         age = int(age_match.group(1))
     # بررسی امتیاز شباهت مکان