Update app.py

app.py

@@ -141,7 +141,7 @@ for message in st.session_state.chat_history:
     if message["role"] == "user":
         st.markdown(f"<div class='user-msg'><strong>You:</strong> {message['content']}</div>", unsafe_allow_html=True)
     else:
-        st.markdown(f"<div class='assistant-msg'><strong>Bot:</strong> {message['content']}</div>", unsafe_allow_html=True)
+        st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {message['content']}</div>", unsafe_allow_html=True)
 
 st.markdown("</div>", unsafe_allow_html=True)
 
@@ -157,15 +157,27 @@ if st.button("Send"):
         )
 
         # Display response
-        st.markdown(f"<div class='assistant-msg'><strong>Bot:</strong> {response}</div>", unsafe_allow_html=True)
+        st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
 
         # Display NASA image if available
         if image_url:
             st.image(image_url, caption="NASA Image of the Day")
 
-        # Follow-up question suggestions
-        follow_up_options = [follow_up, "Explain differently", "Give me an example"]
-        selected_option = st.radio("What would you like to do next?", follow_up_options)
+        # Display a conversational follow-up message
+        st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {follow_up}</div>", unsafe_allow_html=True)
+
+        # Provide a text box for the user's next response
+        next_input = st.text_input("HAL is waiting for your response...")
+
+# If the user enters a follow-up message
+if next_input:
+    response, _, st.session_state.chat_history, _ = get_response(
+        system_message="You are a helpful AI assistant.",
+        user_text=next_input,
+        chat_history=st.session_state.chat_history
+    )
+    st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
+
 
         if st.button("Continue"):
             if selected_option:
@@ -174,4 +186,4 @@ if st.button("Send"):
                     user_text=selected_option,
                     chat_history=st.session_state.chat_history
                 )
-                st.markdown(f"<div class='assistant-msg'><strong>Bot:</strong> {response}</div>", unsafe_allow_html=True)
+                st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)