Upload 2 files

get_chinese_code.py

@@ -53,9 +53,11 @@ def get_chinese_name(path):
             chinese_name.append(df['word'][df['code'] == str(w2)].iloc[0])
             if w3 is not None:
                 chinese_name.append(df['word'][df['code'] == str(w3)].iloc[0])
-            return chinese_name
+            chinese_complete = ""
+            for item in chinese_name:
+                chinese_complete = chinese_complete + item
+            return chinese_complete
         chinese_name = []
+    return ""
 
-    return []
-
-print(get_chinese_name('dontTouchMe/IMG_4495.jpg'))
+# print(get_chinese_name('image\hkid.jpg'))

get_chinese_name.py

@@ -23,9 +23,9 @@ def get_chiname(path):
         model = "gpt-3.5-turbo",
         temperature = 0,
         messages = [
-            {"role": "system", "content": "You are an AI assistant for extracting Chinese name from HKID card."},
+            {"role": "system", "content": "You are an AI assistant for extracting Chinese name (usually in length of three) from Hong Kong ID card."},
             {"role": "user", "content": f"Extract data from the following set of text: {data_set_1}. \
-                You only need to return a dictonary with Chinese name in Chinese. Use double quote!"},
+                You only need to return a dictonary with Chinese name in Chinese. Use double quote."},
         ]
     )