Container Maybe

app.py

@@ -47,14 +47,16 @@ elif st.session_state["step"] == 2:
     selected_mpid = selected_display.split(" ")[0]
 
     if st.button("Get Data"):
-        st.write(f"Material ID: {selected_mpid}")
-        data_dict = get_data_from_mpid(selected_mpid)
-        if data_dict:
-            st.write("Structure Description:")
-            description = get_structure_description(selected_mpid)
-            st.write(description)
-        else:
-            st.write("No data found for this material ID.")
+        with st.spinner("Retrieving data..."):
+            st.write(f"Material ID: {selected_mpid}")
+            with st.container(border=True, height=500):
+                data_dict = get_data_from_mpid(selected_mpid)
+                if data_dict:
+                    st.write("Structure Description:")
+                    description = get_structure_description(selected_mpid)
+                    st.write(description)
+                else:
+                    st.write("No data found for this material ID.")
 
     chemical_name = get_chemical_name(chemical)
     if chemical_name: