Update ui/components.py

ui/components.py

@@ -100,21 +100,14 @@ def show_chat_history(chat_history):
             st.markdown(chat.get('analysis', ''))
             st.markdown("---")
 
-def show_follow_up_section(key_suffix=""):
-    """Show the follow-up question section with enhanced UI"""
-    st.markdown("---")  # Visual separator
-    
-    # Chat container
-    chat_container = st.container()
-    with chat_container:
-        # Show conversation context if exists
-        if 'conversation_context' in st.session_state:
-            for msg in st.session_state.conversation_context:
-                if msg['role'] == 'user':
-                    st.markdown(f"**You:** {msg['content']}")
-                else:
-                    st.markdown(f"**Assistant:** {msg['content']}")
-                st.markdown("---")
+def show_follow_up_section(key_suffix="", previous_response=None):
+    """Show the follow-up question section with enhanced chat UI"""
+    # Show previous response if provided
+    if previous_response:
+        with st.container():
+            st.markdown("**Previous Response:**")
+            st.markdown(previous_response)
+            st.markdown("---")
     
     # Input section
     col1, col2 = st.columns([4, 1])