Update app.py

app.py

@@ -36,8 +36,16 @@ if "rfp_summary" not in st.session_state:
     st.session_state["rfp_summary"] = ""
 if "estimation_data" not in st.session_state:
     st.session_state["estimation_data"] = []
+if "estimation_data_json" not in st.session_state:
+    st.session_state["estimation_data_json"] = {}
 if "is_estimation_data_created" not in st.session_state:
     st.session_state["is_estimation_data_created"] = False
+if "roadmap_data" not in st.session_state:
+    st.session_state["roadmap_data"] = []
+if "roadmap_data_json" not in st.session_state:
+    st.session_state["roadmap_data_json"] = []
+if "is_roadmap_data_created" not in st.session_state:
+    st.session_state["is_roadmap_data_created"] = False
     
 
 def main():
@@ -49,7 +57,7 @@ def main():
     with st.sidebar:
         menu_choice = option_menu(
             menu_title="RFPStoryCraft",
-            options=["Home", "RFP Bot", "User Stories", "Summary", "Estimations"],
+            options=["Home", "RFP Bot", "User Stories", "Summary", "Estimations","RoadMap"],
             icons=["house", "list-task", "book", "book", "list-task"],
         )
 
@@ -157,6 +165,42 @@ def main():
         else:
             st.warning("Plesase Process RFP Details to access this feature", icon="⚠️")
 
+    if menu_choice == "RoadMap":
+        if st.session_state["is_data_processed"] == True:
+            if st.session_state["is_estimation_data_created"] == True:
+                st.title("RoadMap")
+                st.button(
+                    "Generate RoadMap", on_click=lambda: function.generate_roadmap()
+                )
+
+                if st.session_state["is_roadmap_data_created"] == True:
+                    st.button(
+                        "Export RoadMap",
+                        on_click=lambda: export(st.session_state["roadmap_data"]),
+                    )
+                    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)
+
+                    with st.container():
+                        df = pd.DataFrame(st.session_state["roadmap_data"])
+                        st.table(df)
+
+            else:
+                st.warning(
+                    "Please Process the Estimations Data to access this feature",
+                    icon="⚠️",
+                )
+        else:
+            st.warning("Plesase Process RFP Details to access this feature", icon="⚠️")
+            
 
 if __name__ == "__main__":
     # if localStorage.getItem("email"):