Update app.py

app.py

@@ -82,52 +82,54 @@ def tryon(person_img, garment_img, seed, randomize_seed):
 MAX_SEED = 999999
 
 # Set up the Streamlit app
-st.set_page_config(page_title="Virtual-DressUp", page_icon=":guardsman:", layout="wide")
+st.set_page_config(page_title="Virtual-DressUp", page_icon=":dress:", layout="wide")
 
 st.title("Virtual-DressUp")
 st.markdown("""
-**Step 1:** Upload a person image ⬇️  
-**Step 2:** Upload a garment image ⬇️  
-**Step 3:** Press “Run” to get try-on results
+**Project Overview:**  
+This project leverages the advanced virtual try-on model by Kwai-Kolors. We've created a simple and interactive GUI for users to experience virtual try-ons. For more innovative features and models, check out Kwai-Kolors on their official website!
+
+[Explore Kwai-Kolors' website](https://klingai.com/)
+
 """)
 
+# Display the demo image
+st.image("demo img.png", caption="Demo Image", use_column_width=True)
+
+# Image upload columns
 col1, col2 = st.columns(2)
 
 with col1:
-    person_img = st.file_uploader("Person Image", type=["jpg", "jpeg", "png"])
-    
+    person_img = st.file_uploader("Upload Person Image", type=["jpg", "jpeg", "png"])
+
 with col2:
-    garment_img = st.file_uploader("Garment Image", type=["jpg", "jpeg", "png"])
+    garment_img = st.file_uploader("Upload Garment Image", type=["jpg", "jpeg", "png"])
 
-if person_img and garment_img:
-    person_img = np.array(bytearray(person_img.read()), dtype=np.uint8)
-    garment_img = np.array(bytearray(garment_img.read()), dtype=np.uint8)
-    person_img = cv2.imdecode(person_img, cv2.IMREAD_COLOR)
-    garment_img = cv2.imdecode(garment_img, cv2.IMREAD_COLOR)
-    
-    st.sidebar.header("Options")
-    st.sidebar.subheader("Instructions")
-    st.sidebar.markdown("""
-    1. Upload your **person image** 📸  
-    2. Upload your **garment image** 👗  
-    3. Press **“Run”** to get the try-on results 🔍
-    """)
-    
-    seed = st.sidebar.slider("Seed", 0, MAX_SEED, 0)
-    randomize_seed = st.sidebar.checkbox("Random seed", value=True)
+# Always show slider options
+st.sidebar.header("Options")
+seed = st.sidebar.slider("Seed", 0, MAX_SEED, 0)
+randomize_seed = st.sidebar.checkbox("Random seed", value=True)
 
-    if st.sidebar.button("Run"):
+# Show steps in sidebar
+st.sidebar.markdown("""
+**Steps to Follow:**
+1. Upload a person image ⬇️
+2. Upload a garment image ⬇️
+3. Press "Run" to get try-on results 🎉
+""")
+
+if st.sidebar.button("Run"):
+    if person_img and garment_img:
+        person_img = np.array(bytearray(person_img.read()), dtype=np.uint8)
+        garment_img = np.array(bytearray(garment_img.read()), dtype=np.uint8)
+        person_img = cv2.imdecode(person_img, cv2.IMREAD_COLOR)
+        garment_img = cv2.imdecode(garment_img, cv2.IMREAD_COLOR)
+        
         result_img, seed_used, result_info = tryon(person_img, garment_img, seed, randomize_seed)
         if result_info == "Success":
             st.image(result_img, caption="Result", channels="BGR")
             st.sidebar.text(f"Seed used: {seed_used}")
         else:
             st.sidebar.error(result_info)
-else:
-    st.sidebar.warning("Please upload both images to proceed.")
-
-st.markdown("""
-### About This Project
-
-This project features a virtual try-on model created by **Kwai-Kolors**. We provide a user-friendly GUI to showcase this model. For more amazing features and cool stuff, visit [Kwai-Kolors](https://klingai.com/).
-""")
+    else:
+        st.sidebar.warning("Please upload both images to proceed.")