|
import streamlit as st |
|
from src.components.process import all_crew |
|
from dotenv import load_dotenv |
|
load_dotenv() |
|
|
|
|
|
class ContentGenerationUI: |
|
|
|
def generate_content(self, topic): |
|
inputs = { |
|
"topic": topic |
|
} |
|
result = all_crew.kickoff(inputs=inputs) |
|
return result |
|
|
|
def content_generation(self): |
|
if st.session_state.generating: |
|
st.session_state.content_generator = self.generate_content( |
|
st.session_state.topic) |
|
|
|
if st.session_state.content_generator and st.session_state.content_generator != "": |
|
with st.container(): |
|
st.write("Content generated successfully!") |
|
st.download_button( |
|
label="Download Markdown file", |
|
data=st.session_state.content_generator, |
|
file_name="content_generate.md", |
|
mime="text/markdown" |
|
) |
|
st.session_state.generating = False |
|
|
|
def sidebar(self): |
|
with st.sidebar: |
|
st.title("Content Generator Tool") |
|
st.write("This workflow should autonomously process input topics, conduct research, plan content, generate images, optimize for SEO, and perform final editorial checks.") |
|
|
|
topic = st.text_area( |
|
label="Enter the topic: ", |
|
placeholder="Enter the name of the topic to generate the article", |
|
key="topic", |
|
height=175 |
|
) |
|
st.write(f'You wrote {len(topic.split())} words.') |
|
|
|
if st.button("Generate Content"): |
|
st.session_state.generating = True |
|
|
|
def render(self): |
|
st.set_page_config(page_title="Content Generation", page_icon="📃") |
|
tab1, tab2 = st.tabs(["Output", "About Project"]) |
|
|
|
|
|
self.sidebar() |
|
|
|
with tab2: |
|
st.markdown("# About project") |
|
st.write( |
|
""" |
|
### Project Description |
|
Implementing a scalable content team using AI involves creating a framework that blends the strengths of AI technologies with the creative and supervisory capabilities of human team members. This strategy aims to enhance efficiency, creativity, and content output quality. |
|
|
|
This code is a high-level conceptualization and would require adaptation to fit the actual CrewAI framework and toolset specifics. It illustrates how different AI agents, equipped with specialized roles and tools, can collaborate within a content creation process. Each agent focuses on a key area—research, writing, and SEO—streamlining the content development workflow and enhancing output quality through specialized AI-driven tasks. |
|
|
|
### Objective |
|
Implement a content generation workflow using the Crew AI framework. This workflow should autonomously process input topics, conduct research, plan content, generate images, optimize for SEO, and perform final editorial checks. |
|
|
|
### Tools and Frameworks: |
|
* Crew AI framework |
|
* Streamlit - User Interface(UI) |
|
* Python for scripting |
|
* AI models or APIs (e.g., `gemini-pro` for content, `stable-diffusion-xl-base` for images) |
|
|
|
### Prerequisites |
|
To complete this project, you should understand Python programming, data manipulation, visualization libraries such as Pandas and Matplotlib, and machine learning libraries such as Scikit-Learn. Additionally, some background knowledge of natural language processing (NLP) techniques and generate text to image and image to text methods would be helpful. |
|
|
|
### Resources |
|
- Live demo link: [Content Generate - WebApp]() |
|
- Check out [CrewAI](https://docs.crewai.com/) |
|
- Project code [GitHub](https://github.com/Bhavik-Jikadara/Content-Generation-Workflow) |
|
""" |
|
) |
|
|
|
with tab1: |
|
|
|
if "topic" not in st.session_state: |
|
st.session_state.topic = "" |
|
|
|
if "content_generator" not in st.session_state: |
|
st.session_state.content_generator = "" |
|
|
|
if "generating" not in st.session_state: |
|
st.session_state.generating = False |
|
|
|
self.content_generation() |
|
|
|
|
|
if __name__ == "__main__": |
|
ContentGenerationUI().render() |
|
|