quote_proposal.py

import json
import os
from uuid import uuid4

from abstra.tasks import send_task
from abstra.ai import prompt
from abstra.common import get_persistent_dir
from abstra.forms import ListItemSchema, Page

persistent_dir = get_persistent_dir()
save_image_path = persistent_dir / "estimate.png"


page = (
    Page()
    .read("Name", key="name", required=True)
    .read_email("E-mail", key="email", required=True)
    .read("Company", key="company", required=True)
    .read_file("Upload image", key="image", required=True)
    .display_file("example_quote.jpg", download_text="Or use this example image")
    .run()
)


# OCR
transcription = prompt(
    [
        """This image in base64 format is a list of products, you must transcribe the text from the image, extracting the name of each product and the amount of each product. The final output should be a list with square brackets where each item is separated by commas, and each item should follow the following sample of a json dictionary.json dictionary sample: {"product": "sample", "quantity": "sample"}. The quantities of the products can not be negative.""",
        page["image"],
    ],
    format={
        "products": {
            "type": "string",
            "description": "list of products separated by commas, delimited by square brackets, each item following the sample json pattern.",
        }
    },
    temperature=0.3,
)

ocr = json.loads(transcription["products"])


# Check page
item = (
    ListItemSchema()
    .read("Product", key="product")
    .read_number("Quantity", key="quantity")
)
check_stage = (
    Page()
    .display(
        "Please, check the products and quantities extracted from the image.",
        size="medium",
    )
    .read_list(item, initial_value=ocr, key="check", required=True, min=1)
    .run()
)

proposal = check_stage["check"]

for p in proposal:
    p["id"] = str(uuid4())
    p["price"] = ""
    p["database_match"] = ""
    p["product_id"] = ""

finance_email = os.getenv("FINANCE_EMAIL")
print(f"finance email é {finance_email}")

send_task(
    "proposal_data", 
    {
        "proposal": proposal,
        "email": page["email"],
        "company": page["company"].title(),
        "name": page["name"].title(),
        "finance_email": finance_email,
    },
)