Create auth/auth_manager.py

auth/auth_manager.py

@@ -0,0 +1,296 @@
+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)
+        
+        # Store analysis results in session state
+        if 'analysis_results' not in st.session_state:
+            st.session_state.analysis_results = []
+            
+        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:
+                        st.session_state.current_analysis = results[-1]['analysis']
+                        st.session_state.chat_history.extend(results)
+                        st.session_state.analysis_results = results
+
+        # Display analysis results from session state
+        if st.session_state.analysis_results:
+            results = st.session_state.analysis_results
+            
+            # First show comparative analysis if it exists
+            for result in 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")
+                        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.markdown(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):
+                with st.container():
+                    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)
+                    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.markdown(response['analysis'])
+            
+            # Risk warning at the bottom
+            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")# 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"""
+    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()