latex

app.py

@@ -21,10 +21,18 @@ def get_image_from_pdf(pdf_path):
 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}"
-    generate_invoice(image_path, output_pdf, template, output_folder)
-    print(f"Generated invoice: {path_to_output_pdf}")
-    st.query_params["generated_pdf"] = path_to_output_pdf
-    return get_image_from_pdf(path_to_output_pdf)
+    try:
+        status = generate_invoice(image_path, output_pdf, template, output_folder)
+        if status == 0:
+            st.write("Image is irrelevant, upload another one")
+            return
+        print(f"Generated invoice: {path_to_output_pdf}")
+        st.query_params["generated_pdf"] = path_to_output_pdf
+        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")
@@ -48,8 +56,9 @@ with col2:
     if st.query_params.get("image"):
         if st.button("Generate invoice"):
             with st.spinner("Generating..."):
-                st.session_state["other_image"] = display_invoice(st.query_params["image"])
-                st.query_params["status"] = "loaded"
+                st.session_state["invoice"] = display_invoice(st.query_params["image"])
+                if st.session_state["invoice"]:
+                    st.query_params["status"] = "loaded"
     else:
         st.button("Generate invoice", disabled=True)
     if st.query_params.get("generated_pdf"):
@@ -64,4 +73,4 @@ with col2:
 
 with col3:
     if st.query_params.get("status") == "loaded":
-        st.image(st.session_state["other_image"], caption="Generated invoice", use_column_width=True)
+        st.image(st.session_state["invoice"], caption="Generated invoice", use_column_width=True)

invoice_generator.py

@@ -228,7 +228,7 @@ def generate_invoice(image_path, output_file, template, output_folder):
     description = generate_accident_description(client, image_path)
     if not description:
         print(f"Image {image_path} is irrelevant.")
-        return
+        return 0
 
     assert "damage_description" in description, "damage_description not found"
     assert "accident_story" in description, "accident_story not found"
@@ -242,3 +242,4 @@ def generate_invoice(image_path, output_file, template, output_folder):
     complete_pdf(
         f"{output_folder}/{output_file}", image_path, description["accident_story"]
     )
+    return 1