Update app.py

app.py

@@ -125,33 +125,35 @@ def main():
 
     if menu_choice == "Estimations":
         if st.session_state["is_data_processed"] == True:
-            st.title("Estimations")
-            senior_developers = st.text_input(
-                label="Number of Senior Developers",
-                placeholder="Enter here....",
-            )
-            junior_developers = st.text_input(
-                label="Number of Junior Developers",
-                placeholder="Enter here...",
-            )
-            tech_leads = st.text_input(
-                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:
-                st.button(
-                    "Generate Estimations",
-                    on_click=lambda: function.generate_estimations(tech_leads, senior_developers, junior_developers),
+            if st.session_state["is_user_stories_created"] == True:
+                st.title("Estimations")
+                senior_developers = st.text_input(
+                    label="Number of Senior Developers",
+                    placeholder="Enter here....",
                 )
-
-            if st.session_state["is_estimation_data_created"] == True:
+                junior_developers = st.text_input(
+                    label="Number of Junior Developers",
+                    placeholder="Enter here...",
+                )
+                tech_leads = st.text_input(
+                    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:
+                    st.button(
+                        "Generate Estimations",
+                        on_click=lambda: function.generate_estimations(tech_leads, senior_developers, junior_developers),
+                    )
+    
                 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)
-
+                    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)
+            else:
+                st.warning("Plesase Process User Stories to access this feature", icon="⚠️")
         else:
             st.warning("Plesase Process RFP Details to access this feature", icon="⚠️")