Update app.py

app.py

@@ -71,36 +71,17 @@ topic = st.text_input("What topic do you want to write about?")
 
 if st.button('Generate Content'):
     if topic:
-        # Instantiate the ContentAgents class
-        content_agents = ContentAgents()
-        
-        # Run planner, researcher, and writer agents
-        planner = content_agents.planner_agent(topic)
-        researcher = content_agents.researcher_agent(topic)
-        writer = content_agents.writer_agent(topic)
-        
-        # Assuming you have some method to run the agents and get the content
-        content_plan = planner.run()
-        research_content = researcher.run()
-        final_article = writer.run()
-        
-        sections = ["Introduction", "Application", "Definition and Architecture", "How it Works", "Comparison with Other Similar Apps", "References"]
-        
+        # Instantiate ContentCrew and run
+        content_crew = ContentCrew(topic)
+        result = content_crew.run()
+
+        # Display the results
         st.write("## Here is your Content Plan")
         st.write('-------------------------------')
-        st.write(content_plan)
-
-        st.write("## Research Content")
+        st.write(result['Content Plan'])
         st.write('-------------------------------')
-        st.write(research_content)
-
-        st.write("## Final Article")
+        st.write(result['Research Content'])
         st.write('-------------------------------')
-        
-        for section in sections:
-            st.write(f"### {section}")
-            st.write(final_article[section])  # Assuming final_article is a dict with sections as keys
-            image_path = content_agents.generate_image_for_section(section, topic)
-            st.image(image_path, caption=f'{section} Image')
+        st.write(result['Final Article'])
     else:
         st.error("Please enter a topic.")