import os import sys import cv2 import json import random import time import pickle import requests import func_timeout import numpy as np import gradio as gr from PIL import Image from collections import OrderedDict taskType = "3" try: taskType = str(os.environ['taskType']) except Exception as e: print(e) CapVisible = True Param4Label = "" Param4Visible = False Param5Label = "" Param5Visible = False TempInter = True if taskType=="3": # 是否显示token输入框 is_show_token = True # 标题 title = r""" <h1 align="center">Selfit AI App: Poster Background Generation</h1> """ # 标题下的描述 description = r""" Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br> 3. Share app with your friends and enjoy! 😊.<br> """ # 组件信息 TempLabel = "Choose a template" TempVisible = True UserLabel = "product photo" CaptionLabel = "English caption text (optional)" elif taskType=="4": # 是否显示token输入框 TempInter = False is_show_token = True # 标题 title = r""" <h1 align="center">Selfit AI App: AI Model Replace</h1> """ # 标题下的描述 description = r""" Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br> 3. Share app with your friends and enjoy! 😊.<br> """ # 组件信息 TempLabel = "Choose a Model" TempVisible = True UserLabel = "upload photo" CaptionLabel = "English caption text (optional)" elif taskType=="6": # 是否显示token输入框 is_show_token = True # 标题 title = r""" <h1 align="center">Selfit AI App: Image Style Filter</h1> """ # 标题下的描述 description = r""" Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br> 3. Share app with your friends and enjoy! 😊.<br> """ # 组件信息 TempLabel = "Choose a Style" TempVisible = True UserLabel = "upload photo" CaptionLabel = "English caption text (optional)" elif taskType=="5": # 是否显示token输入框 is_show_token = True TempInter = False # 标题 title = r""" <h1 align="center">Selfit AI App: Poster(human) Background Swap</h1> """ # 标题下的描述 description = r""" Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br> 3. Share app with your friends and enjoy! 😊.<br> """ # 组件信息 TempLabel = "background info" TempVisible = True UserLabel = "upload photo" CaptionLabel = "English caption text (optional)" elif taskType=="1": # 是否显示token输入框 is_show_token = True TempInter = False # 标题 title = r""" <h1 align="center">Selfit AI App: Object Segment</h1> """ # 标题下的描述 description = r""" Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br> 3. Share app with your friends and enjoy! 😊.<br> """ # 组件信息 TempLabel = "guide" TempVisible = True UserLabel = "upload photo" CaptionLabel = "English caption text (optional)" CapVisible = False elif taskType=="2": # 是否显示token输入框 is_show_token = True TempInter = False # 标题 title = r""" <h1 align="center">AI Object Removal</h1> """ # 标题下的描述 description = r""" Go to <a href='https://heybeauty.ai/remove-object' target='_blank'><b>HeyBeauty</b></a> for Faster and Free Object Removal! 🤗 .<br> 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br> 3. Share app with your friends and enjoy! 😊.<br> """ # 组件信息 TempLabel = "guide" TempVisible = True UserLabel = "upload and paint photo" CaptionLabel = "English caption text (optional)" CapVisible = False elif taskType=="7": # 是否显示token输入框 is_show_token = True TempInter = False # 标题 title = r""" <h1 align="center">Selfit AI App: Image Outpaint</h1> """ # 标题下的描述 description = r""" Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br> 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br> 3. Share app with your friends and enjoy! 😊.<br> """ # 组件信息 TempLabel = "guide" TempVisible = True UserLabel = "upload and paint photo" CaptionLabel = "English caption text (required)" CapVisible = True elif taskType=="9": # 是否显示token输入框 is_show_token = True TempInter = False # 标题 title = r""" <h1 align="center">Red Panda Image Generator by Recraft</h1> """ # 标题下的描述 description = r""" Go to <a href='https://heybeauty.ai/red-panda-recraft' target='_blank'><b>Recraft</b></a> for Faster and Free Generation! 🤗 .<br> 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. 2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br> 3. Share app with your friends and enjoy! 😊.<br> """ # 组件信息 TempLabel = "guide" TempVisible = True UserLabel = "upload and paint photo" CaptionLabel = "caption text (required, any language)" CapVisible = True else: # 是否显示token输入框 is_show_token = False # 标题 title = r""" <h1 align="center">Selfit AI App: Poster(product) background generation</h1> """ # 标题下的描述 description = r""" <a href='https://heybeauty.ai/extension' target='_blank'><b> Go to HeyBeauty for Faster and Free AI App! 🤗 </b></a>.<br> <b>Official 🤗 Gradio demo</b> for <a href='https://github.com/selfitcamera/Outfit-Anyone-in-the-Wild' target='_blank'><b>Product poster background generation</b></a>.<br> 1. Please wait for a while after submitting the task, you will get the result within 30 seconds. 2. <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>You can find your api key here</b></a> .<br> 3. Share app with your friends and enjoy! 😊.<br> """ if is_show_token: # 当试用次数耗尽,显示的信息 no_more_attempts = "You've exhausted all trial attempts. [Enter Your API Key](https://heybeauty.ai/home/apikeys) to continue..." else: # 当试用次数耗尽,显示的信息 no_more_attempts = "You've exhausted all trial attempts."