Update app.py

app.py

@@ -1,3 +1,61 @@
+import streamlit as st
+from services.claude_service import ClaudeService
+from services.chart_analysis import ChartAnalysisService
+from ui.components import (
+    create_sidebar,
+    show_analysis_section,
+    show_chat_history,
+    show_follow_up_section,
+    show_save_options,
+    create_expertise_selector
+)
+from utils.file_handlers import save_chat_history
+from utils.learning_module import LearningModule
+
+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:
+        st.session_state.current_image = None
+    if 'current_analysis' not in st.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 = Noneimport streamlit as st
+from services.claude_service import ClaudeService
+from services.chart_analysis import ChartAnalysisService
+from ui.components import (
+    create_sidebar,
+    show_analysis_section,
+    show_chat_history,
+    show_follow_up_section,
+    show_save_options,
+    create_expertise_selector
+)
+from utils.file_handlers import save_chat_history
+from utils.learning_module import LearningModule
+
+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:
+        st.session_state.current_image = None
+    if 'current_analysis' not in st.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
+
+
+
 def show_chart_analysis_tab(claude_service, analysis_service):
     """Display chart analysis functionality"""
     # Get user expertise level
@@ -115,139 +173,6 @@ def show_chart_analysis_tab(claude_service, analysis_service):
             else:
                 st.info("Chat saved to session state")# app.py
 
-import streamlit as st
-from services.claude_service import ClaudeService
-from services.chart_analysis import ChartAnalysisService
-from ui.components import (
-    create_sidebar,
-    show_analysis_section,
-    show_chat_history,
-    show_follow_up_section,
-    show_save_options,
-    create_expertise_selector
-)
-from utils.file_handlers import save_chat_history
-from utils.learning_module import LearningModule
-
-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:
-        st.session_state.current_image = None
-    if 'current_analysis' not in st.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
-
-def show_chart_analysis_tab(claude_service, analysis_service):
-    """Display chart analysis functionality"""
-    # Get user expertise level
-    expertise_level = create_expertise_selector()
-    
-    # Create sidebar and get inputs
-    upload_option, uploaded_files, patterns, indicators, comparison_type = create_sidebar()
-    
-    # Main content area
-    col1, col2 = st.columns([2, 1])
-    
-    with col1:
-        st.title("📈 Stock Chart Analysis Assistant")
-        
-        if upload_option == "Ask Question":
-            question = st.text_input(
-                "What would you like to know?",
-                key="main_question_input"
-            )
-        
-        # Main analysis section
-        if uploaded_files:
-            # Store all uploaded images
-            st.session_state.current_images = [file.getvalue() for file in uploaded_files]
-            
-        analyze_clicked = show_analysis_section(uploaded_files)
-        
-        if analyze_clicked:
-            if not uploaded_files:
-                st.warning("Please upload at least one chart.")
-            else:
-                with st.spinner("Analyzing charts..."):
-                    results = analysis_service.analyze_multiple_charts(
-                        st.session_state.current_images,
-                        patterns,
-                        indicators,
-                        comparison_type,
-                        expertise_level
-                    )
-                    
-                    if results:
-                        # Store current analysis
-                        st.session_state.current_analysis = results[-1]['analysis']  # Latest analysis
-                        st.session_state.chat_history.extend(results)
-                        
-                        # Display analyses
-                        # First show comparative analysis if it exists
-                        for result in results:
-                            if result.get('analysis_type') != 'Individual':
-                                st.subheader(f"{result['analysis_type']} Results")
-                                st.write(result['analysis'])
-                                # Add follow-up section immediately
-                                follow_up = show_follow_up_section(f"comparative_followup")
-                                if follow_up:
-                                    with st.spinner("Processing follow-up..."):
-                                        response = analysis_service.handle_follow_up_question(
-                                            follow_up,
-                                            result['analysis'],
-                                            st.session_state.current_images
-                                        )
-                                        if response:
-                                            st.session_state.chat_history.append(response)
-                                            st.write(response['analysis'])
-                        
-                        # Then show individual analyses
-                        individual_analyses = [r for r in results if r.get('analysis_type') == 'Individual']
-                        for idx, result in enumerate(individual_analyses):
-                            st.subheader(f"Analysis Results - Chart {idx + 1}")
-                            st.write(result['analysis'])
-                            # Add follow-up section for each individual analysis
-                            follow_up = show_follow_up_section(f"individual_followup_{idx}")
-                            if follow_up:
-                                with st.spinner("Processing follow-up..."):
-                                    response = analysis_service.handle_follow_up_question(
-                                        follow_up,
-                                        result['analysis'],
-                                        st.session_state.current_images[idx] if idx < len(st.session_state.current_images) else None
-                                    )
-                                    if response:
-                                        st.session_state.chat_history.append(response)
-                                        st.write(response['analysis'])
-                        
-                        # Risk warning
-                        st.warning(
-                            "⚠️ This analysis is AI-generated and for informational purposes only. "
-                            "Do not make trading decisions solely based on this information."
-                        )
-    
-    with col2:
-        # Chat history
-        show_chat_history(st.session_state.chat_history)
-        
-        # Save options
-        save_name = show_save_options()
-        if save_name and st.session_state.chat_history:
-            filename = save_chat_history(
-                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
-            )
-            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"""
@@ -271,6 +196,18 @@ def main():
         initial_sidebar_state="expanded"
     )
     
+    # Initialize authentication
+    auth_manager = AuthManager()
+    
+    # Show login page if not authenticated
+    if not auth_manager.is_authenticated():
+        show_login_page(auth_manager)
+        return
+        
+    # Get user's storage paths
+    storage_paths = auth_manager.get_user_storage_paths()
+    storage_manager = UserStorageManager(storage_paths)
+    
     # Initialize services
     try:
         claude_service = ClaudeService()
@@ -283,14 +220,33 @@ def main():
     # Initialize session state
     init_session_state()
     
+    # Show logout button in sidebar
+    show_logout_button(auth_manager)
+    
+    # Load previous context if available
+    if 'chat_history' not in st.session_state:
+        context = storage_manager.get_context()
+        if context:
+            st.session_state.chat_history = context.get('chat_history', [])
+            st.session_state.current_analysis = context.get('current_analysis')
+            
     # Main application tabs
-    tab1, tab2 = st.tabs(["Chart Analysis", "Learning Center"])
+    tab1, tab2, tab3 = st.tabs(["Chart Analysis", "Learning Center", "Previous Chats"])
     
     with tab1:
-        show_chart_analysis_tab(claude_service, analysis_service)
+        show_chart_analysis_tab(claude_service, analysis_service, storage_manager)
         
     with tab2:
         show_learning_tab(learning_module)
+        
+    with tab3:
+        show_previous_chats_tab(storage_manager)
+        
+    # Save context before closing
+    storage_manager.save_context({
+        'chat_history': st.session_state.chat_history,
+        'current_analysis': st.session_state.current_analysis
+    })
 
 if __name__ == "__main__":
     main()