import os
import gradio as gr
import logging

LOG_FORMAT = '%(asctime)s - %(levelname)s  [%(name)s] %(funcName)s -> %(message)s'
logging.basicConfig(level = logging.DEBUG, format = LOG_FORMAT)
logger = logging.getLogger(__name__)
logger.debug('Logging started')

API_KEY=os.environ.get('HUGGING_FACE_HUB_TOKEN', None)
article = """---
This space was created using [SD Space Creator](https://huggingface.co/spaces/anzorq/sd-space-creator)."""

models = [
"models/ItsJayQz/Marvel_WhatIf_Diffusion",
"models/DGSpitzer/Cyberpunk-Anime-Diffusion",
"models/DGSpitzer/Guan-Yu-Diffusion",
"models/wavymulder/portraitplus",
"models/nitrosocke/classic-anim-diffusion",
"models/22h/vintedois-diffusion-v0-1",
"models/dreamlike-art/dreamlike-diffusion-1.0",
"models/stabilityai/stable-diffusion-2-1"
]

custom_model = "models/dreamlike-art/dreamlike-diffusion-1.0"

def selectModel(message):
    message = message.lower()
    c_model=models[0]
    logging.warning('model selected = '+c_model)
    return c_model

sandbox = gr.Interface.load(
    fn= selectModel,
    name= custom_model, 
    title="""AlStable sandbox""",
    inputs = gr.Textbox(label="Prompt", show_label=False, max_lines=2, placeholder="Enter your prompt", elem_id="input-prompt1", 
              gr.Textbox(label="Model", show_label=False, max_lines=2, placeholder="Enter your style", elem_id="input-prompt2"),
    description="""Demo for <a href="https://huggingface.co/stabilityai/stable-diffusion-2-1">AlStable</a> Stable Diffusion model.""",
    article=article,
    api_key=API_KEY
).queue(concurrency_count=20).launch()