import os import cv2 import numpy as np import json import random from PIL import Image, ImageDraw, ImageFont import asyncio import requests import base64 import gradio as gr machine_number = 0 model = os.path.join(os.path.dirname(__file__), "models/female_model.png") MODEL_MAP = { "AI Model female_model": 'models/female_model6.png', "AI Model female_model1": 'models/female_model3.jpg', "AI Model male_model": 'models/male_model3.png', "AI Model male_model2": 'models/male_model2.png', } # Sample clothing items from your clothes folder SAMPLE_CLOTHES = { "top": [ "clothes/shirt1.jpg", "clothes/shirt2.jpg", "clothes/dress1.jpg", "clothes/jacket1.jpg" ], "bottom": [ "clothes/pants1.jpg", "clothes/skirt1.jpg", "clothes/jeans1.jpg", "clothes/shorts1.jpg" ] } def add_waterprint(img): h, w, _ = img.shape img = cv2.putText(img, 'Powered by OutfitAnyone', (int(0.3*w), h-20), cv2.FONT_HERSHEY_PLAIN, 2, (128, 128, 128), 2, cv2.LINE_AA) return img def load_sample_image(image_path): """Load a sample clothing image and return it as numpy array""" try: if os.path.exists(image_path): img = cv2.imread(image_path) img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) return img else: # Return a placeholder if image doesn't exist return np.zeros((400, 400, 3), dtype=np.uint8) except Exception as e: print(f"Error loading image {image_path}: {e}") return np.zeros((400, 400, 3), dtype=np.uint8) def load_model_image(model_path): """Load a model image and return the file path""" try: if os.path.exists(model_path): return model_path else: print(f"Model image not found: {model_path}") return model # fallback to default except Exception as e: print(f"Error loading model {model_path}: {e}") return model # fallback to default def get_tryon_result(model_name, garment1, garment2, seed=1234): # model_name = "AI Model " + model_name.split("\\")[-1].split(".")[0] # windows # model_name = "AI Model " + model_name.split("/")[-1].split(".")[0] # linux # print(model_name) # garment1 = cv2.imread(garment1) # read path into numpy # garment1 = cv2.cvtColor(garment1, cv2.COLOR_BGR2RGB) if not model_name.startswith("AI Model "): model_name = "AI Model " + model_name.split("/")[-1].split(".")[0] print(model_name) encoded_garment1 = cv2.imencode('.jpg', garment1)[1].tobytes() encoded_garment1 = base64.b64encode(encoded_garment1).decode('utf-8') if garment2 is not None: encoded_garment2 = cv2.imencode('.jpg', garment2)[1].tobytes() encoded_garment2 = base64.b64encode(encoded_garment2).decode('utf-8') else: encoded_garment2 = '' # Fix for the missing environment variable try: url = os.environ['OA_IP_ADDRESS'] # reads your key from Hugging Face secrets except KeyError: print("Error: OA_IP_ADDRESS environment variable not set!") # return a dummy image instead of crashing dummy_img = np.zeros((512, 512, 3), dtype=np.uint8) dummy_img = cv2.putText(dummy_img, 'Error: API key not set', (50, 256), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2) return add_waterprint(dummy_img) headers = {'Content-Type': 'application/json', "Authorization": f"Bearer {os.environ.get('OA_API_KEY', '')}" } seed = random.randint(0, 1222222222) data = { "garment1": encoded_garment1, "garment2": encoded_garment2, "model_name": model_name, "seed": seed } try: response = requests.post(url, headers=headers, data=json.dumps(data)) print("response code", response.status_code) if response.status_code == 200: result = response.json() result = base64.b64decode(result['images'][0]) result_np = np.frombuffer(result, np.uint8) result_img = cv2.imdecode(result_np, cv2.IMREAD_UNCHANGED) else: print('server error!') # Return error image result_img = np.zeros((512, 512, 3), dtype=np.uint8) result_img = cv2.putText(result_img, f'Server Error: {response.status_code}', (50, 256), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2) except Exception as e: print(f"Request error: {e}") result_img = np.zeros((512, 512, 3), dtype=np.uint8) result_img = cv2.putText(result_img, 'Connection Error', (50, 256), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2) final_img = add_waterprint(result_img) return final_img with gr.Blocks(css = ".output-image, .input-image, .image-preview {height: 400px !important} ") as demo: # Step 1: hidden state to store the selected model name selected_model = gr.State("AI Model female_model") gr.HTML( """

Outfit Anyone: Ultra-high quality virtual try-on for Any Clothing and Any Person

v0.9

OutfitAnyone plus version is now online with any model and any cloth: https://www.outfitanyone.life/
Project Page
""") with gr.Row(): with gr.Column(): # Model selection buttons gr.HTML("

Select AI Model:

") with gr.Row(): model_btn1 = gr.Button("Female Model", variant="secondary") model_btn2 = gr.Button("Female Model 1", variant="secondary") model_btn3 = gr.Button("Male Model", variant="secondary") model_btn4 = gr.Button("Male Model 2", variant="secondary") init_image = gr.Image(sources='clipboard', type="filepath", label="model", value=model) with gr.Column(): gr.HTML( """

Models are fixed and cannot be uploaded or modified; we only support users uploading their own garments.

For a one-piece dress or coat, you only need to upload the image to the 'top garment' section and leave the 'lower garment' section empty.

""") # Sample clothing buttons section gr.HTML("

Quick Select Sample Clothes:

") with gr.Row(): gr.HTML("

Top Garments:

") with gr.Row(): top_btn1 = gr.Button("Shirt 1", variant="secondary") top_btn2 = gr.Button("Shirt 2", variant="secondary") top_btn3 = gr.Button("Dress", variant="secondary") top_btn4 = gr.Button("Jacket", variant="secondary") with gr.Row(): gr.HTML("

Bottom Garments:

") with gr.Row(): bottom_btn1 = gr.Button("Pants", variant="secondary") bottom_btn2 = gr.Button("Skirt", variant="secondary") bottom_btn3 = gr.Button("Jeans", variant="secondary") bottom_btn4 = gr.Button("Shorts", variant="secondary") with gr.Row(): garment_top = gr.Image(sources='upload', type="numpy", label="top garment") garment_down = gr.Image(sources='upload', type="numpy", label="lower garment") run_button = gr.Button(value="Run") with gr.Column(): gallery = gr.Image() # Connect model buttons to load model images model_btn1.click( lambda: ("AI Model female_model", load_model_image(MODEL_MAP["AI Model female_model"])), outputs=[selected_model, init_image] ) model_btn2.click( lambda: ("AI Model female_model1", load_model_image(MODEL_MAP["AI Model female_model1"])), outputs=[selected_model, init_image] ) model_btn3.click( lambda: ("AI Model male_model", load_model_image(MODEL_MAP["AI Model male_model"])), outputs=[selected_model, init_image] ) model_btn4.click( lambda: ("AI Model male_model2", load_model_image(MODEL_MAP["AI Model male_model2"])), outputs=[selected_model, init_image] ) # Connect buttons to load sample images top_btn1.click( lambda: load_sample_image(SAMPLE_CLOTHES["top"][0]), outputs=[garment_top] ) top_btn2.click( lambda: load_sample_image(SAMPLE_CLOTHES["top"][1]), outputs=[garment_top] ) top_btn3.click( lambda: load_sample_image(SAMPLE_CLOTHES["top"][2]), outputs=[garment_top] ) top_btn4.click( lambda: load_sample_image(SAMPLE_CLOTHES["top"][3]), outputs=[garment_top] ) bottom_btn1.click( lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][0]), outputs=[garment_down] ) bottom_btn2.click( lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][1]), outputs=[garment_down] ) bottom_btn3.click( lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][2]), outputs=[garment_down] ) bottom_btn4.click( lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][3]), outputs=[garment_down] ) # top_btn1.click( # lambda: load_sample_image(SAMPLE_CLOTHES["top"][0], outputs=[garment_top]), # outputs=[garment_top] # ) # top_btn2.click( # lambda: load_sample_image(SAMPLE_CLOTHES["top"][1], outputs=[garment_top]), # outputs=[garment_top] # ) # top_btn3.click( # lambda: load_sample_image(SAMPLE_CLOTHES["top"][2], outputs=[garment_top]), # outputs=[garment_top] # ) # top_btn4.click( # lambda: load_sample_image(SAMPLE_CLOTHES["top"][3], outputs=[garment_top]), # outputs=[garment_top] # ) # bottom_btn1.click( # lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][0], outputs=[garment_down]), # outputs=[garment_down] # ) # bottom_btn2.click( # lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][1], outputs=[garment_down]), # outputs=[garment_down] # ) # bottom_btn3.click( # lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][2], outputs=[garment_down]), # outputs=[garment_down] # ) # bottom_btn4.click( # lambda: load_sample_image(SAMPLE_CLOTHES["bottom"][3], outputs=[garment_down]), # outputs=[garment_down] # ) run_button.click(fn=get_tryon_result, inputs=[selected_model, garment_top, garment_down], outputs=[gallery], concurrency_limit=2) if __name__ == "__main__": ip = requests.get('http://ifconfig.me/ip', timeout=1).text.strip() print("ip address alibaba", ip) demo.queue(max_size=10) demo.launch()