Update app.py

app.py

@@ -4,6 +4,7 @@ from crewai_tools import WebsiteSearchTool
 from dotenv import load_dotenv
 import os
 from fpdf import FPDF
+from io import BytesIO  # To handle in-memory files
 
 # Load environment variables
 load_dotenv()
@@ -88,10 +89,6 @@ if generate_blog:
             try:
                 result = crew.kickoff(inputs={'topic': news_link})
 
-                # Inspect result attributes for debugging
-                st.subheader("Result Attributes")
-                st.write(dir(result))
-
                 # Access task outputs
                 try:
                     task_outputs = result.tasks_output
@@ -99,12 +96,6 @@ if generate_blog:
                     st.error("The result object does not have 'tasks_output'.")
                     task_outputs = []
 
-                # Display task outputs
-                st.subheader("Task Outputs")
-                for idx, task_output in enumerate(task_outputs):
-                    st.write(f"Task {idx + 1}:")
-                    st.json(task_output)
-
                 # Extract blog content
                 try:
                     blog_content = task_outputs[1].raw
@@ -122,19 +113,24 @@ if generate_blog:
 
                 if save_blog:
                     try:
+                        # Generate PDF in-memory
                         pdf = FPDF()
                         pdf.add_page()
                         pdf.set_font("Arial", size=12)
                         pdf.multi_cell(0, 10, blog_content)
-                        pdf.output("generated_blog.pdf")
-
-                        with open("generated_blog.pdf", "rb") as pdf_file:
-                            st.download_button(
-                                label="Download Blog as PDF",
-                                data=pdf_file,
-                                file_name="generated_blog.pdf",
-                                mime="application/pdf"
-                            )
+                        
+                        # Use BytesIO for in-memory file
+                        pdf_buffer = BytesIO()
+                        pdf.output(pdf_buffer)
+                        pdf_buffer.seek(0)  # Reset buffer pointer
+                        
+                        # Provide download button
+                        st.download_button(
+                            label="Download Blog as PDF",
+                            data=pdf_buffer,
+                            file_name="generated_blog.pdf",
+                            mime="application/pdf"
+                        )
                         st.success("Blog saved as PDF and ready for download!")
                     except Exception as e:
                         st.error(f"Failed to save blog as PDF: {e}")