Update app.py

app.py

@@ -19,7 +19,6 @@ from pages.login import show_login_page, show_logout_button
 from pages.previous_chats import show_previous_chats_tab
 
 def init_session_state():
-    """Initialize session state variables"""
     if 'chat_history' not in st.session_state:
         st.session_state.chat_history = []
     if 'current_image' not in st.session_state:
@@ -28,12 +27,12 @@ def init_session_state():
         st.session_state.current_analysis = None
     if 'current_images' not in st.session_state:
         st.session_state.current_images = []
-    if 'conversation_context' not in st.session_state:
-        st.session_state.conversation_context = []
-    if 'current_conversation_id' not in st.session_state:
-        st.session_state.current_conversation_id = None
     if 'analysis_results' not in st.session_state:
         st.session_state.analysis_results = []
+    if 'followups' not in st.session_state:
+        st.session_state.followups = []
+    if 'conversation_context' not in st.session_state:
+        st.session_state.conversation_context = []
         
 def reset_session_state():
     """Reset session state for a new chat"""
@@ -60,7 +59,7 @@ def show_chart_analysis_tab(claude_service, analysis_service, storage_manager):
                 st.session_state.chat_history,
                 st.session_state.current_images,
                 st.session_state.get('last_saved_chat'),
-                claude_service  # Pass claude_service for smart naming
+                claude_service
             )
             st.success("Previous chat saved! Starting new chat...")
         
@@ -113,14 +112,18 @@ def show_chart_analysis_tab(claude_service, analysis_service, storage_manager):
             results = st.session_state.analysis_results
             
             # First show comparative analysis if it exists
-            for result in results:
+            for idx, result in enumerate(results):
                 if result.get('analysis_type') != 'Individual':
                     with st.container():
                         st.subheader(f"{result['analysis_type']} Results")
                         st.markdown(result['analysis'])
                         
-                        # Add follow-up section immediately after analysis
-                        follow_up = show_follow_up_section(f"comparative_followup")
+                        # Add enhanced follow-up section with context
+                        follow_up = show_follow_up_section(
+                            key_suffix=f"comparative_{idx}",
+                            previous_response=result['analysis']
+                        )
+                        
                         if follow_up:
                             with st.spinner("Processing follow-up..."):
                                 response = analysis_service.handle_follow_up_question(
@@ -130,7 +133,10 @@ def show_chart_analysis_tab(claude_service, analysis_service, storage_manager):
                                 )
                                 if response:
                                     st.session_state.chat_history.append(response)
+                                    
+                                    # Show response in the chat interface
                                     st.markdown(response['analysis'])
+                                    
                                     # Save to storage with smart naming
                                     storage_manager.save_chat(
                                         st.session_state.chat_history,
@@ -146,9 +152,12 @@ def show_chart_analysis_tab(claude_service, analysis_service, storage_manager):
                     st.subheader(f"Analysis Results - Chart {idx + 1}")
                     st.markdown(result['analysis'])
                     
-                    # Add follow-up section for each analysis
-                    container_key = f"individual_followup_{idx}"
-                    follow_up = show_follow_up_section(container_key)
+                    # Add enhanced follow-up section for each analysis
+                    follow_up = show_follow_up_section(
+                        key_suffix=f"individual_{idx}",
+                        previous_response=result['analysis']
+                    )
+                    
                     if follow_up:
                         with st.spinner("Processing follow-up..."):
                             response = analysis_service.handle_follow_up_question(
@@ -158,7 +167,10 @@ def show_chart_analysis_tab(claude_service, analysis_service, storage_manager):
                             )
                             if response:
                                 st.session_state.chat_history.append(response)
+                                
+                                # Show response in the chat interface
                                 st.markdown(response['analysis'])
+                                
                                 # Save to storage with smart naming
                                 storage_manager.save_chat(
                                     st.session_state.chat_history,
@@ -172,25 +184,40 @@ def show_chart_analysis_tab(claude_service, analysis_service, storage_manager):
                 "⚠️ This analysis is AI-generated and for informational purposes only. "
                 "Do not make trading decisions solely based on this information."
             )
-    #Kapil : Have commented it out as it would be good to have more space for analysis
-    #with col2:
-        # Chat history
-       # show_chat_history(st.session_state.chat_history)
+    
+    with col2:
+        # Add a collapsible history section
+        with st.expander("💬 View Chat History", expanded=False):
+            if st.session_state.chat_history:
+                # Show only last 3 interactions by default
+                show_full = st.checkbox("Show full history", value=False)
+                
+                if show_full:
+                    history_to_show = st.session_state.chat_history
+                else:
+                    history_to_show = st.session_state.chat_history[-3:]
+                    if len(st.session_state.chat_history) > 3:
+                        st.info(f"Showing last 3 of {len(st.session_state.chat_history)} interactions")
+                
+                # Display the selected history
+                show_chat_history(history_to_show)
+            else:
+                st.info("No chat history yet")
         
-        # Save options
-     #   save_name = show_save_options()
-      #  if save_name and st.session_state.chat_history:
-       #     filename = storage_manager.save_chat(
-        #        st.session_state.chat_history,
-         #       st.session_state.current_images[0] if st.session_state.current_images else None,
-          #      f"{save_name}.json" if save_name else None,
-           #     claude_service
-            #)
-            #if filename:
-             #   st.success(f"Chat saved as {filename}")
-            #else:
-             #   st.info("Chat saved to session state")
-
+        # Add save options in a separate expander
+        with st.expander("💾 Save Analysis", expanded=False):
+            save_name = show_save_options()
+            if save_name and st.session_state.chat_history:
+                filename = storage_manager.save_chat(
+                    st.session_state.chat_history,
+                    st.session_state.current_images[0] if st.session_state.current_images else None,
+                    f"{save_name}.json" if save_name else None,
+                    claude_service
+                )
+                if filename:
+                    st.success(f"Chat saved as {filename}")
+                else:
+                    st.info("Chat saved to session state")
 def show_learning_tab(learning_module):
     """Display learning center functionality"""
     st.title("📚 Trading Learning Center")