fixed invoices display

app.py

@@ -81,42 +81,27 @@ def get_similar_invoice(invoice_name):
 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
-    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)
+    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"] = 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)
+    except Exception as e:
+        st.write("Could not generate invoice, please try again")
+        print(e)
+        return None
 
 
 st.title("Upload FNOL photo")
@@ -161,4 +146,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.md(get_similar_invoice(st.session_state["invoice_path"]))
+        st.markdown(get_similar_invoice(st.session_state["invoice_path"]))

requirements.txt

@@ -15,4 +15,5 @@ streamlit==1.40.1
 fuzzywuzzy==0.18.0
 openai==1.55.3
 pandas==2.2.3
-pypdf==5.1.0
+pypdf==5.1.0
+tqdm==4.67.1

src/pipelines.py

@@ -93,6 +93,7 @@ class InvoiceGenerator:
         total_cost_line = all_lines[last_cost_line]
         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]
+        print(f"Cost lines: \n{cost_lines}\n")
         costs = [int(line[1]) + int(line[2]) * int(line[3]) for line in cost_lines]
         cost_lines = list(map(lambda x, y: [x[0], x[1], x[2], x[3], str(y)], cost_lines, costs))
         total_cost = sum(costs)
@@ -137,6 +138,7 @@ class InvoiceGenerator:
         generated_invoice = self.model.generate_message_from_image(
             GENERATE_INVOICE_PROMPT(invoice_info, detailed_damage_description), image_path
         ).replace("```markdown", "").replace("```", "")
+        print(f"Generated invoice: \n{generated_invoice}\n")
         generated_invoice = self.check_calculations(generated_invoice)
 
         result["generated_invoice"] = generated_invoice