Update app.py

app.py

@@ -71,15 +71,32 @@ topic = st.text_input("What topic do you want to write about?")
 
 if st.button('Generate Content'):
     if topic:
-        content_crew = ContentCrew(topic)
-        result = content_crew.run()
-        
-        # Debugging: Print the result to ensure it is not empty
-        print("DEBUG: Result received")
-        print(result)
+        # 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()
+
         st.write("## Here is your Content Plan")
         st.write('-------------------------------')
-        st.write(result)
+        st.write(content_plan)
+
+        st.write("## Research Content")
+        st.write('-------------------------------')
+        st.write(research_content)
+
+        st.write("## Final Article")
+        st.write('-------------------------------')
+        st.write(final_article)
+        
+        # Generate and display images
+        sections = ["Introduction", "Application", "Definition and Architecture", "How it Works", "Comparison with Other Similar Apps", "References"]
+        for section in sections:
+            image_path = content_agents.generate_image_for_section(section, topic)
+            st.image(image_path, caption=f'{section} Image')
     else:
-        st.error("Please enter a topic.")
+        st.error("Please enter a topic.")