Update app.py

app.py

@@ -3,9 +3,7 @@ from crewai import Agent, Task, Crew, Process
 from crewai_tools import WebsiteSearchTool
 from dotenv import load_dotenv
 import os
-from fpdf import FPDF
-from io import BytesIO
-import textwrap
+from io import BytesIO  # To handle in-memory files
 
 # Load environment variables
 load_dotenv()
@@ -17,6 +15,14 @@ st.sidebar.header("Input")
 # OpenAI API key input
 user_api_key = st.sidebar.text_input("Enter your OpenAI API Key", type="password")
 news_link = st.sidebar.text_input("Enter News Link", "")
+
+# User inputs for agent configurations
+st.sidebar.subheader("Configure Agents")
+blog_researcher_role = st.sidebar.text_input("Researcher Role", "AI Blog Researcher from News Website")
+blog_researcher_goal = st.sidebar.text_area("Researcher Goal", "Get the relevant latest AI related news from News Website")
+blog_writer_role = st.sidebar.text_input("Writer Role", "Blog Writer")
+blog_writer_goal = st.sidebar.text_area("Writer Goal", "Narrate compelling tech stories about the News Article and add the reference links at the end of the blog.")
+
 generate_blog = st.sidebar.button("Generate Blog")
 
 if generate_blog:
@@ -32,8 +38,8 @@ if generate_blog:
 
             # Create agents
             blog_researcher = Agent(
-                role='AI Blog Researcher from News Website',
-                goal='Get the relevant latest AI related news from News Website',
+                role=blog_researcher_role,
+                goal=blog_researcher_goal,
                 verbose=True,
                 memory=True,
                 backstory=("Expert in understanding videos in AI, Data Science, Machine Learning, and GEN AI."),
@@ -42,8 +48,8 @@ if generate_blog:
             )
 
             blog_writer = Agent(
-                role='Blog Writer',
-                goal='Narrate compelling tech stories about the News Article and add the reference links at the end of the blog.',
+                role=blog_writer_role,
+                goal=blog_writer_goal,
                 verbose=True,
                 memory=True,
                 backstory=(
@@ -57,62 +63,62 @@ if generate_blog:
             # Define tasks
             research_task = Task(
                 description=(
-                    "Research Task: "
-                    "1. Visit the provided news link and analyze the content thoroughly\n"
-                    "2. Extract key information about the AI-related news or development\n"
-                    "3. Identify main points, technological aspects, and potential impact\n"
-                    "4. Gather any relevant background information or context\n"
-                    "5. Note any quotes, statistics, or specific technical details"
+                    "Identify the News Article and get detailed information about the News from the website."
                 ),
-                expected_output=(
-                    "A detailed research report containing:\n"
-                    "- Main story/development summary\n"
-                    "- Technical aspects and innovations\n"
-                    "- Impact and implications\n"
-                    "- Supporting facts and context\n"
-                    "- Notable quotes or statistics"
-                ),
-                agent=blog_researcher
+                expected_output='A comprehensive 3-paragraph-long report based on the {topic} of News.',
+                tools=[web_tool],
+                agent=blog_researcher,
             )
 
             write_task = Task(
                 description=(
-                    "Writing Task:\n"
-                    "1. Use the research findings to craft an engaging blog post\n"
-                    "2. Structure the content with clear introduction, body, and conclusion\n"
-                    "3. Explain technical concepts in an accessible manner\n"
-                    "4. Include relevant examples and real-world applications\n"
-                    "5. Add reference links at the end"
+                    "Get the info from the News Website on the topic {topic}."
                 ),
-                expected_output=(
-                    "A well-structured blog post that includes:\n"
-                    "- Engaging headline and introduction\n"
-                    "- Clear explanation of the AI development\n"
-                    "- Technical details explained simply\n"
-                    "- Impact and future implications\n"
-                    "- Reference links"
-                ),
-                agent=blog_writer
+                expected_output='Summarize the info from the News website on the topic {topic} and create the content for the blog.',
+                tools=[web_tool],
+                agent=blog_writer,
+                async_execution=False,
+                output_file=""  # Provide an empty string or valid file path
             )
 
-            # Create crew - Fixed verbose parameter to be boolean
+            # Create crew
             crew = Crew(
                 agents=[blog_researcher, blog_writer],
                 tasks=[research_task, write_task],
                 process=Process.sequential,
-                verbose=True  # Changed from 2 to True
+                memory=True,
+                cache=True,
+                max_rpm=100,
+                share_crew=True
             )
 
+            # Kickoff the process and fetch the result
             try:
-                # Kickoff the process with the news link as input
-                result = crew.kickoff(inputs={'topic': news_link})
-                
-                # Extract blog content from the result
-                if hasattr(result, 'tasks_output'):
-                    blog_content = result.tasks_output[-1].raw  # Get the last task's output
-                else:
-                    # If tasks_output is not available, try to get the raw result
-                    blog_content = str(result)
+                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
+                except AttributeError:
+                    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
+                except (IndexError, AttributeError):
+                    blog_content = "Unable to fetch blog content from the task outputs."
 
                 # Display the blog content
                 st.subheader("Generated Blog")
@@ -120,66 +126,34 @@ if generate_blog:
 
                 # Actions to save or reject the blog
                 st.subheader("Actions")
-                save_blog = st.button("Save Blog as PDF")
+                save_blog = st.button("Save Blog as Markdown")
                 reject_blog = st.button("Reject Blog")
 
                 if save_blog:
                     try:
-                        # Create PDF with proper text wrapping and formatting
-                        pdf = FPDF()
-                        pdf.add_page()
-                        pdf.set_auto_page_break(auto=True, margin=15)
-                        
-                        # Add title
-                        pdf.set_font("Arial", "B", 16)
-                        pdf.cell(0, 10, "Generated Blog", ln=True, align='C')
-                        pdf.ln(10)
-                        
-                        # Add content with proper formatting
-                        pdf.set_font("Arial", size=12)
-                        
-                        # Process and wrap text
-                        effective_page_width = pdf.w - 2 * pdf.l_margin
-                        
-                        # Split content into paragraphs
-                        paragraphs = blog_content.split('\n')
-                        
-                        for paragraph in paragraphs:
-                            if paragraph.strip():  # Only process non-empty paragraphs
-                                # Wrap text to fit page width
-                                wrapped_text = textwrap.fill(paragraph, width=95)
-                                
-                                # Split wrapped text into lines and write them
-                                lines = wrapped_text.split('\n')
-                                for line in lines:
-                                    pdf.multi_cell(0, 10, line)
-                                pdf.ln(5)  # Add small space between paragraphs
-                        
-                        # Generate PDF bytes
-                        pdf_buffer = BytesIO()
-                        pdf.output(pdf_buffer)
-                        pdf_buffer.seek(0)
-                        
-                        # Create download button
+                        # 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 PDF",
-                            data=pdf_buffer,
-                            file_name="generated_blog.pdf",
-                            mime="application/pdf"
+                            label="Download Blog as Markdown",
+                            data=markdown_buffer,
+                            file_name="generated_blog.md",
+                            mime="text/markdown"
                         )
-                        st.success("Blog saved as PDF and ready for download!")
-                        
+                        st.success("Blog saved as Markdown and ready for download!")
                     except Exception as e:
-                        st.error(f"Failed to save blog as PDF: {str(e)}")
-                        st.error("PDF Generation Error Details:", exc_info=True)
+                        st.error(f"Failed to save blog as Markdown: {e}")
 
                 if reject_blog:
                     st.warning("Blog rejected. No file was saved.")
 
             except Exception as e:
-                st.error(f"An error occurred during the process: {str(e)}")
+                st.error(f"An error occurred during the process: {e}")
 
         else:
             st.error("Please enter a valid news link.")
     else:
-        st.error("Please provide your OpenAI API Key.")
+        st.error("Please provide your OpenAI API Key.")