Update app.py

app.py

@@ -36,6 +36,14 @@ st.markdown("""
     .content {
         margin-bottom: 70px;
     }
+    .spacing {
+        margin: 10px 0;
+    }
+    .centered-button {
+        display: flex;
+        justify-content: center;
+        align-items: center;
+    }
     </style>
     """, unsafe_allow_html=True)
 
@@ -85,14 +93,18 @@ if uploaded_file is not None:
     with col4:
         st.image(sketch, caption='Sketch', use_column_width=True)
 
-    # Provide a download link for the sketch image
-    st.markdown("<br>", unsafe_allow_html=True)  # Add some space before the button
+    # Add some space before the button
+    st.markdown('<div class="spacing"></div>', unsafe_allow_html=True)
+    
+    # Provide a download link for the sketch image in the center
+    st.markdown('<div class="centered-button">', unsafe_allow_html=True)
     btn = st.download_button(
         label="Download Sketch",
         data=Image.fromarray(sketch).tobytes(),
         file_name="sketch.png",
         mime="image/png"
     )
+    st.markdown('</div>', unsafe_allow_html=True)
 else:
     st.info("Please upload an image to convert.")