selfitcamera
init
e4879db
raw
history blame
8.45 kB
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">AI Poster Background Generation</h1>
"""
# 标题下的描述
description = r"""
Go to <a href='https://heybeauty.ai/product-background-swap' 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">Model Head Swap</h1>
"""
# 标题下的描述
description = r"""
Go to <a href='https://heybeauty.ai/model-head-swap' 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."