Update app.py

app.py

@@ -3,7 +3,6 @@ from crewai import Agent, Task, Crew, Process
 from crewai_tools import WebsiteSearchTool
 from dotenv import load_dotenv
 import os
-from io import BytesIO  # To handle in-memory files
 
 # Load environment variables
 load_dotenv()
@@ -97,10 +96,6 @@ if generate_blog:
                 with st.spinner("Processing tasks..."):
                     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
@@ -108,12 +103,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
@@ -124,31 +113,20 @@ if generate_blog:
                 st.subheader("Generated Blog")
                 st.text_area("Blog Content", value=blog_content, height=400)
 
-                # Actions to save or reject the blog
+                # Actions to save the blog
                 st.subheader("Actions")
-                save_blog = st.button("Save Blog as Markdown")
-                reject_blog = st.button("Reject Blog")
+                save_blog = st.button("Save Blog Locally")
 
                 if save_blog:
                     try:
-                        # Save Markdown content in-memory
-                        markdown_buffer = BytesIO()
-                        markdown_buffer.write(blog_content.encode('utf-8'))
-                        markdown_buffer.seek(0)
-
-                        # Provide download button
-                        st.download_button(
-                            label="Download Blog as Markdown",
-                            data=markdown_buffer,
-                            file_name="generated_blog.md",
-                            mime="text/markdown"
-                        )
-                        st.success("Blog saved as Markdown and ready for download!")
-                    except Exception as e:
-                        st.error(f"Failed to save blog as Markdown: {e}")
+                        # Save Markdown content to a local file
+                        file_path = "generated_blog.md"
+                        with open(file_path, "w") as markdown_file:
+                            markdown_file.write(blog_content)
 
-                if reject_blog:
-                    st.warning("Blog rejected. No file was saved.")
+                        st.success(f"Blog saved successfully as '{file_path}' on your local machine!")
+                    except Exception as e:
+                        st.error(f"Failed to save blog locally: {e}")
 
             except Exception as e:
                 st.error(f"An error occurred during the process: {e}")