Update app.py

app.py

@@ -3,32 +3,23 @@ import streamlit as st
 def main():
     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
+    # Create three columns for the drag and drop areas
+    col1, col2, col3 = st.columns(3)
 
-    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
+    # Premium uploader and button
+    premium_file = col1.file_uploader("Drop your Premium CSV here", type=['csv'], key="premium")
     if premium_file is not None:
-        st.button("Proceed with Premium Analysis", key="premium_proceed")
+        col1.button("Proceed with Premium Analysis", key="premium_proceed")
 
+    # Incurred Claims & Recoveries Data uploader and button
+    claims_data_file = col2.file_uploader("Drop your Claims & Recoveries CSV here", type=['csv'], key="claims_data")
     if claims_data_file is not None:
-        st.button("Proceed with Claims Data Analysis", key="claims_data_proceed")
+        col2.button("Proceed with Claims Data Analysis", key="claims_data_proceed")
 
+    # Claims Count uploader and button
+    claims_count_file = col3.file_uploader("Drop your Claims Count CSV here", type=['csv'], key="claims_count")
     if claims_count_file is not None:
-        st.button("Proceed with Claims Count Analysis", key="claims_count_proceed")
+        col3.button("Proceed with Claims Count Analysis", key="claims_count_proceed")
 
 if __name__ == "__main__":
     main()