added examples

algorithm/product.py

@@ -1,8 +1,7 @@
 class Product:
-    def __init__(self, name, price, quantity):
+    def __init__(self, name, price):
         self.name = name
         self.price = price
-        self.quantity = quantity
 
     def __repr__(self):
-        return f"Product(name={self.name}, price={self.price}, quantity={self.quantity})"
+        return f"Product(name={self.name}, price={self.price})"

algorithm/receipt.py

@@ -1,8 +0,0 @@
-class Receipt:
-    def __init__(self, products, tax, total_amount):
-        self.products = products
-        self.tax = tax
-        self.total_amount = total_amount
-
-    def __repr__(self):
-        return f"Receipt(products={self.products}, tax={self.tax}, total_amount={self.total_amount})"

algorithm/receipt_calculation.py

@@ -93,21 +93,21 @@ def calculate_tips_and_taxes(items_table, total_amount, tax, tips):
     if items_table[0][0] == "No items":
         return products, 0
 
-    if total_amount == "Not specified" or total_amount is None:
+    if total_amount == "Not specified" or total_amount == "unknown" or total_amount is None:
         total_amount = "0.0"
 
-    if tax == "Not specified" or tax is None:
+    if tax == "Not specified" or tax == "unknown" or tax is None:
         tax = "0.0"
 
-    if tips == "Not specified" or tips is None:
-        tax = "0.0"
+    if tips == "Not specified" or tips == "unknown" or tips is None:
+        tips = "0.0"
 
     for item in items_table:
         price = item[5]
-        if price == "Not specified":
+        if price == "Not specified" or price == "unknown":
             price = "0.0"
         item_value = float(str(price).replace(",", ".")) if item[5] is not None else 0.0
-        products.append(Product(item[0], item_value, item[3]))
+        products.append(Product(item[0], item_value))
 
     sum_of_product_prices = 0
     for _product in products:

app.py

@@ -14,6 +14,10 @@ from vertex_ai_service import VertexAIService
 model_names = ["gemini-1.5-flash", "gemini-1.5-pro", "gpt-4o-mini", "gpt-4o", "gpt-4-turbo"]
 prompt_names = ["prompt_v1", "prompt_v2", "prompt_v3"]
 example_list = [["./examples/" + example] for example in os.listdir("examples")]
+example_list_us = [["./examples_us/" + example] for example in os.listdir("examples_us")]
+example_list_canada = [["./examples_canada/" + example] for example in os.listdir("examples_canada")]
+example_france = [["./examples_france/" + example] for example in os.listdir("examples_france")]
+
 prompt_default = read_prompt_from_file("prompt_v1.txt")
 system_instruction = read_prompt_from_file("system_instruction.txt")
 
@@ -258,6 +262,19 @@ with gr.Blocks() as iface:
     )
     prompt_radio.change(fn=update_prompt_from_radio, inputs=[prompt_radio], outputs=[custom_prompt])
     gr.Examples(examples=example_list,
-                inputs=[image_input, model_radio, prompt_radio, temperature_slider, custom_prompt])
+                inputs=[image_input, model_radio, prompt_radio, temperature_slider, custom_prompt],
+                label="Examples for general use")
+
+    gr.Examples(examples=example_list_us,
+                inputs=[image_input, model_radio, prompt_radio, temperature_slider, custom_prompt],
+                label="Examples for US")
+
+    gr.Examples(examples=example_list_canada,
+                inputs=[image_input, model_radio, prompt_radio, temperature_slider, custom_prompt],
+                label="Examples for Canada")
+
+    gr.Examples(examples=example_france,
+                inputs=[image_input, model_radio, prompt_radio, temperature_slider, custom_prompt],
+                label="Examples for France")
 
-iface.launch(share=True)
+iface.launch()

convert_to_webp.py

@@ -0,0 +1,30 @@
+from PIL import Image
+import os
+
+
+def convert_to_webp(input_file, output_file=None):
+    try:
+        img = Image.open(input_file)
+        if output_file is None:
+            base_name = os.path.splitext(input_file)[0]
+            output_file = f"{base_name}.webp"
+
+        img.save(output_file, "WEBP")
+        print(f"saved as {output_file}")
+
+    except Exception as e:
+        print(f"error saving: {e}")
+
+
+def process_images_in_directory(directory):
+    print(directory)
+    for filename in os.listdir(directory):
+        if filename.lower().endswith(('.jpg', '.png', '.jpeg')):
+            input_path = os.path.join(directory, filename)
+            convert_to_webp(input_path)
+
+if __name__ == "__main__":
+    current_directory = os.getcwd()
+    print(f"current_directory: {current_directory}")
+    directory = "./examples_france/"
+    process_images_in_directory(directory)