# This file is part of stable-diffusion-webui (https://github.com/sd-webui/stable-diffusion-webui/).

# Copyright 2022 sd-webui team.
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.

# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>. 
# base webui import and utils.
from webui_streamlit import st
from sd_utils import *

# streamlit imports


#other imports
import pandas as pd
from io import StringIO

# Temp imports 


# end of imports
#---------------------------------------------------------------------------------------------------------------

def layout():
    #search = st.text_input(label="Search", placeholder="Type the name of the model you want to search for.", help="")

    csvString = f"""
                    ,Stable Diffusion v1.4            , ./models/ldm/stable-diffusion-v1               , https://huggingface.co/CompVis/stable-diffusion-v-1-4-original                  
                    ,GFPGAN v1.3                      , ./src/gfpgan/experiments/pretrained_models     , https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth                     
                    ,RealESRGAN_x4plus                , ./src/realesrgan/experiments/pretrained_models , https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth            
                    ,RealESRGAN_x4plus_anime_6B       , ./src/realesrgan/experiments/pretrained_models , https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth 
                    ,Waifu Diffusion v1.2             , ./models/custom                                , https://huggingface.co/hakurei/waifu-diffusion
                    ,Waifu Diffusion v1.2 Pruned      , ./models/custom                                , https://huggingface.co/crumb/pruned-waifu-diffusion
                    ,TrinArt Stable Diffusion v2      , ./models/custom                                , https://huggingface.co/naclbit/trinart_stable_diffusion_v2
                    ,Stable Diffusion Concept Library , ./models/custom/sd-concepts-library             , https://github.com/sd-webui/sd-concepts-library
                    """
    colms = st.columns((1, 3, 5, 5))
    columns = ["№",'Model Name','Save Location','Download Link']

    # Convert String into StringIO
    csvStringIO = StringIO(csvString)
    df = pd.read_csv(csvStringIO, sep=",", header=None, names=columns)		

    for col, field_name in zip(colms, columns):
        # table header
        col.write(field_name)

    for x, model_name in enumerate(df["Model Name"]):
        col1, col2, col3, col4 = st.columns((1, 3, 4, 6))
        col1.write(x)  # index
        col2.write(df['Model Name'][x])
        col3.write(df['Save Location'][x])
        col4.write(df['Download Link'][x])