fixed invoices display

app.py

@@ -68,30 +68,55 @@ def read_markdown_file(file_path):
     with open(file_path, "r") as f:
         return f.read()
 
+def get_similar_invoice(invoice_name):
+    try:
+        with open(invoice_json_path, "r") as f:
+            invoice_json = json.load(f)
+        return invoice_json[invoice_name]
+    except Exception as e:
+        print(e)
+        return ""
+
 
 def display_invoice(image_path):
     output_pdf = "invoice_" + os.path.basename(image_path).split(".")[0] + ".pdf"
     path_to_output_pdf = f"{output_folder}/{output_pdf}"
-    try:
-        result = pipeline.generate_invoice(
-            image_path=image_path, output_path=path_to_output_pdf, car_parts=car_parts
-        )
-        if result is None:
-            st.write("Image is irrelevant, upload another one")
-            st.session_state["status"] = "irrelevant"
-            return
-        print(f"Generated invoice: {path_to_output_pdf}")
-        st.session_state["generated_pdf"] = path_to_output_pdf
-        st.session_state["invoice_info"] = result["invoice_info"]
-        st.session_state["invoice_path"] = f"{data_folder}/invoices/{result['invoice_path']}"
-        st.session_state["similar_image"] = f"{data_folder}/images/{result['similar_image']}"
-        st.session_state["damage_description"] = result["damage_description"]
-        st.session_state["detailed_damage_description"] = result["detailed_damage_description"]
-        return get_image_from_pdf(path_to_output_pdf)
-    except Exception as e:
-        st.write("Could not generate invoice, please try again")
-        print(e)
-        return None
+    # try:
+    #     result = pipeline.generate_invoice(
+    #         image_path=image_path, output_path=path_to_output_pdf, car_parts=car_parts
+    #     )
+    #     if result is None:
+    #         st.write("Image is irrelevant, upload another one")
+    #         st.session_state["status"] = "irrelevant"
+    #         return
+    #     print(f"Generated invoice: {path_to_output_pdf}")
+    #     st.session_state["generated_pdf"] = path_to_output_pdf
+    #     st.session_state["invoice_info"] = result["invoice_info"]
+    #     st.session_state["invoice_path"] = f"{data_folder}/invoices/{result['invoice_path']}"
+    #     st.session_state["similar_image"] = f"{data_folder}/images/{result['similar_image']}"
+    #     st.session_state["damage_description"] = result["damage_description"]
+    #     st.session_state["detailed_damage_description"] = result["detailed_damage_description"]
+    #     return get_image_from_pdf(path_to_output_pdf)
+    # except Exception as e:
+    #     st.write("Could not generate invoice, please try again")
+    #     print(e)
+    #     return None
+    result = pipeline.generate_invoice(
+        image_path=image_path, output_path=path_to_output_pdf, car_parts=car_parts
+    )
+    if result is None:
+        st.write("Image is irrelevant, upload another one")
+        st.session_state["status"] = "irrelevant"
+        return
+    print(f"Generated invoice: {path_to_output_pdf}")
+    st.session_state["generated_pdf"] = path_to_output_pdf
+    st.session_state["invoice_info"] = result["invoice_info"]
+    st.session_state["invoice_path"] = result['invoice_path']
+    print(f"Invoice path: {result['invoice_path']}")
+    st.session_state["similar_image"] = f"{data_folder}/images/{result['similar_image']}"
+    st.session_state["damage_description"] = result["damage_description"]
+    st.session_state["detailed_damage_description"] = result["detailed_damage_description"]
+    return get_image_from_pdf(path_to_output_pdf)
 
 
 st.title("Upload FNOL photo")
@@ -136,4 +161,4 @@ with col3:
         st.image(st.session_state["similar_image"], caption="Similar accident", width=300)
         st.write(f"Detailed damage description: {st.session_state['detailed_damage_description']}")
         st.write(f"Damage description: {st.session_state['damage_description']}")
-        st.image(get_image_from_pdf(st.session_state["invoice_path"]), caption="Invoice of similar accident type", use_container_width=True)
+        st.md(get_similar_invoice(st.session_state["invoice_path"]))

src/pipelines.py

@@ -72,10 +72,10 @@ class InvoiceGenerator:
             found_part = get_part_info(part, car_parts)
             if found_part:
                 comparing_results[part] = {
-                    "cost_within_range": car_parts[found_part]["cost_range"][0] <= float(cost) <=
-                                         car_parts[found_part]["cost_range"][1],
-                    "hours_within_range": car_parts[found_part]["hours_range"][0] <= float(hours) <=
-                                          car_parts[found_part]["hours_range"][1],
+                    "cost_within_range": car_parts[found_part]["cost_min"] <= float(cost) <=
+                                         car_parts[found_part]["cost_max"],
+                    "hours_within_range": car_parts[found_part]["hours_min"] <= float(hours) <=
+                                          car_parts[found_part]["hours_max"],
                     "cost_diff": float(cost) - car_parts[found_part]["average_cost"],
                     "hours_diff": float(hours) - car_parts[found_part]["average_hours"],
                     "part_info": found_part