Update app.py

app.py

@@ -20,6 +20,8 @@ def init_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 = []
 
 def main():
     st.set_page_config(
@@ -40,10 +42,7 @@ def main():
     init_session_state()
     
     # Create sidebar and get inputs
-    upload_option, uploaded_file, patterns, indicators = create_sidebar()
-    
-    # Main content area
-    col1, col2 = st.columns([2, 1])
+    upload_option, uploaded_files, patterns, indicators, comparison_type = create_sidebar()
     
     # Main content area
     col1, col2 = st.columns([2, 1])
@@ -82,12 +81,12 @@ def main():
                         st.session_state.chat_history.extend(results)
                         
                         # Display analyses
-                        for result in results:
+                        for idx, result in enumerate(results):
                             analysis_type = result.get('analysis_type', 'Individual')
                             if analysis_type != 'Individual':
                                 st.subheader(f"{analysis_type} Results")
                             else:
-                                st.subheader(f"Analysis Results - Chart {len(st.session_state.chat_history)}")
+                                st.subheader(f"Analysis Results - Chart {idx + 1}")
                             st.write(result['analysis'])
                         
                         # Risk warning
@@ -104,7 +103,7 @@ def main():
                     response = analysis_service.handle_follow_up_question(
                         follow_up,
                         st.session_state.current_analysis,
-                        st.session_state.current_image
+                        st.session_state.current_images[0] if st.session_state.current_images else None
                     )
                     if response:
                         st.session_state.chat_history.append(response)
@@ -119,10 +118,13 @@ def main():
         if save_name and st.session_state.chat_history:
             filename = save_chat_history(
                 st.session_state.chat_history,
-                st.session_state.current_image,
+                st.session_state.current_images[0] if st.session_state.current_images else None,
                 f"{save_name}.json" if save_name else None
             )
-            st.success(f"Chat saved as {filename}")
+            if filename:
+                st.success(f"Chat saved as {filename}")
+            else:
+                st.info("Chat saved to session state")
 
 if __name__ == "__main__":
     main()