Update app.py

app.py

@@ -1,34 +1,34 @@
 import streamlit as st
 
 def main():
-    # Create a container to hold the columns
-    container = st.container()
-
-    # Using a with statement ensures that the columns are at the same level
-    with container:
-        # Create three columns for the drag and drop areas
-        col1, col2, col3 = st.columns(3)
-
-        with col1:
-            st.header("Premium")
-            premium_file = st.file_uploader("Drop your Premium CSV here", type=['csv'], key="premium")
-            if premium_file is not None:
-                # Button for next step of Premium
-                st.button("Proceed with Premium Analysis", key="premium_proceed")
-
-        with col2:
-            st.header("Incurred Claims & Recoveries Data")
-            claims_data_file = st.file_uploader("Drop your Claims & Recoveries CSV here", type=['csv'], key="claims_data")
-            if claims_data_file is not None:
-                # Button for next step of Claims Data
-                st.button("Proceed with Claims Data Analysis", key="claims_data_proceed")
-
-        with col3:
-            st.header("Claims Count")
-            claims_count_file = st.file_uploader("Drop your Claims Count CSV here", type=['csv'], key="claims_count")
-            if claims_count_file is not None:
-                # Button for next step of Claims Count
-                st.button("Proceed with Claims Count Analysis", key="claims_count_proceed")
+    st.title("Data Upload for Analysis")
+
+    # Set the layout for file uploaders
+    layout = st.columns(3)
+    # Initialize a variable for each file to None
+    premium_file = claims_data_file = claims_count_file = None
+
+    with layout[0]:
+        st.header("Premium")
+        premium_file = st.file_uploader("Drop your Premium CSV here", type=['csv'], key="premium")
+
+    with layout[1]:
+        st.header("Incurred Claims & Recoveries Data")
+        claims_data_file = st.file_uploader("Drop your Claims & Recoveries CSV here", type=['csv'], key="claims_data")
+
+    with layout[2]:
+        st.header("Claims Count")
+        claims_count_file = st.file_uploader("Drop your Claims Count CSV here", type=['csv'], key="claims_count")
+
+    # Check if files are uploaded and if so, display the proceed buttons
+    if premium_file is not None:
+        st.button("Proceed with Premium Analysis", key="premium_proceed")
+
+    if claims_data_file is not None:
+        st.button("Proceed with Claims Data Analysis", key="claims_data_proceed")
+
+    if claims_count_file is not None:
+        st.button("Proceed with Claims Count Analysis", key="claims_count_proceed")
 
 if __name__ == "__main__":
     main()