Update app.py

app.py

@@ -207,61 +207,53 @@ if st.session_state.df is not None:
                 report_result = crew_report.kickoff(inputs={"query": query + " Provide detailed analysis but DO NOT include Conclusion."})
                 conclusion_result = crew_conclusion.kickoff(inputs={"query": query + " Provide ONLY the most important insights in 3-5 concise lines."})
 
-                # Convert results to string
-                report_text = str(report_result)
-                conclusion_text = str(conclusion_result)
+                st.markdown(str(report_result) if report_result else "⚠️ No Report Generated.")
 
-                st.markdown(report_text if report_text else "⚠️ No Report Generated.")
-
-                # Download Buttons for Tab 1
-                tab1_txt = save_as_txt(report_text, "Tab1_Report.txt")
-                tab1_pdf = save_as_pdf(report_text, "Tab1_Report.pdf")
-                st.download_button("Download Tab 1 Report as TXT", open(tab1_txt, "rb"), file_name="Tab1_Report.txt")
-                st.download_button("Download Tab 1 Report as PDF", open(tab1_pdf, "rb"), file_name="Tab1_Report.pdf")
-
-                # Visualizations with captions
                 fig_salary = px.box(st.session_state.df, x="job_title", y="salary_in_usd", title="Salary Distribution by Job Title")
-                st.plotly_chart(fig_salary)
+                st.plotly_chart(fig_salary, use_container_width=True, key="fig_salary")
                 st.caption("📊 Salary distribution across different job titles.")
 
-                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")
-                st.plotly_chart(fig_experience)
-                st.caption("📊 Average salary based on experience level.")
+                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")
+                st.plotly_chart(fig_experience, use_container_width=True, key="fig_experience")
+                st.caption("📊 Average salary by experience level.")
 
                 fig_employment = px.box(st.session_state.df, x="employment_type", y="salary_in_usd", title="Salary Distribution by Employment Type")
-                st.plotly_chart(fig_employment)
+                st.plotly_chart(fig_employment, use_container_width=True, key="fig_employment")
                 st.caption("📊 Salary distribution across employment types.")
 
-                st.markdown(conclusion_text if conclusion_text else "⚠️ No Conclusion Generated.")
-
     # Full Data Visualization Tab
     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)
+        st.plotly_chart(fig1, key="fig1")
         st.caption("📊 Frequency of each job title in the dataset.")
 
         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)
+        st.plotly_chart(fig2, key="fig2")
         st.caption("📊 Average salary for each experience level.")
 
         fig3 = px.box(st.session_state.df, x="employment_type", y="salary_in_usd", title="Salary Distribution by Employment Type")
-        st.plotly_chart(fig3)
+        st.plotly_chart(fig3, key="fig3")
         st.caption("📊 Salary distribution across employment types.")
 
-        tab2_content = "Comprehensive Data Visualizations:\n- Job Title Frequency\n- Average Salary by Experience Level\n- Salary Distribution by Employment Type\n"
+        # Restored Summary for Tab 2
+        tab2_content = "Comprehensive Data Visualizations:\n"
+        tab2_content += "- Job Title Frequency\n"
+        tab2_content += "- Average Salary by Experience Level\n"
+        tab2_content += "- Salary Distribution by Employment Type\n"
+
         tab2_txt = save_as_txt(tab2_content, "Tab2_Visualizations.txt")
         tab2_pdf = save_as_pdf(tab2_content, "Tab2_Visualizations.pdf")
-
-        st.download_button("Download Tab 2 Summary as TXT", open(tab2_txt, "rb"), file_name="Tab2_Visualizations.txt")
-        st.download_button("Download Tab 2 Summary as PDF", open(tab2_pdf, "rb"), file_name="Tab2_Visualizations.pdf")
+        st.download_button("📥 Download Tab 2 Summary as TXT", open(tab2_txt, "rb"), file_name="Tab2_Visualizations.txt")
+        st.download_button("📥 Download Tab 2 Summary as PDF", open(tab2_pdf, "rb"), file_name="Tab2_Visualizations.pdf")
 
     temp_dir.cleanup()
 else:
     st.info("Please load a dataset to proceed.")
-    
+
 # Sidebar Reference
 with st.sidebar:
     st.header("📚 Reference:")