updated ui

app.py

@@ -48,17 +48,18 @@ def share():
     st.success(f"Share URL: {url}")
 
 with st.sidebar:
-    if st.button("Share"):
+    if st.button("Share", use_container_width=True):
         share()
 
     cols = st.columns(2)
     with cols[0]:
-        if st.button("Undo"):
-            undo()
-    with cols[1]:
-        if st.button("Restart"):
+        if st.button("Restart", type="primary", use_container_width=True):
             st.session_state.messages = []
             clear_uploader()
+    
+    with cols[1]:
+        if st.button("Undo", use_container_width=True):
+            undo()
 
     with st.expander("Advanced Configuration"):
         st.subheader("Temperature")