Spaces:

Springs-apps
/

PrintProntoDemo

Sleeping

File size: 6,592 Bytes

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 = """
# PrintPronto
Product: Business card

Color:
- Black (default)
- Blue
- Brown
- Orange
- Pink

Sizes:
3.5x2: multiplier 1.0
2.5x2.5: multiplier 0.95
2.125x3.375: multiplier 0.9
Custom size: This vendor doesn't support custom sizes

Material:
Standard (default): multiplier 1.0
Uncoated: multiplier 0.9

Base Prices (Quantity,Price per Unit $):
100, 0.23
250, 0.2
500, 1.9
1000, 1.75
"""

printer_2 = """
# BannerLord
Product: Business card

Color:
- Black (default)
- Blue

Sizes:
3.5x2: multiplier 1.0
2.5x2.5: multiplier 0.95
2.125x3.375: multiplier 0.9
3x2: multiplier 1.5
4x3: multiplier 1.5
Custom size: multiplier 2.0

Material:
Standard (default): multiplier 1.0
Uncoated: multiplier 0.95

Base Prices (Quantity,Price per Unit $):
100, 0.25
250, 0.22
500, 0.2
1000, 1.85
"""

printer_3 = """
# PrintMaster
Product: Business card
Color:
- Black (default)
- Blue
- Brown

Sizes:
3.5x2: multiplier 1.0
2.5x2.5: multiplier 0.95
2.125x3.375: multiplier 0.9
3x2: multiplier 1.5
4x3: multiplier 1.5
Custom size: multiplier 2.0

Material:
Standard (default): multiplier 1.0
Uncoated: multiplier 0.85

Base Prices (Quantity,Price per Unit $):
100, 0.21
250, 0.2
500, 0.18
1000, 0.17
"""


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, \n"
        "Answer:\n"
        "PrinterPronto: Doesn't support custom size\n"
        "###\n"
        "BannerLord: Total for 600: $524.40\n"
        "###\n"
        "PrintMaster: Total for 600: $497.40\n\n"
        "If user haven't specified one of the parameters, use the default one"
        "Your response should display ONLY final pricing for each available option, nothing more"
        "The '###' symbol must separate each printer's price calculation.\n"
    )

    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.HTML(label="Product Image"),
        gr.Textbox(label="AI response: text mode"),
    ],
    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)