Update app.py

app.py

@@ -38,14 +38,11 @@ class ContentCrew:
 
         return result
 
+# Function to display the result in Streamlit
 def display_result(result):
     st.write("## Research on Topic")
     st.write('-------------------------------')
 
-    # Debugging: Print the result to ensure it has content
-    st.write("DEBUG: Displaying result")
-    st.json(result)  # Streamlit's way to print JSON data
-
     if 'content_plan' in result:
         st.write("### Content Plan")
         st.write(result['content_plan'])
@@ -57,9 +54,11 @@ def display_result(result):
 
     if 'images' in result:
         st.write("### Images")
-        for idx, image in enumerate(result['images']):
-            image_path = f'image_{idx}.png'
-            st.image(image_path, caption=f"Figure {idx+1}: {image['description']}")
+        for image in result['images']:
+            st.image(image['url'], caption=image['description'])
+
+    if 'content' in result:
+        st.markdown(result['content'])
 
 
 # Streamlit UI