import gradio as gr

theme = gr.themes.Base(
    text_size=gr.themes.Size(lg="16px", 
                             md="20px", # body
                             sm="12px", 
                             xl="40px", # h2
                             xs="10px", 
                             xxl="50px", # h1 
                             xxs="9px"),
    #spacing_size="lg",
)

css = """
h1, h2{
    text-align: center;
}

"""

with gr.Blocks(theme=theme, css=css) as demo:
    # Title
    gr.Markdown("""
    # AI and Fashion E-Commerce
    <center> by <a href="https://www.tonyassi.com/" target="_blank">Tony Assi</a>  </center>
    """)

    # Virtual Try-On
    gr.Markdown("""
    ---
    ## Virtual Try-On
    ![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/648a824a8ca6cf9857d1349c/mARmPyep7W3MwGmvR_zWN.jpeg)

    Virtually try-on clothing by submitting your image and a clothing image. A novel approach to VTO using IP-Adapter inpainting and body segmentation. [Read more](https://huggingface.co/blog/tonyassi/virtual-try-on-ip-adapter)

    [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/tonyassi/fashion-try-on)

    [![Google Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1d-NJNLdX6dqs5Qfm1UYDQknBrhwZXx7R?usp=sharing)
    """)

    # Sales Forecasting with Image Regression
    gr.Markdown("""

    ---
    ## Sales Forecasting with Image Regression
    ![image/png](https://cdn-uploads.huggingface.co/production/uploads/648a824a8ca6cf9857d1349c/mrtzCY91RnLtnbx4LIj4x.png)
    Predict sales from a product image. Image regression training, hosting, and inference. [Read more](https://huggingface.co/blog/tonyassi/image-regression)

    [![Github](https://img.shields.io/badge/Github-Repo-blue)](https://github.com/TonyAssi/ImageRegression)

    [![Model](https://img.shields.io/badge/%F0%9F%A4%97-Models-yellow)](https://huggingface.co/tonyassi/sales-prediction)

    """)    

    # Text-to-Image Clothing Designer
    gr.Markdown("""

    ---
    ## Text-to-Image Clothing Designer
    ![image/png](https://cdn-uploads.huggingface.co/production/uploads/648a824a8ca6cf9857d1349c/Y-EuOGASjO-GMUzjlBCIh.png)

    This text-to-image model was fine-tuned on [Lucy in the Sky](https://www.lucyinthesky.com/) product image - product text pairs. Text input is limited to the language of the product catalog it was trained on which forces the model to input images in a simialr style to the product images.

    [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/LucyintheSky/lucy-text-to-image)
    """)  


demo.launch()