Update app.py

app.py

@@ -170,6 +170,7 @@ if st.session_state.df is not None:
             with st.spinner("Processing query..."):
                 inputs = {"query": query}
                 result = crew.kickoff(inputs=inputs)
+
                 st.markdown("### Analysis Report:")
 
                 # Collect all generated visualizations
@@ -182,39 +183,36 @@ if st.session_state.df is not None:
                     visualizations.append(fig_salary)
 
                 # Experience Level Visualization
-                if "experience" in query.lower():
-                    fig_experience = px.bar(
-                        st.session_state.df.groupby("experience_level")["salary_in_usd"].mean().reset_index(),
-                        x="experience_level", y="salary_in_usd",
-                        title="Average Salary by Experience Level"
-                    )
-                    visualizations.append(fig_experience)
+                fig_experience = px.bar(
+                    st.session_state.df.groupby("experience_level")["salary_in_usd"].mean().reset_index(),
+                    x="experience_level", y="salary_in_usd",
+                    title="Average Salary by Experience Level"
+                )
+                visualizations.append(fig_experience)
 
                 # Employment Type Visualization
-                if "employment" in query.lower():
-                    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)
+                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
+                # Insert Visual Insights before Conclusion
                 insert_section = "## Conclusion"
-                if insert_section in result and visualizations:
+                if insert_section in result:
                     parts = result.split(insert_section)
-                    st.markdown(parts[0])  # Content before Conclusion
+                    st.markdown(parts[0])  # Show content before Conclusion
 
-                    # Insert Visual Insights Section
+                    # Insert Visual Insights Section BEFORE Conclusion
                     st.markdown("## 📊 Visual Insights")
                     for fig in visualizations:
                         st.plotly_chart(fig, use_container_width=True)
 
-                    # Continue with Conclusion
+                    # Show the Conclusion after Visual Insights
                     st.markdown(insert_section + parts[1])
                 else:
                     st.markdown(result)
-                    if visualizations:
-                        st.markdown("## 📊 Visual Insights")
-                        for fig in visualizations:
-                            st.plotly_chart(fig, use_container_width=True)
+                    st.markdown("## 📊 Visual Insights")
+                    for fig in visualizations:
+                        st.plotly_chart(fig, use_container_width=True)
 
     # Tab 2: Full Data Visualization
     with tab2: