Spaces:
Build error
Build error
File size: 4,572 Bytes
92467f3 af729f8 92467f3 af729f8 92467f3 7911d4b 92467f3 7911d4b 92467f3 7911d4b 92467f3 7911d4b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 |
import os
import streamlit as st
from dotenv import load_dotenv
from crewai import Agent, Task, Crew, Process
from crewai_tools import SerperDevTool
from langchain_openai import ChatOpenAI
load_dotenv()
SERPER_API_KEY = os.getenv("SERPER_API_KEY")
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
llm = ChatOpenAI(api_key=OPENAI_API_KEY, model='gpt-4o-mini')
search_tool = SerperDevTool()
def write_social_media_post(user_input):
social_media_writer = Agent(
role='Social Media Post Writer',
goal='Create engaging social media posts.',
backstory="You specialize in crafting concise and compelling social media content.",
verbose=True,
allow_delegation=True,
tools=[search_tool],
llm=llm
)
social_media_task = Task(
description=f"Write posts for Twitter, Instagram, Facebook, and LinkedIn on the topic '{user_input}'. Tailor the posts to each platform's tone. Generate the posts in both Swedish and English.",
expected_output="Well-written posts for Twitter, Instagram, Facebook, and LinkedIn in both Swedish and English.",
agent=social_media_writer,
)
crew = Crew(
tasks=[social_media_task],
agents=[social_media_writer],
verbose=True,
process=Process.sequential,
)
result = crew.kickoff()
return result
def write_blog_post(user_input):
blog_post_writer = Agent(
role='Blog Writer',
goal='Generate informative and engaging blogs.',
backstory="You excel at writing detailed and engaging blog articles.",
verbose=True,
allow_delegation=True,
tools=[search_tool],
llm=llm
)
blog_writer_task = Task(
description=f"Write a blog article on the topic of '{user_input}' that includes sections such as introduction, body (with subheadings), conclusion, and sources cited at the end. Generate the blog in both Swedish and English.",
expected_output="A well-written blog article in both Swedish and English with proper headings and sources cited at the end.",
agent=blog_post_writer,
)
crew = Crew(
tasks=[blog_writer_task],
agents=[blog_post_writer],
verbose=True,
process=Process.sequential,
)
result = crew.kickoff()
return result
def write_blog_post_from_info(user_input):
blog_writer_from_info = Agent(
role='Blog Writer',
goal=f"Generate a blog based on the provided user informations: {user_input}",
backstory="You excel at writing detailed and engaging blog articles.",
verbose=True,
allow_delegation=True,
llm = llm
)
blog_reviewer = Agent(
role='Blog Reviewer',
goal="Review the blog post for quality, coherence, and accuracy.",
backstory="You have a keen eye for detail and ensure content is of high quality.",
verbose=True,
allow_delegation=True,
llm = llm
)
post_writer_from_info_task = Task(
description=f"Write a blog post in both Swedish and English based on provided user information: '{user_input}'. The blog must include sections such as introduction, body (with subheadings), conclusion.",
expected_output="A well-written blog post in both Swedish and English with proper headings.",
agent=blog_writer_from_info,
)
post_reviewer_task = Task(
description="Review the blog post for quality, and coherence.",
expected_output="A reviewed blog post with necessary corrections.",
agent=blog_reviewer,
)
crew = Crew(
tasks=[post_writer_from_info_task, post_reviewer_task],
agents=[blog_writer_from_info, blog_reviewer],
verbose=True,
process=Process.sequential,
)
result = crew.kickoff()
return result
st.title("Content Generator")
option = st.selectbox("Choose the content type you want to generate:",
("Select an option", "Social Media Agent", "Blog Article Agent", "Compose Agent"))
user_input = st.text_area("Enter the topic or information for your content:")
if st.button("Generate Content"):
if option == "Social Media Agent":
result = write_social_media_post(user_input)
st.markdown(result)
elif option == "Blog Article Agent":
result = write_blog_post(user_input)
st.markdown(result)
elif option == "Compose Agent":
result = write_blog_post_from_info(user_input)
st.markdown(result)
else:
st.write("Please select a valid option.") |