Update app.py

app.py

@@ -181,14 +181,8 @@ def main():
                     if st.button("Generate Roadmap and Download"):
                       st.info("Generating roadmap... Please wait.")
                       # Generate the roadmap image
-                      roadmap_image_data = generate_roadmap_image()
-
-                      # Display the generated image
-                      st.image(roadmap_image_data, caption='Project Roadmap', use_column_width=True)
-
-                       # Provide a download link for the image
-                      st.markdown(get_binary_file_downloader_html(roadmap_image_data, 'project_roadmap.png', 'Download Image'), unsafe_allow_html=True)
-
+                      generate_roadmap_image()
+                        
                     with st.container():
                         df = pd.DataFrame(st.session_state["roadmap_data"])
                         st.table(df)