Update app.py

app.py

@@ -1,13 +1,14 @@
 import streamlit as st
-from crewai import Crew, Agent, Process, Task
-from tools import tool
+from crewai import Agent, Task, Crew
 from langchain_google_genai import ChatGoogleGenerativeAI
+from dotenv import load_dotenv
+import os
 
-# Define your API keys directly in the code
+# Load environment variables
+load_dotenv()
 GOOGLE_API_KEY = "AIzaSyDy02GtpV6-VcdeCZNLJT-c4kWuPxBRbrI"
-SERPER_API_KEY = "uxSCra7XOaiojmWo3MIQ2xzFWkX2306yJY7YIVMsT771qLCaOJejKKQB7nKbRXd7"
 
-# Initialize Google Generative AI model
+# Define LLM
 llm = ChatGoogleGenerativeAI(
     model="gemini-1.5-flash",
     verbose=True,
@@ -15,90 +16,112 @@ llm = ChatGoogleGenerativeAI(
     google_api_key=GOOGLE_API_KEY
 )
 
-# Initialize the tool for internet searching capabilities
-from crewai_tools import SerperDevTool
-tool = SerperDevTool(api_key=SERPER_API_KEY)
+# Define Agents
+planner = Agent(
+    role="Content Planner",
+    goal="Plan engaging and factually accurate content on {topic}",
+    backstory="You're working on planning a blog article "
+              "about the topic: {topic}."
+              "You collect information that helps the "
+              "audience learn something "
+              "and make informed decisions. "
+              "Your work is the basis for "
+              "the Content Writer to write an article on this topic.",
+    allow_delegation=False,
+    verbose=True
+)
 
-# Define agents
-news_researcher = Agent(
-    role="Senior Researcher",
-    goal='Uncover groundbreaking technologies in {topic}',
-    verbose=True,
-    memory=True,
-    backstory=(
-        "Driven by curiosity, you're at the forefront of"
-        "innovation, eager to explore and share knowledge that could change"
-        "the world."
-    ),
-    tools=[tool],
-    llm=llm,
-    allow_delegation=True
+writer = Agent(
+    role="Content Writer",
+    goal="Write a compelling and well-structured blog post on {topic}.",
+    backstory="You're a writer who uses the content plan to create a detailed blog post. "
+              "Ensure it aligns with SEO best practices and the brand's voice.",
+    allow_delegation=False,
+    verbose=True
 )
 
-news_writer = Agent(
-    role='Writer',
-    goal='Narrate compelling tech stories about {topic}',
-    verbose=True,
-    memory=True,
-    backstory=(
-        "With a flair for simplifying complex topics, you craft"
-        "engaging narratives that captivate and educate, bringing new"
-        "discoveries to light in an accessible manner."
-    ),
-    tools=[tool],
-    llm=llm,
-    allow_delegation=False
+editor = Agent(
+    role="Editor",
+    goal="Edit a given blog post to align with the writing style of the organization.",
+    backstory="You are an editor who receives a blog post "
+              "from the Content Writer. "
+              "Your goal is to review the blog post "
+              "to ensure that it follows journalistic best practices,"
+              "provides balanced viewpoints "
+              "when providing opinions or assertions, "
+              "and also avoids major controversial topics "
+              "or opinions when possible.",
+    allow_delegation=False,
+    verbose=True
 )
 
-# Define tasks
-research_task = Task(
+# Define Tasks
+plan = Task(
     description=(
-        "Identify the next big trend in {topic}."
-        "Focus on identifying pros and cons and the overall narrative."
-        "Your final report should clearly articulate the key points,"
-        "its market opportunities, and potential risks."
+        "1. Prioritize the latest trends, key players, "
+        "and noteworthy news on {topic}.\n"
+        "2. Identify the target audience, considering "
+        "their interests and pain points.\n"
+        "3. Develop a detailed content outline including "
+        "an introduction, key points, and a call to action.\n"
+        "4. Include SEO keywords and relevant data or sources."
     ),
-    expected_output='A comprehensive 3 paragraphs long report on the latest AI trends.',
-    tools=[tool],
-    agent=news_researcher,
+    expected_output="A comprehensive content plan document "
+                    "with an outline, audience analysis, "
+                    "SEO keywords, and resources.",
+    agent=planner,
 )
 
-write_task = Task(
+write = Task(
     description=(
-        "Compose an insightful article on {topic}."
-        "Focus on the latest trends and how it's impacting the industry."
-        "This article should be easy to understand, engaging, and positive."
+        "1. Use the content plan to craft a compelling "
+        "blog post on {topic}.\n"
+        "2. Incorporate SEO keywords naturally.\n"
+        "3. Sections/Subtitles are properly named "
+        "in an engaging manner.\n"
+        "4. Ensure the post is structured with an "
+        "engaging introduction, insightful body, "
+        "and a summarizing conclusion.\n"
+        "5. Proofread for grammatical errors and "
+        "alignment with the brand's voice.\n"
     ),
-    expected_output='A 4 paragraph article on {topic} advancements formatted as markdown.',
-    tools=[tool],
-    agent=news_writer,
-    async_execution=False,
-    output_file='new-blog-post.md'  # Example of output customization
+    expected_output="A well-written blog post "
+                    "in markdown format, ready for publication, "
+                    "each section should have 2 or 3 paragraphs.",
+    agent=writer,
+)
+
+edit = Task(
+    description=("Proofread the given blog post for "
+                 "grammatical errors and "
+                 "alignment with the brand's voice."),
+    expected_output="A well-written blog post in markdown format, "
+                    "ready for publication, "
+                    "each section should have 2 or 3 paragraphs.",
+    agent=editor
 )
 
 # Define Crew
 crew = Crew(
-    agents=[news_researcher, news_writer],
-    tasks=[research_task, write_task],
-    process=Process.sequential,
+    agents=[planner, writer, editor],
+    tasks=[plan, write, edit],
+    verbose=2
 )
 
-# Streamlit UI
-st.set_page_config(
-    page_title="Tech Research and Writing",
-    page_icon="🔬",
-    layout="wide",
-    menu_items={"About": "# Made by Prathamesh Khade"}
-)
+# Streamlit App
+def main():
+    st.title("Content Creation Assistant")
 
-st.title("Tech Research and Writing")
+    # Input for topic
+    topic = st.text_input("Enter the topic for the blog post:")
 
-topic = st.text_input("Enter a topic to research and write about:")
+    if st.button("Generate Content"):
+        if topic:
+            with st.spinner('Generating content...'):
+                result = crew.kickoff(inputs={"topic": topic})
+                st.markdown(result)
+        else:
+            st.error("Please enter a topic.")
 
-if st.button("Start Task"):
-    if topic:
-        result = crew.kickoff(inputs={'topic': topic})
-        st.write("Task Results:")
-        st.write(result)
-    else:
-        st.error("Please enter a topic.")
+if __name__ == "__main__":
+    main()