Update app.py

app.py

@@ -86,12 +86,36 @@ def main():
                             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("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(