Update app.py

app.py

@@ -80,29 +80,6 @@ elif input_option == "Upload CSV File":
         except Exception as e:
             st.error(f"Error loading file: {e}")
 
-def save_as_txt(content, filename):
-    with open(filename, "w") as f:
-        f.write(content)
-    return filename
-
-def save_as_pdf(content, filename):
-    from fpdf import FPDF
-    pdf = FPDF()
-    pdf.add_page()
-    pdf.set_font("Arial", size=12)
-    for line in content.split('\n'):
-        pdf.multi_cell(0, 10, line)
-    pdf.output(filename)
-    return filename
-
-# Show Dataset Preview Only After Loading
-if st.session_state.df is not None and st.session_state.show_preview:
-    st.subheader("📂 Dataset Preview")
-    st.dataframe(st.session_state.df.head())
-
-import tempfile
-from fpdf import FPDF
-
 # Helper Functions for Download
 def save_as_txt(content, filename):
     with open(filename, "w") as f:
@@ -171,7 +148,7 @@ if st.session_state.df is not None:
 
     conclusion_writer = Agent(
         role="Conclusion Specialist",
-        goal="Summarize findings into a clear and concise 3-5 line Conclusion highlighting only the most important aspects.",
+        goal="Summarize findings into a clear and concise 3-5 line Conclusion highlighting only the most important insights.",
         backstory="An expert in crafting impactful and clear conclusions.",
         llm=llm,
     )
@@ -227,26 +204,27 @@ if st.session_state.df is not None:
         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..."):
-                report_inputs = {"query": query + " Provide detailed analysis but DO NOT include Conclusion."}
-                report_result = crew_report.kickoff(inputs=report_inputs)
+                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."})
 
-                conclusion_inputs = {"query": query + " Provide ONLY the most important insights in 3-5 concise lines."}
-                conclusion_result = crew_conclusion.kickoff(inputs=conclusion_inputs)
+                # Convert results to string
+                report_text = str(report_result)
+                conclusion_text = str(conclusion_result)
 
-                st.markdown(report_result if report_result else "⚠️ No Report Generated.")
+                st.markdown(report_text if report_text else "⚠️ No Report Generated.")
 
-                if report_result:
-                    tab1_txt = save_as_txt(report_result, "Tab1_Report.txt")
-                    tab1_pdf = save_as_pdf(report_result, "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")
+                # 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.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")
+                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.")
 
@@ -254,7 +232,7 @@ if st.session_state.df is not None:
                 st.plotly_chart(fig_employment)
                 st.caption("📊 Salary distribution across employment types.")
 
-                st.markdown(conclusion_result if conclusion_result else "⚠️ No Conclusion Generated.")
+                st.markdown(conclusion_text if conclusion_text else "⚠️ No Conclusion Generated.")
 
     # Full Data Visualization Tab
     with tab2:
@@ -271,15 +249,12 @@ if st.session_state.df is not None:
 
         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.caption("📊 Salary distribution for each employment type.")
-
-        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"
+        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"
         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")