Update app.py

app.py

@@ -93,7 +93,12 @@ if st.session_state.logged_in:
         st.chat_message("user").markdown(prompt)
     
         # Add user message to chat history
-        st.session_state.messages.append({"role": "system", "content": f"""
+        # st.session_state.messages.append()
+        st.session_state.messages.append({"role": "user", "content": prompt})
+    
+        # API Call
+        bot = ChatBot(
+            protocol={"role": "system", "content": f"""
         You are a helpful assistant assiting users on GPU selections.
         Here's the data:
         {df.to_markdown(index=False)}
@@ -101,11 +106,8 @@ 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.
-        """})
-        st.session_state.messages.append({"role": "user", "content": prompt})
-    
-        # API Call
-        bot = ChatBot()
+        """}
+        )
         bot.history = st.session_state.messages.copy()  # Update history from messages
         response = bot.generate_response(prompt)