Update app.py

app.py

@@ -33,7 +33,8 @@ def main():
         prompt_1 = prompt_1.strip()
         
         # Save 'prompt_1' to the chat history
-        st.session_state['chat_history'] += "<div style='color: blue;'>You: </div><br><hr style='border: 1px solid black;'>"
+        #st.session_state['chat_history'] += "<div style='color: blue;'>You: </div><br><hr style='border: 1px solid black;'>"
+        st.session_state['chat_history'] += "<div style='color: blue;'>You: </div>"
         st.session_state['chat_history'] += f"{prompt_1}"
         st.session_state['chat_history'] += "<br><hr style='border: 1px solid black;'>"
 
@@ -59,7 +60,8 @@ def main():
         answers_1 = '\n\n'.join(out_list)
         
         # Save GPT's response to the chat history
-        st.session_state['chat_history'] += "<div style='color: red;'>MidikanGPT-4: </div><br><hr style='border: 1px solid black;'>"
+        #st.session_state['chat_history'] += "<div style='color: red;'>MidikanGPT-4: </div><br><hr style='border: 1px solid black;'>"
+        st.session_state['chat_history'] += "<div style='color: red;'>MidikanGPT-4: </div>"
         st.session_state['chat_history'] += f"{answers_1}"
         st.session_state['chat_history'] += "<br><hr style='border: 1px solid black;'>"
         
@@ -68,7 +70,8 @@ def main():
         prompt_2 = prompt_2.strip()
         
         # Save 'prompt_2' to the chat history
-        st.session_state['chat_history'] += "<div style='color: blue;'>You: </div><br><hr style='border: 1px solid black;'>"
+        #st.session_state['chat_history'] += "<div style='color: blue;'>You: </div><br><hr style='border: 1px solid black;'>"
+        st.session_state['chat_history'] += "<div style='color: blue;'>You: </div>"
         st.session_state['chat_history'] += f"{prompt_2}"
         st.session_state['chat_history'] += "<br><hr style='border: 1px solid black;'>"
         
@@ -85,7 +88,8 @@ def main():
         answers_2 = response['choices'][0]['message']['content']
         
         # Save GPT's response to the chat history
-        st.session_state['chat_history'] += "<div style='color: red;'>MidikanGPT-4: </div><br><hr style='border: 1px solid black;'>"
+        #st.session_state['chat_history'] += "<div style='color: red;'>MidikanGPT-4: </div><br><hr style='border: 1px solid black;'>"
+        st.session_state['chat_history'] += "<div style='color: red;'>MidikanGPT-4: </div>"
         st.session_state['chat_history'] += f"{answers_2}"
         st.session_state['chat_history'] += "<br><hr style='border: 1px solid black;'>"
         
@@ -93,7 +97,8 @@ def main():
         prompt_3 = answers_1 + '\n\n' + answers_2 + '\n\n' + "You are a resolver tasked with 1) finding which of the 3 answer options the researcher thought was best 2) improving that answer, and 3) printing the improved answer in full. Let's work this out in a step by step way to be sure we have the right answer:"
 
         # Save the end of 'prompt_3' to the chat history
-        st.session_state['chat_history'] += "<div style='color: blue;'>You: </div><br><hr style='border: 1px solid black;'>"
+        #st.session_state['chat_history'] += "<div style='color: blue;'>You: </div><br><hr style='border: 1px solid black;'>"
+        st.session_state['chat_history'] += "<div style='color: blue;'>You: </div>"
         st.session_state['chat_history'] += "You are a resolver tasked with 1) finding which of the 3 answer options the researcher thought was best 2) improving that answer, and 3) printing the improved answer in full. Let's work this out in a step by step way to be sure we have the right answer:"
         st.session_state['chat_history'] += "<br><hr style='border: 1px solid black;'>"
         
@@ -110,7 +115,8 @@ def main():
         answers_3 = response['choices'][0]['message']['content']
         
         # Save GPT's response to the chat history
-        st.session_state['chat_history'] += "<div style='color: red;'>MidikanGPT-4: </div><br><hr style='border: 1px solid black;'>"
+        #st.session_state['chat_history'] += "<div style='color: red;'>MidikanGPT-4: </div><br><hr style='border: 1px solid black;'>"
+        st.session_state['chat_history'] += "<div style='color: red;'>MidikanGPT-4: </div>"
         st.session_state['chat_history'] += f"{answers_3}"
         st.session_state['chat_history'] += "<br><hr style='border: 1px solid black;'>"