Upload 7 files

Procfile

@@ -0,0 +1 @@
+web: streamlit run app.py

README.md

@@ -1,12 +1,56 @@
----
-title: Vton 002
-emoji: 🏢
-colorFrom: pink
-colorTo: blue
-sdk: streamlit
-sdk_version: 1.42.2
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Virtual Try-On Application
+
+This application uses the IDM-VTON model via Replicate API to perform virtual try-on of garments on human images.
+
+## Requirements
+
+- Python 3.6+
+- `replicate` Python package
+
+## Installation
+
+1. Install the required package:
+
+```bash
+pip install replicate
+```
+
+2. Set up your Replicate API token as an environment variable:
+
+```bash
+export REPLICATE_API_TOKEN='your_token_here'
+```
+
+You can get your API token from [Replicate](https://replicate.com/account/api-tokens) after creating an account.
+
+## Usage
+
+Run the script with:
+
+```bash
+python vton.py
+```
+
+By default, the script uses example images. To use your own images, modify the `garm_img`, `human_img`, and `garment_des` variables in the script.
+
+### Function Usage
+
+You can also import the function in your own Python code:
+
+```python
+from vton import virtual_try_on
+
+output_path = virtual_try_on(
+    garm_img_url="URL_TO_GARMENT_IMAGE",
+    human_img_url="URL_TO_HUMAN_IMAGE",
+    garment_des="Description of the garment"
+)
+```
+
+## Output
+
+The script will save the generated try-on image as `output.jpg` in the current directory.
+
+## Model Information
+
+This application uses the IDM-VTON model created by cuuupid on Replicate. The model performs virtual try-on by overlaying garments onto human images based on the provided description.

app.py

@@ -0,0 +1,148 @@
+import streamlit as st
+import os
+from vton import virtual_try_on
+import requests
+from PIL import Image
+from io import BytesIO
+
+# Set page config for a wider layout and custom title
+st.set_page_config(
+    page_title="Virtual Try-On App",
+    page_icon="👔",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
+# Custom CSS for better styling
+st.markdown("""
+<style>
+    .stApp {
+        max-width: 1200px;
+        margin: 0 auto;
+    }
+    .upload-box {
+        border: 2px dashed #4c4c4c;
+        border-radius: 10px;
+        padding: 20px;
+        text-align: center;
+        margin: 10px 0;
+    }
+    .output-box {
+        border: 1px solid #e0e0e0;
+        border-radius: 10px;
+        padding: 20px;
+        margin: 20px 0;
+        background-color: #f8f9fa;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+# Title and description
+st.title("✨ Virtual Try-On Application")
+st.markdown("""Upload a garment image and a person's photo to see how the garment would look on them!""")
+
+# Create two columns for input images
+col1, col2 = st.columns(2)
+
+# Garment image upload
+with col1:
+    st.subheader("Garment Image")
+    garment_source = st.radio(
+        "Choose garment image source:",
+        ["Upload", "URL"],
+        key="garment_source"
+    )
+    
+    if garment_source == "Upload":
+        garment_file = st.file_uploader("Upload garment image", type=["jpg", "jpeg", "png", "webp"], key="garment_upload")
+        if garment_file:
+            st.image(garment_file, caption="Uploaded Garment", use_container_width=True)
+            # Save uploaded file to temporary URL
+            garment_img = Image.open(garment_file)
+            garment_buffer = BytesIO()
+            garment_img.save(garment_buffer, format="PNG")
+            garment_url = "temp_garment.png"
+            with open(garment_url, "wb") as f:
+                f.write(garment_buffer.getvalue())
+    else:
+        garment_url = st.text_input("Enter garment image URL", key="garment_url")
+        if garment_url:
+            try:
+                st.image(garment_url, caption="Garment from URL", use_container_width=True)
+            except:
+                st.error("Unable to load image from URL")
+
+# Person image upload
+with col2:
+    st.subheader("Person Image")
+    person_source = st.radio(
+        "Choose person image source:",
+        ["Upload", "URL"],
+        key="person_source"
+    )
+    
+    if person_source == "Upload":
+        person_file = st.file_uploader("Upload person image", type=["jpg", "jpeg", "png", "webp"], key="person_upload")
+        if person_file:
+            st.image(person_file, caption="Uploaded Person", use_container_width=True)
+            # Save uploaded file to temporary URL
+            person_img = Image.open(person_file)
+            person_buffer = BytesIO()
+            person_img.save(person_buffer, format="PNG")
+            person_url = "temp_person.png"
+            with open(person_url, "wb") as f:
+                f.write(person_buffer.getvalue())
+    else:
+        person_url = st.text_input("Enter person image URL", key="person_url")
+        if person_url:
+            try:
+                st.image(person_url, caption="Person from URL", use_container_width=True)
+            except:
+                st.error("Unable to load image from URL")
+
+# Garment description input
+st.subheader("Garment Description")
+garment_desc = st.text_area(
+    "Describe the garment (e.g., color, style, type)",
+    height=100,
+    placeholder="Example: A cute pink top with floral pattern"
+)
+
+# Check for API token
+if "REPLICATE_API_TOKEN" not in os.environ:
+    st.warning(
+        "⚠️ REPLICATE_API_TOKEN is not set. Please set it using: "
+        "export REPLICATE_API_TOKEN='your_token_here'"
+    )
+
+# Process button
+if st.button("Generate Try-On", type="primary"):
+    if not (garment_url and person_url and garment_desc):
+        st.error("Please provide all required inputs!")
+    else:
+        try:
+            with st.spinner("🔄 Processing virtual try-on..."):
+                output_path = virtual_try_on(garment_url, person_url, garment_desc)
+                
+                # Display result in a styled box
+                st.markdown("<div class='output-box'>", unsafe_allow_html=True)
+                st.subheader("🎉 Try-On Result")
+                st.image(output_path, caption="Virtual Try-On Result", use_container_width=True)
+                st.markdown("</div>", unsafe_allow_html=True)
+                
+                # Cleanup temporary files
+                if os.path.exists("temp_garment.png"):
+                    os.remove("temp_garment.png")
+                if os.path.exists("temp_person.png"):
+                    os.remove("temp_person.png")
+                    
+        except Exception as e:
+            st.error(f"Error occurred: {str(e)}")
+
+# Footer
+st.markdown("---")
+st.markdown("""
+<div style='text-align: center; color: #666;'>
+    <p>Powered by IDM-VTON model via Replicate API</p>
+</div>
+""", unsafe_allow_html=True)

packages.txt

@@ -0,0 +1,3 @@
+python3-dev
+python3-pip
+python3-venv

requirements.txt

@@ -0,0 +1,4 @@
+replicate==0.22.0
+streamlit==1.31.0
+Pillow==10.2.0
+requests==2.31.0

vton.py

@@ -0,0 +1,67 @@
+import replicate
+import requests
+import os
+
+def virtual_try_on(garm_img_url, human_img_url, garment_des):
+    """
+    Perform virtual try-on using the IDM-VTON model on Replicate.
+    
+    Args:
+        garm_img_url (str): Path or URL of the garment image
+        human_img_url (str): Path or URL of the human image
+        garment_des (str): Description of the garment
+        
+    Returns:
+        str: Path to the saved output image
+    """
+    # Convert local files to data URLs if needed
+    def file_to_data_url(file_path):
+        if file_path.startswith('http'):
+            response = requests.get(file_path)
+            img_data = response.content
+        else:
+            with open(file_path, 'rb') as f:
+                img_data = f.read()
+        import base64
+        encoded = base64.b64encode(img_data).decode('utf-8')
+        return f"data:image/png;base64,{encoded}"
+    
+    # Prepare input for the model
+    input_data = {
+        "garm_img": file_to_data_url(garm_img_url),
+        "human_img": file_to_data_url(human_img_url),
+        "garment_des": garment_des
+    }
+    
+    print("Sending request to IDM-VTON model...")
+    # Run the model on Replicate
+    output = replicate.run(
+        "cuuupid/idm-vton:c871bb9b046607b680449ecbae55fd8c6d945e0a1948644bf2361b3d021d3ff4",
+        input=input_data
+    )
+    
+    # Save the output image
+    output_path = "output.jpg"
+    with open(output_path, "wb") as file:
+        file.write(output.read())
+    
+    print(f"Virtual try-on complete! Output saved to {output_path}")
+    return output_path
+
+# Example usage
+if __name__ == "__main__":
+    # Example inputs
+    garm_img = "https://replicate.delivery/pbxt/KgwTlZyFx5aUU3gc5gMiKuD5nNPTgliMlLUWx160G4z99YjO/sweater.webp"
+    human_img = "https://replicate.delivery/pbxt/KgwTlhCMvDagRrcVzZJbuozNJ8esPqiNAIJS3eMgHrYuHmW4/KakaoTalk_Photo_2024-04-04-21-44-45.png"
+    garment_des = "cute pink top"
+    
+    # Check if REPLICATE_API_TOKEN is set
+    if "REPLICATE_API_TOKEN" not in os.environ:
+        print("Warning: REPLICATE_API_TOKEN environment variable is not set.")
+        print("Please set it using: export REPLICATE_API_TOKEN='your_token_here'")
+    
+    try:
+        output_path = virtual_try_on(garm_img, human_img, garment_des)
+        print(f"Success! Try-on image saved to {output_path}")
+    except Exception as e:
+        print(f"Error occurred: {e}")