Update app.py

app.py

@@ -100,25 +100,37 @@ def ask_ds(message, history):
     global division
     global authenticated
     global extractions
+    global EXTRACTIONS_PATH
     
     if len(message) == 0:
         yield None
 
     if authenticated == False:
-        
+
+        divisions = {'1': 'ime', '2': 'pas', '3': 'peer disability'}
         if division == None:
-            if message.lower().strip() in ['ime', 'peer disability', 'pas']:
+            if message.lower().strip() in list(divisions.values()):
                 division = message.lower().strip().replace(' ', '_')
                 yield "[1] CSR\n[2] QA"
+            elif message.lower().strip() in list(divisions.keys()):
+                division = divisions[message.lower().strip()].replace(' ', '_')
+                yield "[1] CSR\n[2] QA"
             else:
                 yield "Please select a valid choice."
         elif employee_type == None:
-            if message.lower().strip() in ['csr', 'qa']:
+            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 = 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 = 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."
             else:
                 yield "Please select a valid choice."   
     else: