Update app.py

app.py

@@ -6,10 +6,15 @@ import pandas as pd
 import base64
 from Functions import RFPProcessor
 # from localStoragePy import localStoragePy
-from Utils import export,clear_rfp_data, generate_roadmap_image
+from Utils import (
+    export,
+    clear_rfp_data,
+    get_binary_file_downloader_html,
+    generate_roadmap_image,
+)
 # from Login_and_email import *
-
-# localStorage = localStoragePy("RFP", "json")
+from Proposal import prop
+#localStorage = localStoragePy("RFP", "json")
 
 # Initialize session states
 if "generated" not in st.session_state:
@@ -46,7 +51,9 @@ 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
-    
+if "is_presentation_created" not in st.session_state:
+    st.session_state["is_presentation_created"] = False
+
 
 def main():
     function = RFPProcessor()
@@ -57,7 +64,15 @@ def main():
     with st.sidebar:
         menu_choice = option_menu(
             menu_title="RFPStoryCraft",
-            options=["Home", "RFP Bot", "User Stories", "Summary", "Estimations","RoadMap"],
+            options=[
+                "Home",
+                "RFP Bot",
+                "User Stories",
+                "Summary",
+                "Estimations",
+                "RoadMap",
+                "Proposal",
+            ],
             icons=["house", "list-task", "book", "book", "list-task"],
         )
 
@@ -75,7 +90,9 @@ def main():
                 type="default",
                 placeholder="Project Name",
             )
-            file = st.file_uploader("Document", type="pdf")
+            file = st.file_uploader(
+                "Document", type=["pdf", "txt", "docx"], accept_multiple_files=False
+            )
 
             submitted = st.form_submit_button("Process Data")
 
@@ -116,10 +133,13 @@ def main():
                 on_click=function.genrate_user_stories,
             )
             if st.session_state["is_user_stories_created"] == True:
-                st.button("Export Stories", on_click=lambda: export(st.session_state["user_stories_data"]))
+                st.button(
+                    "Export Stories",
+                    on_click=lambda: export(st.session_state["user_stories_data"]),
+                )
                 with st.container():
-                 df = pd.DataFrame(st.session_state["user_stories_data"])
-                 st.table(df)
+                    df = pd.DataFrame(st.session_state["user_stories_data"])
+                    st.table(df)
         else:
             st.warning("Plesase Process RFP Details to access this feature", icon="⚠️")
 
@@ -147,21 +167,35 @@ def main():
                     label="Number of Tech Leads",
                     placeholder="Enter here....",
                 )
-    
-                if senior_developers and junior_developers and tech_leads and st.session_state["is_user_stories_created"] == True:
+
+                if (
+                    senior_developers
+                    and junior_developers
+                    and tech_leads
+                    and st.session_state["is_user_stories_created"] == True
+                ):
                     st.button(
                         "Generate Estimations",
-                        on_click=lambda: function.generate_estimations(tech_leads, senior_developers, junior_developers),
+                        on_click=lambda: function.generate_estimations(
+                            tech_leads, senior_developers, junior_developers
+                        ),
                     )
-    
-                if st.session_state["is_estimation_data_created"] == True:
+
                     if st.session_state["is_estimation_data_created"] == True:
-                     st.button("Export Stories", on_click = lambda: export(st.session_state["estimation_data"]))
-                    with st.container():
-                        df = pd.DataFrame(st.session_state["estimation_data"])
-                        st.table(df)
+                        st.button(
+                            "Export Stories",
+                            on_click=lambda: export(
+                                st.session_state["estimation_data"]
+                            ),
+                        )
+
+                        with st.container():
+                            df = pd.DataFrame(st.session_state["estimation_data"])
+                            st.table(df)
             else:
-                st.warning("Plesase Process User Stories to access this feature", icon="⚠️")
+                st.warning(
+                    "Plesase Process User Stories to access this feature", icon="⚠️"
+                )
         else:
             st.warning("Plesase Process RFP Details to access this feature", icon="⚠️")
 
@@ -179,10 +213,10 @@ def main():
                         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
-                      generate_roadmap_image()
-                        
+                        st.info("Generating roadmap... Please wait.")
+                        # Generate the roadmap image
+                        generate_roadmap_image()
+
                     with st.container():
                         df = pd.DataFrame(st.session_state["roadmap_data"])
                         st.table(df)
@@ -194,10 +228,38 @@ def main():
                 )
         else:
             st.warning("Plesase Process RFP Details to access this feature", icon="⚠️")
-            
+
+    if menu_choice == "Proposal":
+        if st.session_state["is_data_processed"] == True:
+            if st.session_state["is_user_stories_created"] == True:
+                st.title("Proposal")
+                num_slides = st.number_input(
+                    "Enter the number of slides", min_value=1, max_value=20, value=None
+                )
+                if num_slides:
+                    st.button(
+                        "Generate Presentation", on_click=lambda: prop(num_slides)
+                    )
+
+                    if st.session_state["is_presentation_created"] == True:
+                        st.success("Presentation Created!", icon="✅")
+                        with open("Generated_Presentation.pptx", "rb") as pptx_file:
+                            pptx_bytes = pptx_file.read()
+                            st.download_button(
+                                label="Download Presentation",
+                                data=pptx_bytes,
+                                file_name="Generated_Presentation_from_code.pptx",
+                                mime="application/vnd.openxmlformats-officedocument.presentationml.presentation",
+                            )
+
+            else:
+                st.warning(
+                    "Please Process the User Stories 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"):
-    main()
-    # else:
-    #     authenticate(localStorage)
+        main()