Update app.py

app.py

@@ -6,7 +6,7 @@ from huggingface_hub import InferenceApi, login, InferenceClient
 # Streamlit app configuration
 st.set_page_config(page_title="Medical Chatbot")
 st.title("Medical Chatbot 🤖")
-
+st.subheader("", divider='rainbow')
 # Get the Hugging Face token from environment variables
 hf_token = os.getenv("HF_TOKEN")
 if hf_token is None:
@@ -54,11 +54,10 @@ if 'messages' not in st.session_state:
         {"role": "system", "content": "You are a knowledgeable and empathetic medical assistant providing accurate and compassionate health advice based on user input."}
     ]
 
-# Display chat history
-# for message in st.session_state.messages:
-#     with st.chat_message(message["role"]):
-#         st.markdown(message["content"])
-st.subheader("", divider='rainbow')
+Display chat history
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
 
 def respond(message, history, max_tokens, temperature, top_p):
     # Prepare the list of messages for the chat completion