Update app.py

app.py

@@ -4,7 +4,8 @@ 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
+from io import BytesIO
+import textwrap
 
 # Load environment variables
 load_dotenv()
@@ -56,22 +57,42 @@ if generate_blog:
             # Define tasks
             research_task = Task(
                 description=(
-                    "Identify the News Article and get detailed information about the News from the website."
+                    "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"
                 ),
-                expected_output='A comprehensive 3-paragraph-long report based on the {topic} of News.',
-                tools=[web_tool],
-                agent=blog_researcher,
+                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
             )
 
             write_task = Task(
                 description=(
-                    "Get the info from the News Website on the topic {topic}."
+                    "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"
                 ),
-                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
+                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
             )
 
             # Create crew
@@ -79,28 +100,19 @@ if generate_blog:
                 agents=[blog_researcher, blog_writer],
                 tasks=[research_task, write_task],
                 process=Process.sequential,
-                memory=True,
-                cache=True,
-                max_rpm=100,
-                share_crew=True
+                verbose=2
             )
 
-            # Kickoff the process and fetch the result
             try:
-                result = crew.kickoff(inputs={'topic': news_link})
-
-                # Access task outputs
-                try:
-                    task_outputs = result.tasks_output
-                except AttributeError:
-                    st.error("The result object does not have 'tasks_output'.")
-                    task_outputs = []
-
-                # Extract blog content
-                try:
-                    blog_content = task_outputs[1].raw
-                except (IndexError, AttributeError):
-                    blog_content = "Unable to fetch blog content from the task outputs."
+                # Kickoff the process
+                result = crew.kickoff()
+                
+                # Extract blog content from the result
+                if hasattr(result, 'task_output'):
+                    blog_content = result.task_output
+                else:
+                    # If task_output is not available, try to get the raw result
+                    blog_content = str(result)
 
                 # Display the blog content
                 st.subheader("Generated Blog")
@@ -113,18 +125,42 @@ if generate_blog:
 
                 if save_blog:
                     try:
-                        # Generate PDF in-memory
+                        # 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)
-                        pdf.multi_cell(0, 10, blog_content)
                         
-                        # Use BytesIO for in-memory file
+                        # 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)  # Reset buffer pointer
+                        pdf_buffer.seek(0)
                         
-                        # Provide download button
+                        # Create download button
                         st.download_button(
                             label="Download Blog as PDF",
                             data=pdf_buffer,
@@ -132,16 +168,18 @@ if generate_blog:
                             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}")
+                        st.error(f"Failed to save blog as PDF: {str(e)}")
+                        st.error("PDF Generation Error Details:", exc_info=True)
 
                 if reject_blog:
                     st.warning("Blog rejected. No file was saved.")
 
             except Exception as e:
-                st.error(f"An error occurred during the process: {e}")
+                st.error(f"An error occurred during the process: {str(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.")