demo v2

app.py

@@ -3,17 +3,17 @@ import gradio as gr
 from openai import OpenAI
 import os
 import dotenv
-
+import re
 
 dotenv.load_dotenv()
 
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 script_dir = os.path.dirname(os.path.abspath(__file__))
-
+COMPANIES = ["PrintPronto", "BannerLord", "PrintMaster"]
 printer_1 = """
 # PrintPronto
-Product: Business card
+## Product: Business cards
 
 Color:
 - Black (default)
@@ -23,7 +23,7 @@ Color:
 - Pink
 
 Sizes:
-3.5x2: multiplier 1.0
+3.5x2: multiplier 1.0 (default)
 2.5x2.5: multiplier 0.95
 2.125x3.375: multiplier 0.9
 Custom size: This vendor doesn't support custom sizes
@@ -36,24 +36,91 @@ Base Prices (Quantity,Price per Unit $):
 100, 0.23
 250, 0.2
 500, 0.19
-1000, 0.175
+1000, 0.17
+
+## Product: Fabric Banners
+
+Sizes:
+2.5x4: multiplier 1.0
+4x4: multiplier 1.1
+2.5x6: multiplier 1.0 (default)
+2.5x8: multiplier 1.2
+4x6: multiplier 1.5
+Custom size: This vendor doesn't support custom sizes
+
+Base Prices (Quantity,Price per Unit $):
+1, 230.00
+5, 210.00
+10, 201.50
+25, 195.26
+
+## Product: Bumper Stickers
+
+Shapes:
+Rectangles: multiplier 1.0
+Squares: multiplier 1.0
+Circles: multiplier 1.0
+Ovals: multiplier 1.0
+Rounded Rectangles: multiplier 1.0
+Custom shape: This vendor doesn't support custom shapes
+
+Sizes:
+2x3: multiplier 1.0
+2x4: multiplier 1.1
+3x4: multiplier 1.0 (default)
+3x6: multiplier 1.2
+2x8: multiplier 1.5
+Custom size: This vendor doesn't support custom sizes
+
+Base Prices (Quantity,Price per Unit $):
+50, 1.80
+100, 1.10
+250, 0.80
+500, 0.44
+1000, 0.24
+
+## Product: Paper Stickers
+
+Shapes:
+Rectangles: multiplier 1.0 
+Squares: multiplier 1.0
+Circles: multiplier 1.0
+Ovals: multiplier 1.0
+Custom size: This vendor doesn't support custom shapes
+
+Sizes:
+0.79: multiplier 1.0 (default)
+1: multiplier 1.1
+1.18: multiplier 1.0
+1.26: multiplier 1.2
+1.38: multiplier 1.5
+Custom size: This vendor doesn't support custom sizes
+
+Material:
+- Gloss Paper (default)
+- Matte Paper
+
+Base Prices (Quantity,Price per Unit $):
+250, 0,40
+500, 0.22
+1000, 0.12
 """
 
 printer_2 = """
 # BannerLord
-Product: Business card
+## Product: Business cards
 
 Color:
 - Black (default)
 - Blue
 
 Sizes:
-3.5x2: multiplier 1.0
+3.5x2: multiplier 1.0 (default)
 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
+Custom size: This vendor supports custom sizes with a multiplier of 2.0
 
 Material:
 - Standard (default)
@@ -63,24 +130,92 @@ Base Prices (Quantity,Price per Unit $):
 100, 0.25
 250, 0.22
 500, 0.2
-1000, 0.185
+1000, 0.19
+
+## Product: Fabric Banners
+
+Sizes:
+2.5x4: multiplier 1.0
+4x4: multiplier 1.1
+2.5x6: multiplier 1.0 (default)
+2.5x8: multiplier 1.2
+4x6: multiplier 1.5
+Custom size: This vendor supports custom sizes with a multiplier of 2.5
+
+Base Prices (Quantity,Price per Unit $):
+1, 228.00
+5, 205.00
+10, 204.90
+25, 193.56
+
+## Product: Bumper Stickers
+
+Shapes:
+Rectangles: multiplier 1.0 
+Squares: multiplier 1.0
+Circles: multiplier 1.0
+Ovals: multiplier 1.0
+Rounded Rectangles: multiplier 1.0
+Custom shape: This vendor supports custom shapes with a multiplier of 1.5
+
+Sizes:
+2x3: multiplier 1.0
+2x4: multiplier 1.1
+3x4: multiplier 1.0 (default)
+3x6: multiplier 1.2
+2x8: multiplier 1.5
+Custom size: This vendor supports custom sizes with a multiplier of 2.0
+
+Base Prices (Quantity,Price per Unit $):
+50, 1.90
+100, 1.15
+250, 0.85
+500, 0.45
+1000, 0.25
+
+## Product: Paper Stickers
+
+Shapes:
+Rectangles: multiplier 1.0
+Squares: multiplier 1.0
+Circles: multiplier 1.0
+Ovals: multiplier 1.0
+Custom size: This vendor supports custom sizes with a multiplier of 2.0
+
+Sizes:
+0.79: multiplier 1.0 (default)
+1: multiplier 1.1
+1.18: multiplier 1.0
+1.26: multiplier 1.2
+1.38: multiplier 1.5
+Custom size: This vendor supports custom sizes with a multiplier of 2.0
+
+Material:
+- Gloss Paper (default)
+- Matte Paper
+
+Base Prices (Quantity,Price per Unit $):
+250, 0.42
+500, 0.23
+1000, 0.13
 """
 
 printer_3 = """
 # PrintMaster
-Product: Business card
+## Product: Business cards
+
 Color:
 - Black (default)
 - Blue
 - Brown
 
 Sizes:
-3.5x2: multiplier 1.0
+3.5x2: multiplier 1.0 (default)
 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
+Custom size: This vendor supports custom sizes with a multiplier of 2.0
 
 Material:
 - Standard (default)
@@ -89,36 +224,178 @@ Material:
 Base Prices (Quantity,Price per Unit $):
 100, 0.21
 250, 0.2
-500, 0.18
-1000, 0.17
+500, 0.19
+1000, 0.18
+
+## Product: Fabric Banners
+
+Sizes:
+2.5x4: multiplier 1.0 (default)
+4x4: multiplier 1.1
+2.5x6: multiplier 1.0
+2.5x8: multiplier 1.2
+4x6: multiplier 1.5
+Custom size: This vendor supports custom sizes with a multiplier of 2.2
+
+Base Prices (Quantity,Price per Unit $):
+1, 231.00
+5, 209.50
+10, 203.45
+25, 194.00
+
+## Product: Bumper Stickers
+
+Shapes:
+Rectangles: multiplier 1.0
+Squares: multiplier 1.0
+Circles: multiplier 1.0
+Ovals: multiplier 1.0
+Rounded Rectangles: multiplier 1.0
+Custom shape: This vendor supports custom shapes with a multiplier of 1.5
+
+Sizes:
+2x3: multiplier 1.0 (default)
+2x4: multiplier 1.1
+3x4: multiplier 1.0
+3x6: multiplier 1.2
+2x8: multiplier 1.5
+Custom size: This vendor supports custom sizes with a multiplier of 2.0
+
+Base Prices (Quantity,Price per Unit $):
+50, 1.85
+100, 1.10
+250, 0.82
+500, 0.43
+1000, 0.23
+
+## Product: Paper Stickers
+
+Shapes:
+Rectangles: multiplier 1.0
+Squares: multiplier 1.0
+Circles: multiplier 1.0
+Ovals: multiplier 1.0
+Custom size: This vendor supports custom sizes with a multiplier of 2.0
+
+Sizes:
+0.79: multiplier 1.0 (default)
+1: multiplier 1.1
+1.18: multiplier 1.0
+1.26: multiplier 1.2
+1.38: multiplier 1.5
+Custom size: This vendor supports custom sizes with a multiplier of 2.0
+
+Material:
+- Gloss Paper (default)
+- Matte Paper
+
+Base Prices (Quantity,Price per Unit $):
+250, 0.40
+500, 0.21
+1000, 0.11
 """
 
+IMAGES = {
+    "Business Cards": {
+        "images": [
+            os.path.join(script_dir, "public/src/painted-edge-business-card-05.png"),
+            os.path.join(
+                script_dir,
+                "public/src/painted-edge-cards---w-brown-box-_4-4-color_.png",
+            ),
+            os.path.join(script_dir, "public/src/silk-business-card-03_1.png"),
+        ],
+        "links": [
+            "https://www.example.com/product_image.jpg",
+            "https://www.example.com/product_image.jpg",
+            "https://www.example.com/product_image.jpg",
+        ],
+    },
+    "Fabric Banners": {
+        "images": [
+            os.path.join(script_dir, "public/src/03.jpg"),
+            os.path.join(
+                script_dir,
+                "public/src/fabric-banner-01.webp",
+            ),
+            os.path.join(script_dir, "public/src/fabric-banner-02.webp"),
+        ],
+        "links": [
+            "https://www.example.com/product_image.jpg",
+            "https://www.example.com/product_image.jpg",
+            "https://www.example.com/product_image.jpg",
+        ],
+    },
+    "Bumper Stickers": {
+        "images": [
+            os.path.join(script_dir, "public/src/clear-bumper-sticker.jpg"),
+            os.path.join(
+                script_dir,
+                "public/src/copy_of_untitled_design_9_.webp",
+            ),
+            os.path.join(script_dir, "public/src/removable-bumper-stickers.jpg"),
+        ],
+        "links": [
+            "https://www.example.com/product_image.jpg",
+            "https://www.example.com/product_image.jpg",
+            "https://www.example.com/product_image.jpg",
+        ],
+    },
+    "Paper Stickers": {
+        "images": [
+            os.path.join(script_dir, "public/src/us_images_1_.webp"),
+            os.path.join(
+                script_dir,
+                "public/src/copy_of_us_images_1__1.webp",
+            ),
+            os.path.join(script_dir, "public/src/copy_of_us_images_1__1.webp"),
+        ],
+        "links": [
+            "https://www.example.com/product_image.jpg",
+            "https://www.example.com/product_image.jpg",
+            "https://www.example.com/product_image.jpg",
+        ],
+    },
+}
+
 
 def find_best_price(request):
     chat_prompt = (
-        "You are a customer assistant and you have to find the best price for the customer. "
+        "You are a customer assistant tasked with finding the best price for multiple items from different printers.\n"
         "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"
+        "For each product in request, you should:\n"
         "1. Check if the requested size is available. If not, check if vendor supports custom sizes.\n"
+        "If user hasn't specified one of the parameters (size, shape, etc), use the default one.\n"
+        "Use custom size multiplier only if user requested size and vendor supports custom sizes.\n"
         "2. Calculate the final price by:\n"
         "   - Finding the nearest lower quantity in the base price list.\n"
         "   - Multiplying by the size multiplier\n"
-        "   - Multiplying by the user requested amount. IMPORTANT. Do no multiply by nearest lower quantity!"
+        "   - Multiplying by the user requested amount. IMPORTANT. Do not multiply by nearest lower quantity!\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"
+        "User: I want 25 banners in size 6x6, 2000 business cards and 5000 circle stickers in size 6x6 on matte paper\n"
         "Answer:\n"
-        "PrintPronto: Doesn't support custom size\n"
-        "###\n"
-        "BannerLord: amount (600) * multiplier for custom size (2.0) * price per unit for nearest pack size ($0.2). Total for 600: $240\n"
-        "###\n"
-        "PrintMaster: amount (600) * multiplier for custom size (2.0) * price per unit for nearest pack size ($0.18). Total for 600: $216\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"
+        "Banners:\n"
+        "- PrintPronto: Doesn't support custom size\n"
+        "- BannerLord: amount (25) * multiplier for custom size (2.5) * price per unit for nearest pack size ($193.56). Total for 25: $12097.50\n"
+        "- PrintMaster: amount (25) * multiplier for custom size (2.2) * price per unit for nearest pack size ($194.00). Total for 25: $10670.00\n\n"
+        "Business cards:\n"
+        "- PrintPronto: amount (2000) * multiplier for default size (1.0) * price per unit for nearest pack size ($0.17). Total for 2000: $340.00\n"
+        "- BannerLord: amount (2000) * multiplier for default size (1.0) * price per unit for nearest pack size ($0.19). Total for 2000: $380.00\n"
+        "- PrintMaster: amount (2000) * multiplier for default size (1.0) * price per unit for nearest pack size ($0.18). Total for 2000: $360.00\n\n"
+        " Paper stickers:\n"
+        "- PrintPronto: Doesn't support custom size\n"
+        "- BannerLord: amount (5000) * multiplier for circle shape (1.0) * multiplier for custom size (2.0) * price per unit for nearest pack size ($0.13). Total for 5000: $1300.00\n"
+        "- PrintMaster: amount (5000) * multiplier for circle shape (1.0) * multiplier for custom size (2.0) * price per unit for nearest pack size ($0.11). Total for 5000: $1100.00\n\n"
+        "Your response should display ONLY final pricing for each available option, nothing more.\n"
+        "At the end of your response, write summary in the following format:\n"
+        "Full order quote: *all items, that user's requested*\n"
+        "- Printer1: $price of full request\n"
+        "- Printer2: $price of full request\n"
+        "- Printer3: $price of full request\n"
     )
 
     chat_response = client.chat.completions.create(
@@ -127,51 +404,82 @@ def find_best_price(request):
             {"role": "system", "content": chat_prompt},
             {"role": "user", "content": request},
         ],
+        temperature=0.5,
     )
 
     chat_text = chat_response.choices[0].message.content
+    chat_text = chat_text.replace("###", "")
+
+    images_to_use = extract_requested_products(
+        request=request, product_names=list(IMAGES.keys())
+    )
+    if images_to_use is None:
+        html_page = (
+            "<html><body><h3>Requested product is not available.</h3></body></html>"
+        )
+        chat_text = "Requested product is not available."
+        return html_page, chat_text
 
-    html_page = render_html_page(chat_text)
+    html_page = render_html_page(chat_text, images_to_use)
 
-    chat_text = chat_text.replace("###", "")
     return html_page, chat_text
 
 
-def render_html_page(chat_text):
-    formatting_prompt = (
-        "format this text for product card. It needs to be like this:\n"
-        "Vendor name 1: $price for 'amount'"
-        "###"
-        "Vendor name 2: $price for 'amount'"
-        "###"
-        "And so on for the rest of the vendors"
+def extract_total_prices(text):
+    """
+    Extracts total prices and availability information from a text block,
+    preserving the order of appearance.
+
+    Args:
+        text: The input text.
+
+    Returns:
+        A list of strings containing prices and availability messages in order.
+    """
+    results = []
+    # Combine patterns using the OR operator | and capture groups
+    pattern = (
+        r"(Total for \d+: (\$\d+(?:,\d{3})*(?:\.\d{2})?))|(Doesn't support custom size)"
     )
 
-    chat_response = client.chat.completions.create(
+    matches = re.findall(pattern, text)
+
+    for match in matches:
+        # Check which group matched
+        if match[0]:  # Price match
+            results.append(match[1])
+        elif match[2]:  # Availability match
+            results.append(match[2])
+    return results
+
+
+def extract_requested_products(request, product_names):
+    product_list = "\n".join(f"- {product}" for product in product_names)
+    prompt = (
+        "Here are the available product names:\n"
+        f"{product_list}\n"
+        "Identify any products explicitly or implicitly mentioned in the user's request. "
+        "If no products match, respond with 'None'. Output the result as a Python list. "
+        "Your response should contain only the list or 'None'."
+    )
+
+    response = client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
-            {"role": "system", "content": formatting_prompt},
-            {"role": "user", "content": chat_text},
+            {"role": "system", "content": prompt},
+            {"role": "user", "content": request},
         ],
     )
-    formatted_text = chat_response.choices[0].message.content
-    images = [
-        os.path.join(script_dir, "public/src/painted-edge-business-card-05.png"),
-        os.path.join(
-            script_dir, "public/src/painted-edge-cards---w-brown-box-_4-4-color_.png"
-        ),
-        os.path.join(script_dir, "public/src/silk-business-card-03_1.png"),
-    ]
-    product_links = [
-        "https://www.example.com/product_image.jpg",
-        "https://www.example.com/product_image.jpg",
-        "https://www.example.com/product_image.jpg",
-    ]
+    response_text = response.choices[0].message.content
+    return eval(response_text)
 
-    # Split chat_text into product descriptions
-    product_descriptions = formatted_text.split("###")
 
-    # Generate HTML content
+def render_html_page(chat_text, images_to_use):
+    if images_to_use is None:
+        return "<html><body><h3>Requested product is not available.</h3></body></html>"
+
+    # product_descriptions = chat_text.split("\n")
+
     html_content = """
     <html>
         <head>
@@ -185,7 +493,6 @@ def render_html_page(chat_text):
                     text-align: center;
                     width: 200px;
                     display: inline-block;
-                    vertical-align: top;
                 }
                 .product-card img {
                     max-width: 100%;
@@ -206,30 +513,30 @@ def render_html_page(chat_text):
         </head>
         <body>
     """
-
-    for i, description in enumerate(product_descriptions):
-        if (
-            "Doesn't support" in description.lower()
-            or "Not available" in description.lower()
-        ):
-            continue
-        image = images[i % len(images)]
+    processed_images = [
+        image for img in images_to_use for image in IMAGES[img]["images"]
+    ]
+    processed_links = [link for img in images_to_use for link in IMAGES[img]["links"]]
+    print(chat_text)
+    processed_prices = extract_total_prices(chat_text)
+    print(processed_prices)
+    # print(product_descriptions)
+    for i in range(len(processed_images)):
+        print(i)
+        image = processed_images[i]
         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>
+            <p>{COMPANIES[i % 3]+': '+processed_prices[i]}</p>
+            <a href="{processed_links[i]}" target="_blank"><button>Buy Now</button></a>
+            
         </div>
         """
 
-    html_content += """
-        </body>
-    </html>
-    """
+    html_content += "</body></html>"
     return html_content
 
 
@@ -247,7 +554,7 @@ iface = gr.Interface(
     ],
     title="Get Instant Quote",
     description=f"""
-    <div style="display: flex; justify-content: center; align-items: center; text-align: center;  margin-bottom: 20px;">
+    <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>
     """,
@@ -256,7 +563,7 @@ iface = gr.Interface(
 
 
 def auth_function(username, password):
-    valid_users = {"admin": "demo4anthony1", "user1": "pass456"}
+    valid_users = {"admin": "demo4anthony1", "user": "strongpassword123"}
     return username in valid_users and valid_users[username] == password