Update app.py

app.py

@@ -5,7 +5,7 @@ from huggingface_hub import InferenceClient
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 # Streamlit app configuration
-st.set_page_config(page_title="Health Care ChatBot", layout="wide")
+st.set_page_config(page_title="Health Care ChatBot")
 st.title("Health Care ChatBot")
 
 # Initialize session state for messages if not present
@@ -50,24 +50,24 @@ with st.sidebar:
     top_p = st.slider("Top-p (nucleus sampling)", 0.1, 1.0, 0.95)
 
 # Display chat messages from history
-# for message in st.session_state.messages:
-#     if message["role"] == "user":
-#         # User message on the right
-#         col1, col2 = st.columns([1, 4])
-#         with col2:
-#             with st.chat_message("user"):
-#                 st.write(message["content"])
-#         with col1:
-#             st.write("")  # Empty space on the left for alignment
+for message in st.session_state.messages:
+    if message["role"] == "user":
+        # User message on the right
+        col1, col2 = st.columns([1, 4])
+        with col2:
+            with st.chat_message("user"):
+                st.write(message["content"])
+        with col1:
+            st.write("")  # Empty space on the left for alignment
         
-#     elif message["role"] == "assistant":
-#         # Assistant message on the left
-#         col1, col2 = st.columns([4, 1])
-#         with col1:
-#             with st.chat_message("assistant"):
-#                 st.write(message["content"])
-#         with col2:
-#             st.write("")  # Empty space on the right for alignment
+    elif message["role"] == "assistant":
+        # Assistant message on the left
+        col1, col2 = st.columns([4, 1])
+        with col1:
+            with st.chat_message("assistant"):
+                st.write(message["content"])
+        with col2:
+            st.write("")  # Empty space on the right for alignment
 
 # Keep user input box at the bottom
 st.divider()  # Optional, to visually separate chat history from input box