Update app.py

app.py

@@ -93,12 +93,7 @@ if st.session_state.logged_in:
         st.chat_message("user").markdown(prompt)
     
         # Add user message to chat history
-        # st.session_state.messages.append()
-        st.session_state.messages.append({"role": "user", "content": prompt})
-    
-        # API Call
-        bot = ChatBot(
-            protocol={"role": "system", "content": f"""
+        st.session_state.messages.append({"role": "system", "content": f"""
         You are a helpful assistant assiting users on GPU selections.
         Here's the data:
         {df.to_markdown(index=False)}
@@ -106,10 +101,14 @@ if st.session_state.logged_in:
         User may ask what is the best GPU selection.
         You will need to ask user: 1) type of task, 2) size of data, 3) size of models.
         You will then make a suggestion of what type of GPU or instance is the best for the user.
-        """},
-            conversation=st.session_state.messages.copy()
-        )
-        # bot.conversation = st.session_state.messages.copy()  # Update history from messages
+        """})
+        st.session_state.messages.append({"role": "user", "content": prompt})
+    
+        # API Call
+        bot = ChatBot()
+        st.write(st.session_state.messages.copy())
+        st.type(st.session_state.messages.copy())
+        bot.history = st.session_state.messages.copy()  # Update history from messages
         response = bot.generate_response(prompt)
     
         # Display assistant response in chat message container