config.py

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/keys' 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/keys' 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/keys' 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/keys' 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/keys' 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">Selfit AI App: Object Remove</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/keys' 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/keys' 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

    Param4Label = "How much to expand width (0 - 2)"
    Param4Visible = True
    Param5Label = "How much to expand height (0 - 2)"
    Param5Visible = 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/keys' 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/keys) to continue..."
else:
    # 当试用次数耗尽，显示的信息
    no_more_attempts = "You've exhausted all trial attempts."