Update app.py

app.py

@@ -4,17 +4,14 @@ from dotenv import load_dotenv
 from crewai import Agent, Task, Crew, Process
 from crewai_tools import SerperDevTool
 from langchain_openai import ChatOpenAI
-import markdown
-
-def convert_markdown_to_html(markdown_text):
-    return markdown.markdown(markdown_text)
+from langchain_core.output_parsers import StrOutputParser
 
 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')
-
+output_parser = StrOutputParser()
 search_tool = SerperDevTool()
 def write_social_media_post(user_input):
     social_media_writer = Agent(
@@ -53,16 +50,7 @@ def write_blog_post(user_input):
         allow_delegation=True,
         tools=[search_tool],
         llm=llm
-    )
-
-    blog_post_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
-    )
+    )    
 
     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.",
@@ -70,15 +58,9 @@ def write_blog_post(user_input):
         agent=blog_post_writer,
     )
 
-    blog_post_reviewer_task = Task(
-        description="Review the blog post for quality, and coherence.",
-        expected_output="A reviewed blog post with necessary corrections.",
-        agent=blog_post_reviewer,
-    )
-
     crew = Crew(
-        tasks=[blog_writer_task, blog_post_reviewer_task],
-        agents=[blog_post_writer, blog_post_reviewer],
+        tasks=[blog_writer_task],
+        agents=[blog_post_writer],
         verbose=True,
         process=Process.sequential,
     )
@@ -138,18 +120,15 @@ 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)
-        html_output = convert_markdown_to_html(result)
-        st.markdown(html_output, unsafe_allow_html=True)
-        #st.write(result)
+        output = output_parser.parse(result)
+        st.write(output)
     elif option == "Blog Article Agent":
         result = write_blog_post(user_input)
-        html_output = convert_markdown_to_html(result)
-        st.markdown(html_output, unsafe_allow_html=True)
-        #st.write(result)
+        output = output_parser.parse(result)
+        st.write(output)
     elif option == "Compose Agent":
         result = write_blog_post_from_info(user_input)
-        html_output = convert_markdown_to_html(result)
-        st.markdown(html_output, unsafe_allow_html=True)
-        #st.write(result)
+        output = output_parser.parse(result)
+        st.write(output)
     else:
         st.write("Please select a valid option.")