Update app.py

app.py

@@ -21,12 +21,6 @@ def main():
     if 'chat_history' not in st.session_state:
         st.session_state['chat_history'] = ''
     
-    
-    
-    
-    #st.session_state['chat_history'].append("<div style='color: blue;'>You: </div>")
-    #st.session_state['chat_history'].append("<div style='color: red;'>MidikanGPT-4: </div>")
-    
     # Handle the send button click
     if send_button:
         # Set the system prompt
@@ -41,6 +35,7 @@ def main():
         # 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'] += f"{prompt_1}"
+        st.session_state['chat_history'] += "<br><hr style='border: 1px solid black;'>"
 
         # Hit the OpenAI API with 'prompt_1'
         response = openai.ChatCompletion.create(
@@ -64,14 +59,18 @@ def main():
         answers_1 = '\n\n'.join(out_list)
         
         # Save GPT's response to the chat history
-        st.session_state['chat_history'].append(f"<div style='color: red;'>MidikanGPT-4: {answers_1}</div>")
+        st.session_state['chat_history'] += "<div style='color: red;'>MidikanGPT-4: </div><br><hr style='border: 1px solid black;'>"
+        st.session_state['chat_history'] += f"{answers_1}"
+        st.session_state['chat_history'] += "<br><hr style='border: 1px solid black;'>"
         
         # Prepare 'prompt_2'
         prompt_2 = context + '\n\n' + txt_1 + question + '\n\n' + answers_1 + '\n\n' + "You are a researcher tasked with investigating the 3 answer options provided. List the flaws and faulty logic of each answer option. Let's work this out in a step by step way to be sure we have all the errors:"
         prompt_2 = prompt_2.strip()
         
         # Save 'prompt_2' to the chat history
-        st.session_state['chat_history'].append(f"You: {prompt_2}")
+        st.session_state['chat_history'] += "<div style='color: blue;'>You: </div><br><hr style='border: 1px solid black;'>"
+        st.session_state['chat_history'] += f"{prompt_2}"
+        st.session_state['chat_history'] += "<br><hr style='border: 1px solid black;'>"
         
         # Hit the OpenAI API with 'prompt_2'
         response = openai.ChatCompletion.create(
@@ -86,10 +85,17 @@ def main():
         answers_2 = response['choices'][0]['message']['content']
         
         # Save GPT's response to the chat history
-        st.session_state['chat_history'].append(f"SmartGPT-4: {answers_2}")
+        st.session_state['chat_history'] += "<div style='color: red;'>MidikanGPT-4: </div><br><hr style='border: 1px solid black;'>"
+        st.session_state['chat_history'] += f"{answers_2}"
+        st.session_state['chat_history'] += "<br><hr style='border: 1px solid black;'>"
         
         # Prepare 'prompt_3'
         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'] += "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;'>"
         
         # Hit the OpenAI API with 'prompt_3'
         response = openai.ChatCompletion.create(
@@ -104,12 +110,12 @@ def main():
         answers_3 = response['choices'][0]['message']['content']
         
         # Save GPT's response to the chat history
-        st.session_state['chat_history'].append(f"SmartGPT-4: {answers_3}")
+        st.session_state['chat_history'] += "<div style='color: red;'>MidikanGPT-4: </div><br><hr style='border: 1px solid black;'>"
+        st.session_state['chat_history'] += f"{answers_3}"
+        st.session_state['chat_history'] += "<br><hr style='border: 1px solid black;'>"
         
     # Creating an area to display the chat history
     st.markdown(st.session_state['chat_history'], unsafe_allow_html=True)
-    for message in st.session_state['chat_history']:
-        st.markdown(message)
 
 
 if __name__ == "__main__":