import base64
import gradio as gr
from openai import OpenAI
import os
import dotenv


dotenv.load_dotenv()

client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

script_dir = os.path.dirname(os.path.abspath(__file__))

printer_1 = """
# PrinterOne
Product: Business Cards
Surface Options:
plastic:
    multiplier: 1.0
cardboard:
    multiplier: 0.9
wood:
    multiplier: 1.3

Sizes:
3x3 cm: multiplier 1.0
6x6 cm: multiplier 1.8

Finishing:
gloss: multiplier 1.1
matt: multiplier 1.0

Label Types:
permanent: multiplier 1.0
repositionable: multiplier 1.15
removable: multiplier 1.2

Base Prices (Quantity,Price per Unit $):
25,2.0
50,1.6
100,1.0
200,0.7
500,0.36
1000,0.2
2000,0.14
5000,0.12
10000,0.09
20000,0.07
"""

printer_2 = """
# BannerLord
Product: Business Cards
Surface Options:
plastic:
    multiplier: 1.0
cardboard:
    multiplier: 0.85
wood:
    multiplier: 1.4

Sizes:
3x3 cm: multiplier 1.0
6x6 cm: multiplier 1.9

Finishing:
gloss: multiplier 1.15
matt: multiplier 1.0

Label Types:
permanent: multiplier 1.0
repositionable: multiplier 1.2
removable: multiplier 1.25

Base Prices (Quantity,Price per Unit $):
25,2.4
50,1.8
100,1.2
200,0.75
500,0.4
1000,0.22
2000,0.15
5000,0.14
10000,0.1
20000,0.075
"""

printer_3 = """
# PrintMaster
Product: Business Cards
Surface Options:
plastic:
    multiplier: 1.0
cardboard:
    multiplier: 1.1
wood:
    multiplier: 1.25

Sizes:
3x3 cm: multiplier 1.0
6x6 cm: multiplier 1.85

Finishing:
gloss: multiplier 1.12
matt: multiplier 1.0

Label Types:
permanent: multiplier 1.0
repositionable: multiplier 1.18
removable: multiplier 1.22

Base Prices (Quantity,Price per Unit $):
50,1.4
75,1.2
100,1.05
150,1.0
200,0.875
250,0.72
500,0.4
750,0.4
1000,0.35
1500,0.3
2000,0.25
"""


def find_best_price(request):
    chat_prompt = (
        "You are a customer assistant and you have to find the best price for the customer. "
        "Here are the prices and options from the printers:\n"
        f"Printer 1: {printer_1}\n"
        f"Printer 2: {printer_2}\n"
        f"Printer 3: {printer_3}\n"
        "For each request, you should:\n"
        "1. Check if the requested size (3x3 or 6x6 cm) is available. If not, suggest the nearest available size.\n"
        "2. Calculate the final price by:\n"
        "   - Finding the nearest lower quantity in the base price list\n"
        "   - Multiplying by the surface option multiplier\n"
        "   - Multiplying by the size multiplier\n"
        "   - Multiplying by the finishing multiplier\n"
        "   - Multiplying by the label type multiplier\n"
        "3. Do not use linear interpolation for quantities - use the nearest lower quantity pricing.\n\n"
        "Example:\n"
        "User: I want to print 600 business cards, 6x6 cm size, premium white surface, gloss finish, permanent labels\n"
        "Answer:\n"
        "PrinterOne: 500 cards at base price $0.36. With multipliers (surface: 1.0, size: 1.8, finish: 1.1, label: 1.0) = $0.713 per unit. Total for 600: $427.80\n"
        "###\n"
        "BannerLord: 500 cards at base price $0.4. With multipliers (surface: 1.0, size: 1.9, finish: 1.15, label: 1.0) = $0.874 per unit. Total for 600: $524.40\n"
        "###\n"
        "PrintMaster: 500 cards at base price $0.4. With multipliers (surface: 1.0, size: 1.85, finish: 1.12, label: 1.0) = $0.829 per unit. Total for 600: $497.40\n\n"
        "The '###' symbol must separate each printer's price calculation. Show all calculations clearly as in the example above.\n"
        "For each printer, first state the base quantity and price, then show all applicable multipliers, the final per-unit price, and the total cost."
    )

    chat_response = client.chat.completions.create(
        model="gpt-4o-mini",
        messages=[
            {"role": "system", "content": chat_prompt},
            {"role": "user", "content": request},
        ],
    )

    chat_text = chat_response.choices[0].message.content

    html_page = render_html_page(chat_text)

    chat_text = chat_text.replace("###", "")
    return chat_text, html_page


def render_html_page(chat_text):
    images = [
        os.path.join(script_dir, "public/src/brochure3-400x400.jpg"),
        os.path.join(script_dir, "public/src/quality-unfolded-brochures_1_.png"),
        os.path.join(script_dir, "public/src/unfolded-glossy-brochures.png"),
    ]
    product_links = [
        "https://www.example.com/product_image.jpg",
        "https://www.example.com/product_image.jpg",
        "https://www.example.com/product_image.jpg",
    ]

    # Split chat_text into product descriptions
    product_descriptions = chat_text.split("###")

    # Generate HTML content
    html_content = """
    <html>
        <head>
            <title>Product Cards</title>
            <style>
                .product-card {
                    border: 1px solid #ccc;
                    border-radius: 5px;
                    padding: 16px;
                    margin: 16px;
                    text-align: center;
                    width: 200px;
                    display: inline-block;
                    vertical-align: top;
                }
                .product-card img {
                    max-width: 100%;
                    height: auto;
                }
                .product-card button {
                    background-color: #4CAF50;
                    color: white;
                    padding: 10px 20px;
                    border: none;
                    border-radius: 5px;
                    cursor: pointer;
                }
                .product-card button:hover {
                    background-color: #45a049;
                }
            </style>
        </head>
        <body>
    """

    for i, description in enumerate(product_descriptions):
        image = images[i % len(images)]
        with open(image, "rb") as img_file:
            base64_image = base64.b64encode(img_file.read()).decode("utf-8")

        product_link = product_links[i % len(product_links)]
        html_content += f"""
        <div class="product-card">
            <img src="data:image/jpeg;base64,{base64_image}" alt="Product Image">
            <p>{description}</p>
            <a href="{product_link}" target="_blank"><button>Buy Now</button></a>
        </div>
        """

    html_content += """
        </body>
    </html>
    """
    return html_content


logo = os.path.join(script_dir, "public/src/logo.svg")
with open(logo, "rb") as logo_file:
    base64_logo = base64.b64encode(logo_file.read()).decode("utf-8")


iface = gr.Interface(
    fn=find_best_price,
    inputs=gr.Textbox(lines=3, placeholder="Enter what are you looking for"),
    outputs=[
        gr.Textbox(label="Result"),
        gr.HTML(label="Product Image"),
    ],
    title="Get Instant Quote",
    description=f"""
    <div style="display: flex; justify-content: center; align-items: center; text-align: center;  margin-bottom: 20px;">
        <img src="data:image/svg+xml;base64,{base64_logo}" alt="Logo" style="width: 150px; height: auto;">
    </div>
    """,
    submit_btn="Get quote",
)


def auth_function(username, password):
    valid_users = {"admin": "password123", "user1": "pass456"}
    return username in valid_users and valid_users[username] == password


iface.launch(auth=auth_function, share=True, ssr_mode=False)