Update app.py

app.py

@@ -1,4 +1,119 @@
-# app.py
+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