Update app.py

app.py

@@ -113,29 +113,31 @@ def ask_ds(message, history):
         divisions = {'1': 'ime', '2': 'pas', '3': 'peer disability'}
         if division == None:
             if message.lower().strip() in list(divisions.values()):
-                division = message.lower().strip().replace(' ', '_')
+                division = message.lower().strip()
                 yield "[1] CSR\n[2] QA"
             elif message.lower().strip() in list(divisions.keys()):
-                division = divisions[message.lower().strip()].replace(' ', '_')
+                division = divisions[message.lower().strip()]
                 yield "[1] CSR\n[2] QA"
             else:
                 yield "Please select a valid choice."
                 
         elif employee_type == None:
+
             if division == 'peer disability':
                 division = 'dis'
+                
             employee_types = {'1': 'csr', '2': 'qa'}
             if message.lower().strip() in list(employee_types.values()):
                 employee_type = message.lower().strip()
                 authenticated = True
-                EXTRACTIONS_PATH = EXTRACTIONS_PATH.replace('{employee-type}', employee_type).replace('{division}', division[:3])
+                EXTRACTIONS_PATH = EXTRACTIONS_PATH.replace('{employee_type}', employee_type).replace('{division}', division)
                 yield EXTRACTIONS_PATH
                 extractions = read_json_from_s3()
                 yield "Welcome to Ask Dane Street! Whether you're new to the team or just looking for some quick information, I'm here to guide you through our company's literature and platform. Simply ask your question, and I'll provide you with the most relevant information I can."
             elif message.lower().strip() in list(employee_types.keys()):
                 employee_type = employee_types[message.lower().strip()]
                 authenticated = True
-                EXTRACTIONS_PATH = EXTRACTIONS_PATH.replace('{employee_type}', employee_type).replace('{division}', division[:3])
+                EXTRACTIONS_PATH = EXTRACTIONS_PATH.replace('{employee_type}', employee_type).replace('{division}', division)
                 yield EXTRACTIONS_PATH
                 extractions = read_json_from_s3()
                 yield "Welcome to Ask Dane Street! Whether you're new to the team or just looking for some quick information, I'm here to guide you through our company's literature and platform. Simply ask your question, and I'll provide you with the most relevant information I can."