Update app.py

app.py

@@ -163,23 +163,54 @@ if st.session_state.df is not None:
     # UI: Tabs for Query Results and General Insights
     tab1, tab2 = st.tabs(["🔍 Query Insights + Viz", "📊 Full Data Viz"])
 
+    # Tab 1: Query Insights + Visualization
     with tab1:
         query = st.text_area("Enter Query:", value="Provide insights into the salary of a Principal Data Scientist.")
-        
+        if st.button("Submit Query"):
+            with st.spinner("Processing query..."):
+                inputs = {"query": query}
+                result = crew.kickoff(inputs=inputs)
+                st.markdown("### Analysis Report:")
+
+                # Create visualization if the query is about salary
+                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 "5. Company Size"
+                    insert_section = "5. Company Size"
+                    if insert_section in result:
+                        # Split the report at the "Company Size" section
+                        parts = result.split(insert_section)
+                        st.markdown(parts[0])  # Display everything before "Company Size"
+                        st.markdown(f"## {insert_section}{parts[1].split('6.')[0]}")  # Show the "Company Size" content
+                        
+                        # Insert the visualization here
+                        st.plotly_chart(fig, use_container_width=True)
+
+                        # Continue with the rest of the report
+                        st.markdown("## 6." + parts[1].split("6.")[1])  # Display everything after "Company Size"
+                    else:
+                        # If "Company Size" not found, show full report and plot at the end
+                        st.markdown(result)
+                        st.plotly_chart(fig, use_container_width=True)
+                else:
+                    st.markdown(result)
+
+    # Tab 2: Full Data Visualization
     with tab2:
         st.subheader("📊 Comprehensive Data Visualizations")
-        
+
         fig1 = px.histogram(st.session_state.df, x="job_title", title="Job Title Frequency")
         st.plotly_chart(fig1)
 
-        fig2 = 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")
+        fig2 = 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"
+        )
         st.plotly_chart(fig2)
 
     temp_dir.cleanup()
 else:
     st.info("Please load a dataset to proceed.")
-
-with st.sidebar:
-    st.header("📚 Reference:")
-    st.markdown("[SQL Agents w CrewAI & Llama 3 - Plaban Nayak](https://github.com/plaban1981/Agents/blob/main/SQL_Agents_with_CrewAI_and_Llama_3.ipynb)")