Upload model0.py

model0.py

@@ -3,6 +3,7 @@ import openai
 from dotenv import load_dotenv
 import os
 import json
+import checkTool
 
 def model0(path):
     ocr = CnOcr(rec_model_name='en_PP-OCRv3')
@@ -19,25 +20,29 @@ def model0(path):
         if item['text'] not in invalid_list:
             data_set_1.append(item['text'])
 
+    print(f'All data here: {data_set_1}')
+
     completion = openai.ChatCompletion.create(
         model = "gpt-3.5-turbo",
         temperature = 0,
         messages = [
             {"role": "system", "content": "You are an AI assistant for extracting data from HKID card with following information \
-                (name, HKID number, date of issue) from HKID card. Uppercase and lowercase letters are the same. Store the results in \
+                (name, date of birth, date of issue, HKID number) from HKID card. Uppercase and lowercase letters are the same. Store the results in \
                 dictionary format"},
             {"role": "user", "content": f"Extract data from the following set of text: {data_set_1}. \
                 You have three types of data to extract. \
                 1. id card holder full name (it noramlly is a chinese name, including surname and family \
                 name in English spelling, and it may be separate in different fields in the data set for surname and family name \
                 sometimes) \
-                2. issue date (should be a date with month and day, e.g. 19-97 is the required format, but 26-11-18 is not \
-                because date of issue of have 5 characters) Only choose valid format!!! \
-                3. HKID number (The standard format of HKID number is @123456(#) e.g. A123456(7) is a valid HKID number. \
+                2. date of birth (should be a date with year, month and day, e.g. 23-02-2003 is the required format, but 26-11 is not \
+                because date of birth should have 10 characters) Only choose valid format!!!\
+                3. date of issue (a string with format xx-xx) \
+                4. HKID number (The standard format of HKID number is @123456(#) e.g. A123456(7) is a valid HKID number. \
                 (a) @ represents any one or two capital letters of the alphabet. \
                 (b) # is the check digit which has 11 possible values from 0 to 9 and A.) \
                 Remember to include the check digit with () \
                 Only reply a dictionary. No need to add other words or explanation. Use double quote for dictionary."},
+                
         ]
     )
 
@@ -46,9 +51,17 @@ def model0(path):
     print(data)
 
     id_data = json.loads(data)
+
+    name = id_data["name"]
+    dateofbirth = id_data["date of birth"]
+    issuedate = id_data["date of issue"]
+    hkid = id_data["HKID number"]
+    if checkTool.validate_hkid(hkid=hkid):
+        valid_hkid = 'True'
+    else:
+        valid_hkid = 'False'
+    name = checkTool.seperate_name(name)
     
     print(id_data)
-    return
+    return [name, valid_hkid, hkid, issuedate, dateofbirth]
     # return [name, valid_hkid, hkid, issuedate]
-
-model0('dontTouchMe/IMG_4499.jpg')