Upload app.py

app.py

@@ -125,7 +125,8 @@ class chat_manager():
 
       top_hits, search_string, _ = websearch_node(query, self.embed_model)
       local_chat_history.append(search_string)
-      self.chat_history.append(local_chat_history)
+      self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+      self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
       with open('chat_session_history.txt', 'w') as f:
         pp.pp(self.chat_history, stream=f)
 
@@ -180,7 +181,8 @@ manual mode. Reference the previous conversation in the context if needed."
       response = parts[1].strip('\n').strip('<end_of_turn>')
       self.latest_response = response
       local_chat_history.append(response)
-      self.chat_history.append(local_chat_history)
+      self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+      self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
       with open('chat_session_history.txt', 'w') as f:
         pp.pp(self.chat_history, stream=f)
       
@@ -221,7 +223,8 @@ manual mode. Reference the previous conversation in the context if needed."
           response = response.replace(m, f'```{m}```')
 
       local_chat_history.append(response)
-      self.chat_history.append(local_chat_history)
+      self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+      self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
       with open('chat_session_history.txt', 'w') as f:
         pp.pp(self.chat_history, stream=f)
 
@@ -248,7 +251,8 @@ manual mode. Reference the previous conversation in the context if needed."
           response += f'**{i+1}**. {list_name}\n'
         
         local_chat_history.append(response)
-        self.chat_history.append(local_chat_history)
+        self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+        self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
         with open('chat_session_history.txt', 'w') as f:
           pp.pp(self.chat_history, stream=f)
         return '', self.chat_history, None
@@ -264,7 +268,8 @@ manual mode. Reference the previous conversation in the context if needed."
         response = 'Invalid input. Please enter 1, 2, or 3 to choose one of the tools above, or hit enter to accept the #1 tool choice. \
 or enter "edit" to edit a list.'
         local_chat_history.append(response)
-        self.chat_history.append(local_chat_history)
+        self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+        self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
         with open('chat_session_history.txt', 'w') as f:
           pp.pp(self.chat_history, stream=f)
 
@@ -289,7 +294,8 @@ or enter "edit" to edit a list.'
       data_request += 'Please provide the necessary information to proceed.'
       if missing_data:
         local_chat_history.append(data_request)
-        self.chat_history.append(local_chat_history)
+        self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+        self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
         with open('chat_session_history.txt', 'w') as f:
           pp.pp(self.chat_history, stream=f)
         self.chat_idx = 999
@@ -320,7 +326,8 @@ or enter "edit" to edit a list.'
             self.results_string = self.results_string.replace(m, f'```{m}```')
 
         local_chat_history.append(self.results_string)
-        self.chat_history.append(local_chat_history)
+        self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+        self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
         with open('chat_session_history.txt', 'w') as f:
           pp.pp(self.chat_history, stream=f)
         try:
@@ -371,7 +378,8 @@ or enriching information where appropriate."
               response = response.replace(m, f'```{m}```')
 
       local_chat_history.append(response)
-      self.chat_history.append(local_chat_history)
+      self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+      self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
       with open('chat_session_history.txt', 'w') as f:
         pp.pp(self.chat_history, stream=f)
       self.chat_idx += 1
@@ -437,7 +445,8 @@ or enriching information where appropriate."
               response = response.replace(m, f'```{m}```')
 
       local_chat_history.append(response)
-      self.chat_history.append(local_chat_history)
+      self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+      self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
       with open('chat_session_history.txt', 'w') as f:
         pp.pp(self.chat_history, stream=f)
 
@@ -493,7 +502,8 @@ or enriching information where appropriate."
               response = response.replace(m, f'```{m}```')
 
       local_chat_history.append(response)
-      self.chat_history.append(local_chat_history)
+      self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+      self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
       with open('chat_session_history.txt', 'w') as f:
         pp.pp(self.chat_history, stream=f)
 
@@ -520,7 +530,8 @@ or enriching information where appropriate."
         self.chat_idx = 501
 
       local_chat_history.append(response)
-      self.chat_history.append(local_chat_history)
+      self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+      self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
       with open('chat_session_history.txt', 'w') as f:
         pp.pp(self.chat_history, stream=f)
       return '', self.chat_history, None  
@@ -545,7 +556,8 @@ or enriching information where appropriate."
           response = 'Invalid input. Please enter the numbers corresponding to the items you wish to keep.'
           self.chat_idx = 502
           local_chat_history.append(response)
-          self.chat_history.append(local_chat_history)
+          self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+          self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
           with open('chat_session_history.txt', 'w') as f:
             pp.pp(self.chat_history, stream=f)
           return '', self.chat_history, None
@@ -616,7 +628,8 @@ or enriching information where appropriate."
         self.chat_idx = 502
 
       local_chat_history.append(response)
-      self.chat_history.append(local_chat_history)
+      self.chat_history.append({'role': 'user', 'content': local_chat_history[0]})
+      self.chat_history.append({'role': 'assistant', 'content': local_chat_history[1]})
       with open('chat_session_history.txt', 'w') as f:
         pp.pp(self.chat_history, stream=f)