update streamlit_app.py 2

streamlit_app.py

@@ -41,8 +41,13 @@ def main():
 
     # Display chat history
     for message in st.session_state.messages:
-        with st.chat_message(message["role"]):
-            st.markdown(message["content"])
+        role = message["role"]
+        content = message["content"]
+        if hasattr(st, "chat_message"):
+            with st.chat_message(role):
+                st.markdown(content)
+        else:
+            st.write(f"**{role.capitalize()}:** {content}")
 
     # Chat input and response handling
     # Check if st.chat_input is available (Streamlit 1.2 or higher)
@@ -55,13 +60,19 @@ def main():
         st.session_state.messages.append({"role": "user", "content": prompt})
 
         # Display user message
-        with st.chat_message("user"):
-            st.markdown(prompt)
+        if hasattr(st, "chat_message"):
+            with st.chat_message("user"):
+                st.markdown(prompt)
+        else:
+            st.write(f"**User:** {prompt}")
 
         # Generate response (cached if already asked before)
-        with st.chat_message("assistant"):
-            response = cached_response(prompt)
-            st.markdown(response)
+        if hasattr(st, "chat_message"):
+            with st.chat_message("assistant"):
+                response = cached_response(prompt)
+                st.markdown(response)
+        else:
+            st.write(f"**Assistant:** {response}")
 
         # Store bot response in chat history
         st.session_state.messages.append({"role": "assistant", "content": response})