Update pages/Data_Collection.py

pages/Data_Collection.py

@@ -88,39 +88,23 @@ st.markdown(
     "</p>", 
     unsafe_allow_html=True
 )
+# Buttons for each stage
+st.markdown("### Select a Lifecycle Stage to Learn More:")
+col1 = st.columns(2)
 
-# Add buttons for each data type
-if st.button("Learn More About Structured Data"):
-    st.markdown(
-        "<p style='font-size: 16px; color: White;'>Structured data is stored in a well-defined schema, such as tables or databases. Common examples include Excel files, CSV files, and SQL databases.</p>", 
-        unsafe_allow_html=True
-    )
-    st.markdown("<h3 style='text-align: left; color: Black;'>Structured Data Formats</h3>", unsafe_allow_html=True)
-    if st.button("Excel"):
-        st.markdown(
-            "<p style='font-size: 16px; color: White;'>Excel is a spreadsheet format used to organize data into rows and columns. It supports formulas, charts, and more.</p>",
-            unsafe_allow_html=True
+with col1:
+    if st.button("More info on structured data"):
+        if.st.button("SQL"):
+            st.switch_page("pages/excel_Files.py")
+        display_lifecycle_stage(
+            "Problem Statement",
         )
-        st.switch_page("pages/excel_Files.py")
     if st.button("CSV"):
-        st.markdown(
-            "<p style='font-size: 16px; color: White;'>CSV (Comma-Separated Values) is a plain text format where each row represents a record, and fields are separated by commas.</p>",
-            unsafe_allow_html=True
+        display_lifecycle_stage(
+            "CSV"
         )
-    if st.button("SQL Database"):
-        st.markdown(
-            "<p style='font-size: 16px; color: White;'>SQL Databases store structured data in tables and allow querying using SQL (Structured Query Language).</p>",
-            unsafe_allow_html=True
+    if st.button("SQL"):
+        display_lifecycle_stage(
+            "SQL"
         )
-
-if st.button("Learn More About Unstructured Data"):
-    st.markdown(
-        "<p style='font-size: 16px; color: White;'>Unstructured data includes multimedia files, text documents, and other formats that lack a predefined schema. Examples are images, videos, and free-text files.</p>", 
-        unsafe_allow_html=True
-    )
-
-if st.button("Learn More About Semi-Structured Data"):
-    st.markdown(
-        "<p style='font-size: 16px; color: White;'>Semi-structured data lies between structured and unstructured data. Formats like JSON, XML, and YAML are common examples of semi-structured data.</p>", 
-        unsafe_allow_html=True
-    )
+