Update ui/components.py

ui/components.py

@@ -74,7 +74,6 @@ def show_chat_history(chat_history):
     """Display chat history"""
     st.subheader("Chat History")
     for idx, chat in enumerate(chat_history):
-        # Create a unique identifier for each analysis
         timestamp = chat.get('timestamp', 'No date')
         analysis_type = chat.get('analysis_type', 'Analysis')
         label = f"{analysis_type} from {timestamp[:16]}"
@@ -85,16 +84,26 @@ def show_chat_history(chat_history):
             if chat.get('question'):
                 st.write(f"Follow-up: {chat['question']}")
 
-def show_follow_up_section(current_analysis):
+def show_follow_up_section(key_suffix=""):
     """Show the follow-up question section"""
     st.subheader("Continue Analysis")
     follow_up = st.text_input(
         "Ask a follow-up question:",
-        key="followup_input"
+        key=f"followup_input_{key_suffix}"
     )
-    send_btn = st.button("Send", key="followup_send_btn")
+    send_btn = st.button("Send", key=f"followup_send_btn_{key_suffix}")
     return follow_up if send_btn else None
 
+def show_save_options():
+    """Show save chat options"""
+    st.subheader("Save Analysis")
+    save_name = st.text_input(
+        "Save chat as (optional):",
+        key="save_chat_name"
+    )
+    save_btn = st.button("Save", key="save_chat_btn")
+    return save_name if save_btn else None
+
 def create_expertise_selector():
     """Create expertise level selector"""
     if 'expertise_level' not in st.session_state: