Update app.py

app.py

@@ -172,39 +172,57 @@ if st.session_state.df is not None:
                 result = crew.kickoff(inputs=inputs)
                 st.markdown("### Analysis Report:")
 
-                # Create visualization if the query is about salary
+                # Collect all generated visualizations
+                visualizations = []
+
+                # Salary Visualization
                 if "salary" in query.lower():
-                    fig = px.box(st.session_state.df, x="job_title", y="salary_in_usd",
-                                 title="Salary Distribution by Job Title")
-
-                    # Insert visualization after "4. Influence of Company Size" for better flow
-                    insert_section = "4. Influence of Company Size"
-                    if insert_section in result:
-                        parts = result.split(insert_section)
-                        st.markdown(parts[0])  # Show content before the section
-                        
-                        # Insert the visualization here
-                        st.markdown(f"## {insert_section}{parts[1].split('5.')[0]}")
+                    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
+                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)
+
+                # 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)
+
+                # Insert "Visual Insights" before Conclusion
+                insert_section = "## Conclusion"
+                if insert_section in result and visualizations:
+                    parts = result.split(insert_section)
+                    st.markdown(parts[0])  # Content before Conclusion
+
+                    # Insert Visual Insights Section
+                    st.markdown("## 📊 Visual Insights")
+                    for fig in visualizations:
                         st.plotly_chart(fig, use_container_width=True)
 
-                        # Continue with the rest of the report
-                        st.markdown("## 5." + parts[1].split("5.")[1])
-                    else:
-                        # Default behavior if the section isn't found
-                        st.markdown(result)
-                        st.plotly_chart(fig, use_container_width=True)
+                    # Continue with Conclusion
+                    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)
 
     # Tab 2: Full Data Visualization
     with tab2:
         st.subheader("📊 Comprehensive Data Visualizations")
 
-        # Histogram of job titles
         fig1 = px.histogram(st.session_state.df, x="job_title", title="Job Title Frequency")
         st.plotly_chart(fig1)
 
-        # Bar chart of average salary by experience level
         fig2 = px.bar(
             st.session_state.df.groupby("experience_level")["salary_in_usd"].mean().reset_index(),
             x="experience_level", y="salary_in_usd",
@@ -212,18 +230,15 @@ if st.session_state.df is not None:
         )
         st.plotly_chart(fig2)
 
-        # Salary by Employment Type
         fig3 = px.box(st.session_state.df, x="employment_type", y="salary_in_usd",
                       title="Salary Distribution by Employment Type")
         st.plotly_chart(fig3)
 
-        # Salary by Company Size (if available)
         if "company_size" in st.session_state.df.columns:
             fig4 = px.box(st.session_state.df, x="company_size", y="salary_in_usd",
                           title="Salary Distribution by Company Size")
             st.plotly_chart(fig4)
 
-        # Salary by Region (if available)
         if "region" in st.session_state.df.columns:
             fig5 = px.box(st.session_state.df, x="region", y="salary_in_usd",
                           title="Salary Distribution by Region")