Update app.py

app.py

@@ -8,9 +8,11 @@ from ui.components import (
     show_analysis_section,
     show_chat_history,
     show_follow_up_section,
-    show_save_options
+    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"""
@@ -23,23 +25,10 @@ def init_session_state():
     if 'current_images' not in st.session_state:
         st.session_state.current_images = []
 
-def main():
-    st.set_page_config(
-        page_title="Stock Chart Assistant",
-        layout="wide",
-        initial_sidebar_state="expanded"
-    )
-    
-    # Initialize services
-    try:
-        claude_service = ClaudeService()
-        analysis_service = ChartAnalysisService(claude_service)
-    except ValueError as e:
-        st.error(str(e))
-        return
-    
-    # Initialize session state
-    init_session_state()
+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()
@@ -87,7 +76,7 @@ def main():
                                 st.subheader(f"{result['analysis_type']} Results")
                                 st.write(result['analysis'])
                                 # Add follow-up section immediately
-                                follow_up = show_follow_up_section("comparative_followup")
+                                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(
@@ -122,20 +111,6 @@ def main():
                             "⚠️ This analysis is AI-generated and for informational purposes only. "
                             "Do not make trading decisions solely based on this information."
                         )
-        
-        # Follow-up section
-        if st.session_state.current_analysis:
-            follow_up = show_follow_up_section(st.session_state.current_analysis)
-            if follow_up:
-                with st.spinner("Processing follow-up..."):
-                    response = analysis_service.handle_follow_up_question(
-                        follow_up,
-                        st.session_state.current_analysis,
-                        st.session_state.current_images[0] if st.session_state.current_images else None
-                    )
-                    if response:
-                        st.session_state.chat_history.append(response)
-                        st.write(response['analysis'])
     
     with col2:
         # Chat history
@@ -154,5 +129,48 @@ def main():
             else:
                 st.info("Chat saved to session state")
 
+def show_learning_tab(learning_module):
+    """Display learning center functionality"""
+    st.title("📚 Trading Learning Center")
+    
+    # Create tabs for different learning options
+    learn_tab1, learn_tab2 = st.tabs(["Structured Courses", "Custom Learning"])
+    
+    with learn_tab1:
+        st.header("Learn Trading Step by Step")
+        learning_module.display_course_selection()
+        
+    with learn_tab2:
+        st.header("Ask Any Trading Question")
+        learning_module.display_custom_learning()
+
+def main():
+    st.set_page_config(
+        page_title="Stock Chart Assistant",
+        layout="wide",
+        initial_sidebar_state="expanded"
+    )
+    
+    # Initialize services
+    try:
+        claude_service = ClaudeService()
+        analysis_service = ChartAnalysisService(claude_service)
+        learning_module = LearningModule(claude_service)
+    except ValueError as e:
+        st.error(str(e))
+        return
+    
+    # Initialize session state
+    init_session_state()
+    
+    # Main application tabs
+    tab1, tab2 = st.tabs(["Chart Analysis", "Learning Center"])
+    
+    with tab1:
+        show_chart_analysis_tab(claude_service, analysis_service)
+        
+    with tab2:
+        show_learning_tab(learning_module)
+
 if __name__ == "__main__":
     main()