Update helpers.py

helpers.py

@@ -381,31 +381,99 @@ def define_metadata(input_text):
                   'Định phí bảo hiểm và Quản trị rủi ro',
                   'Công nghệ thông tin',
                   'An toàn thông tin']
-    #cond1 cond2 la str, con3 la list ten cac nganh
-    result = {}
+        result = {}
+    # Xác định loại tài liệu
     if condition3 in input_text:
         result['Tai lieu ve'] = 'Đề cương'
     elif condition1 in input_text:
         result['Tai lieu ve'] = 'Chương trình đào tạo'
     elif condition2 in input_text:
         result['Tai lieu ve'] = 'Đề án'
-    for cond in condition4:
-        if cond in input_text:
+
+    # Nếu tài liệu là "Đề án", thêm tất cả các ngành và khoa theo mapping dưới đây.
+    if result.get('Tai lieu ve') == 'Đề án':
+        all_nganh = []
+        all_khoa = []
+        for cond in condition4:
+            # Xác định khoa dựa trên tên ngành
             if cond in ['An toàn thông tin', 'Công nghệ thông tin', 'Khoa học máy tính', 'Kỹ thuật phần mềm']:
-                result['Khoa'] = 'Công nghệ thông tin (FIT)'
+                khoa = 'Công nghệ thông tin (FIT)'
+                cond_value = cond
             elif cond in ['Toán kinh tế', 'Phân tích dữ liệu trong Kinh tế', 'Định phí bảo hiểm và Quản trị rủi ro']:
-                result['Khoa'] = 'Toán Kinh tế (MFE)'
+                khoa = 'Toán Kinh tế (MFE)'
                 if cond == 'Toán kinh tế':
-                    cond == 'Toán kinh tế (TOKT)'
+                    cond_value = 'Toán kinh tế (TOKT)'
                 elif cond == 'Phân tích dữ liệu trong Kinh tế':
-                    cond == 'Phân tích dữ liệu trong Kinh tế (DSEB)'
+                    cond_value = 'Phân tích dữ liệu trong Kinh tế (DSEB)'
                 elif cond == 'Định phí bảo hiểm và Quản trị rủi ro':
-                    cond == 'Định phí bảo hiểm và Quản trị rủi ro (Actuary)'
+                    cond_value = 'Định phí bảo hiểm và Quản trị rủi ro (Actuary)'
             elif cond in ['Khoa học dữ liệu', 'Trí tuệ nhân tạo']:
-                result['Khoa'] = 'Khoa học dữ liệu và Trí tuệ nhân tạo (FDA)'
+                khoa = 'Khoa học dữ liệu và Trí tuệ nhân tạo (FDA)'
+                cond_value = cond
             elif cond == 'Thống kê kinh tế':
-                result['Khoa'] = 'Thống kê'
+                khoa = 'Thống kê'
+                cond_value = cond
             elif cond in ['Hệ thống thông tin', 'Hệ thống thông tin quản lý']:
-                result['Khoa'] = 'Hệ thống thông tin quản lý (MIS)'
-            result['Nganh'] = cond
-    return result
+                khoa = 'Hệ thống thông tin quản lý (MIS)'
+                cond_value = cond
+            else:
+                khoa = None
+                cond_value = cond
+            all_nganh.append(cond_value)
+            if khoa is not None:
+                all_khoa.append(khoa)
+        result['Nganh'] = all_nganh
+        result['Khoa'] = all_khoa
+    else:
+        # Nếu không phải "Đề án", duyệt từng điều kiện trong condition4 dựa trên input_text
+        for cond in condition4:
+            if cond in input_text:
+                if cond in ['An toàn thông tin', 'Công nghệ thông tin', 'Khoa học máy tính', 'Kỹ thuật phần mềm']:
+                    result['Khoa'] = 'Công nghệ thông tin (FIT)'
+                    result['Nganh'] = cond
+                elif cond in ['Toán kinh tế', 'Phân tích dữ liệu trong Kinh tế', 'Định phí bảo hiểm và Quản trị rủi ro']:
+                    result['Khoa'] = 'Toán Kinh tế (MFE)'
+                    if cond == 'Toán kinh tế':
+                        result['Nganh'] = 'Toán kinh tế (TOKT)'
+                    elif cond == 'Phân tích dữ liệu trong Kinh tế':
+                        result['Nganh'] = 'Phân tích dữ liệu trong Kinh tế (DSEB)'
+                    elif cond == 'Định phí bảo hiểm và Quản trị rủi ro':
+                        result['Nganh'] = 'Định phí bảo hiểm và Quản trị rủi ro (Actuary)'
+                elif cond in ['Khoa học dữ liệu', 'Trí tuệ nhân tạo']:
+                    result['Khoa'] = 'Khoa học dữ liệu và Trí tuệ nhân tạo (FDA)'
+                    result['Nganh'] = cond
+                elif cond == 'Thống kê kinh tế':
+                    result['Khoa'] = 'Thống kê'
+                    result['Nganh'] = cond
+                elif cond in ['Hệ thống thông tin', 'Hệ thống thông tin quản lý']:
+                    result['Khoa'] = 'Hệ thống thông tin quản lý (MIS)'
+                    result['Nganh'] = cond
+    return result
+    # #cond1 cond2 la str, con3 la list ten cac nganh
+    # result = {}
+    # if condition3 in input_text:
+    #     result['Tai lieu ve'] = 'Đề cương'
+    # elif condition1 in input_text:
+    #     result['Tai lieu ve'] = 'Chương trình đào tạo'
+    # elif condition2 in input_text:
+    #     result['Tai lieu ve'] = 'Đề án'
+    # for cond in condition4:
+    #     if cond in input_text:
+    #         if cond in ['An toàn thông tin', 'Công nghệ thông tin', 'Khoa học máy tính', 'Kỹ thuật phần mềm']:
+    #             result['Khoa'] = 'Công nghệ thông tin (FIT)'
+    #         elif cond in ['Toán kinh tế', 'Phân tích dữ liệu trong Kinh tế', 'Định phí bảo hiểm và Quản trị rủi ro']:
+    #             result['Khoa'] = 'Toán Kinh tế (MFE)'
+    #             if cond == 'Toán kinh tế':
+    #                 cond == 'Toán kinh tế (TOKT)'
+    #             elif cond == 'Phân tích dữ liệu trong Kinh tế':
+    #                 cond == 'Phân tích dữ liệu trong Kinh tế (DSEB)'
+    #             elif cond == 'Định phí bảo hiểm và Quản trị rủi ro':
+    #                 cond == 'Định phí bảo hiểm và Quản trị rủi ro (Actuary)'
+    #         elif cond in ['Khoa học dữ liệu', 'Trí tuệ nhân tạo']:
+    #             result['Khoa'] = 'Khoa học dữ liệu và Trí tuệ nhân tạo (FDA)'
+    #         elif cond == 'Thống kê kinh tế':
+    #             result['Khoa'] = 'Thống kê'
+    #         elif cond in ['Hệ thống thông tin', 'Hệ thống thông tin quản lý']:
+    #             result['Khoa'] = 'Hệ thống thông tin quản lý (MIS)'
+    #         result['Nganh'] = cond
+    # return result