Update app.py

app.py

@@ -173,16 +173,13 @@ if st.session_state.df is not None:
 
                 st.markdown("### Analysis Report:")
 
-                # Collect all generated visualizations
+                # Generate relevant visualizations
                 visualizations = []
 
-                # Salary Visualization
-                if "salary" in query.lower():
-                    fig_salary = px.box(st.session_state.df, x="job_title", y="salary_in_usd",
-                                        title="Salary Distribution by Job Title")
-                    visualizations.append(fig_salary)
+                fig_salary = px.box(st.session_state.df, x="job_title", y="salary_in_usd",
+                                    title="Salary Distribution by Job Title")
+                visualizations.append(fig_salary)
 
-                # Experience Level Visualization
                 fig_experience = px.bar(
                     st.session_state.df.groupby("experience_level")["salary_in_usd"].mean().reset_index(),
                     x="experience_level", y="salary_in_usd",
@@ -190,25 +187,25 @@ if st.session_state.df is not None:
                 )
                 visualizations.append(fig_experience)
 
-                # Employment Type Visualization
                 fig_employment = px.box(st.session_state.df, x="employment_type", y="salary_in_usd",
                                         title="Salary Distribution by Employment Type")
                 visualizations.append(fig_employment)
 
-                # Insert Visual Insights before Conclusion
+                # Split the result and insert visualizations before Conclusion
                 insert_section = "## Conclusion"
                 if insert_section in result:
-                    parts = result.split(insert_section)
-                    st.markdown(parts[0])  # Show content before Conclusion
+                    parts = result.split(insert_section, 1)
+                    st.markdown(parts[0])  # Before Conclusion
 
-                    # Insert Visual Insights Section BEFORE Conclusion
+                    # Insert Visual Insights Section before Conclusion
                     st.markdown("## 📊 Visual Insights")
                     for fig in visualizations:
                         st.plotly_chart(fig, use_container_width=True)
 
-                    # Show the Conclusion after Visual Insights
+                    # Show Conclusion
                     st.markdown(insert_section + parts[1])
                 else:
+                    # Default if Conclusion not found
                     st.markdown(result)
                     st.markdown("## 📊 Visual Insights")
                     for fig in visualizations: