updated comparing results

app.py

@@ -36,6 +36,7 @@ invoice_json_path = f"{data_folder}/invoices.json"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 with open(f"{data_folder}/car_parts.json", "r") as f:
     car_parts = json.load(f)
+    print("Successfully loaded car parts")
 
 faiss_index_path = f"{data_folder}/invoice_index.faiss"
 print("Index file size:", os.path.getsize(faiss_index_path))

src/pipelines.py

@@ -47,6 +47,7 @@ class InvoiceGenerator:
 
     @staticmethod
     def check_within_range(generated_invoice, car_parts):
+        print("Comparing results")
         def get_part_info(part_name, car_parts):
             part_name = part_name.lower()
             max_match = [None, 0]
@@ -62,13 +63,14 @@ class InvoiceGenerator:
         last_cost_line = -2 if all_lines[-1] == '' else -1
         lines = generated_invoice.split("\n")[first_cost_line:last_cost_line]
         cost_lines = [[line.strip() for line in cost_line.split("|")] for cost_line in lines]
+        start_index = 0 if cost_lines[0][0] != "" else 1
 
         comparing_results = {}
 
         for line in cost_lines:
-            part = line[0]
-            cost = line[1]
-            hours = line[2]
+            part = line[start_index]
+            cost = line[start_index + 1]
+            hours = line[start_index + 2]
             found_part = get_part_info(part, car_parts)
             if found_part:
                 comparing_results[part] = {

src/prompts.py

@@ -58,15 +58,3 @@ Be thorough and specific, covering all visible damage in the image.
 Write no more than 3-4 sentences describing the damage in detail.
 Do not add any additional information beyond the visible damage in the image and any comments.
 """
-
-
-ESTIMATE_COST_OF_CAR_PARTS_REPLACEMENT_PROMPT = """
-I will provide you with a list of car parts that need to be replaced due to damage in a car accident (in German).
-Your task is to estimate the cost of each part in Euros (EUR) based on the damage description provided.
-Also, you need to estimate approximate labor hours required to replace each part.
-There may be duplicate parts in the list (for example, for right and left sides of the car).
-You may also expand the list with new parts which are not in the list but may be necessary for the repair.
-Delete any duplicate entries and provide the cost and labor hours for one side only.
-Sort the parts in the list in ascending order based on name of each part (in alphabetical order).
-Your output should be a python dictionary where the keys are the names of the parts and the values are lists containing the estimated cost and labor hours for each part.
-"""