Update app.py

app.py

@@ -34,13 +34,15 @@ def load_icd_codes_from_files():
                 file_path = os.path.join(directory_path, file_name)
                 with open(file_path, "r", encoding="utf-8") as file:
                     for line in file:
-                        parts = line.strip().split("\t")  # Adjust delimiter as needed
-                        if len(parts) >= 2:
+                        # Split the line into code and description
+                        parts = line.strip().split(maxsplit=1)
+                        if len(parts) == 2:
                             code = parts[0].strip()
                             description = parts[1].strip()
                             icd_codes[code] = description
     else:
         print(f"Directory {directory_path} does not exist!")
+
     return icd_codes
 
 # Load CPT codes from files
@@ -54,13 +56,15 @@ def load_cpt_codes_from_files():
                 file_path = os.path.join(directory_path, file_name)
                 with open(file_path, "r", encoding="utf-8") as file:
                     for line in file:
-                        parts = line.strip().split("\t")  # Adjust delimiter as needed
-                        if len(parts) >= 2:
+                        # Split the line into code and description
+                        parts = line.strip().split(maxsplit=1)
+                        if len(parts) == 2:
                             code = parts[0].strip()
                             description = parts[1].strip()
                             cpt_codes[code] = description
     else:
         print(f"Directory {directory_path} does not exist!")
+
     return cpt_codes
 
 # Load ICD and CPT codes dynamically