import gradio as gr
from huggingface_hub import InferenceClient
import gradio as gr
import random
import os


checkpoint = "mistralai/Mixtral-8x7B-Instruct-v0.1"
client = InferenceClient(checkpoint)


def format_prompt(raw_Prompt):
    better_prompt = f"[INST] {raw_Prompt} [/INST]"
    return better_prompt


def findShops(target_platform, target_region, target_field, target_rangeFrom, target_rangeTo, target_crossSelling, target_similarArticles, target_amount, accessKey):
    if accessKey != os.environ['ACCESS_TOKEN']:
        return 'You are not authorized to use this software.'
    if target_platform == 'All':
        basePrompt = f"""
        Search for {target_amount} Shops with a website.
        """
    else:
        basePrompt = f"""
        Search for {target_amount} Shops that use the platform {target_platform}.
        """
    regionPrompt = f"""
    Make sure the recommended shops are located in {target_region}.
    """
    rangePrompt = f"""
    The websites of the recommended shops should have between {target_rangeFrom} and {target_rangeTo} monthly visitors."""
    crossSellingPrompt = """
    The recommended shops should be potential customers for a software that improves the cross-selling on their website."""
    similarArticlesPrompt = """
    The recommended shops should be potential customers for a software that can recommend similar products to the customers."""
    fieldPrompt = f"""
    Focus on shops that have a focus on the topic'{target_field}'."""
    stylePrompt = """
    Your response must be a list in markdown format.
    Each element of the list must follow these rules:
    - The title for each element must be the name of the shop.
    - The information coming after the title must each start in a new line.
    - The first information for each element must be a link to the website.
    - The next information must always be the number of monthly website visitors.
    - The next information must always be a summary of what is sold in the shop.
   """
    fullPrompt = basePrompt
    fullPrompt += regionPrompt
    fullPrompt += rangePrompt
    if target_crossSelling:
        fullPrompt += crossSellingPrompt
    if target_similarArticles:
        fullPrompt += similarArticlesPrompt
    if target_field != 'All':
        fullPrompt += fieldPrompt
    fullPrompt += stylePrompt
    generate_kwargs = dict(
        temperature=0.2,
        max_new_tokens=10000,
        top_p=0.9,
        repetition_penalty=1.0,
        do_sample=True,
        seed=random.randint(0, 100000),
    )
    stream = client.text_generation(format_prompt(fullPrompt),
                                    **generate_kwargs,
                                    stream=True,
                                    details=True,
                                    return_full_text=False)
    output = ""
    for response in stream:
        output += response.token.text
    output = output.removesuffix('</s>')
    return output


with gr.Blocks(title="Shop finder",
               theme=gr.themes.Soft(primary_hue="purple")) as demo:
    gr.Markdown(
        "##  Shop finder")
    gr.Markdown(
        "Choose the right settings and click **FIND** to search for shops.")
    with gr.Group():
        gr.Markdown("**FILTERS**")
        with gr.Column():
            platform = gr.Dropdown(choices=['All', 'Shopware', 'Shopify'],
                                   value='Shopware',
                                   interactive=True,
                                   label="PLATFORM")
            region = gr.Dropdown(choices=['Germany', 'England', 'France', 'Italy', 'Austria'],
                                 value='Germany',
                                 interactive=True,
                                 label="REGION")
            field = gr.Dropdown(choices=['All', 'Fashion', 'Outdoor', 'Home & Living',
                                         'Beauty & Cosmetics', 'Luggage & Bags',
                                         'Office & School materials','Pets','Sports',
                                        'Tech','Toys','Jewelry & Accessories','Food'],
                                value='All',
                                interactive=True,
                                label="FIELD")
    with gr.Group():
        gr.Markdown("**MONTHLY VISITORS**")
        with gr.Row():
            range_from = gr.Slider(minimum=0,
                                   maximum=1_000_000,
                                   value=100,
                                   step=100,
                                   label="FROM",
                                   interactive=True)
            range_to = gr.Slider(minimum=0,
                                 maximum=1_000_000,
                                 value=10_000,
                                 step=100,
                                 label="TO",
                                 interactive=True)
    with gr.Group():
        gr.Markdown("**POTENTIAL PRODUCT**")
        with gr.Row():
            opt_crossSelling = gr.Checkbox(value=False,
                                           label='CROSSELLING POTENTIAL',
                                           interactive=True)
            opt_similarArticles = gr.Checkbox(value=False,
                                              label='SIMILARITY POTENTIAL',
                                              interactive=True)
    numberOfShops = gr.Slider(minimum=1,
                              maximum=50,
                              value=10,
                              step=1,
                              label="AMOUNT",
                              interactive=True)

    btn = gr.Button("FIND")
    with gr.Group():
        gr.Markdown("**RESULTS**")
        out = gr.Markdown(value=' <br /> <br /> <br /> <br /> <br /> <br />',
                          label='POTENTIAL SHOPS')
    with gr.Group():
        gr.Markdown("*ACCESS SECRET*")
        loginKey = gr.Textbox(max_lines=1,
                              interactive=True,
                              type='password',
                              show_label=False)
    btn.click(fn=findShops,
              inputs=[platform, region, field,
                      range_from, range_to,
                      opt_crossSelling,
                      opt_similarArticles,
                      numberOfShops,
                      loginKey],
              outputs=out)

demo.launch(show_api=False)