latex

app.py

@@ -7,7 +7,7 @@ from invoice_generator import generate_invoice
 
 st.set_page_config(page_title="Invoice generator", layout="wide")
 output_folder = "output"
-template = f"{output_folder}/template.tex"
+template = "template.tex"
 
 
 def get_image_from_pdf(pdf_path):
@@ -20,9 +20,11 @@ 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: {output_folder}/{output_pdf}")
-    return get_image_from_pdf(f"{output_folder}/{output_pdf}")
+    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)
 
 
 st.title("Upload FNOL photo")
@@ -32,22 +34,34 @@ col1, col2, col3 = st.columns([4, 1, 4])
 with col1:
     uploaded_image = st.file_uploader("Upload photo", type=["jpg", "jpeg", "png"])
     if uploaded_image:
-        image = Image.open(uploaded_image)
-        image_path = f"{output_folder}/{str(uuid4())[:5]}.png"
-        image.save(image_path)
-        print(f"Image: {image_path}")
-        st.image(image, caption="Uploaded photo", width=300)
-        st.query_params["image"] = image_path
+        try:
+            image = Image.open(uploaded_image)
+            image_path = f"{output_folder}/{str(uuid4())[:5]}.png"
+            image.save(image_path)
+            print(f"Image: {image_path}")
+            st.image(image, caption="Uploaded photo", width=300)
+            st.query_params["image"] = image_path
+        except Exception as e:
+            st.write(f"Coudn't load image: {e}")
 
 with col2:
     if st.query_params.get("image"):
         if st.button("Generate invoice"):
             with st.spinner("Generating..."):
-                other_image = display_invoice(st.query_params["image"])
+                st.session_state["other_image"] = display_invoice(st.query_params["image"])
                 st.query_params["status"] = "loaded"
     else:
         st.button("Generate invoice", disabled=True)
+    if st.query_params.get("generated_pdf"):
+        with open(st.query_params["generated_pdf"], "rb") as f:
+            file_data = f.read()
+        st.download_button(
+            label="Download invoice",
+            data=file_data,
+            file_name="generated_invoice.pdf",
+            mime="application/pdf"
+        )
 
 with col3:
     if st.query_params.get("status") == "loaded":
-        st.image(other_image, caption="Generated invoice", use_column_width=True)
+        st.image(st.session_state["other_image"], caption="Generated invoice", use_column_width=True)

invoice_generator.py

@@ -121,6 +121,8 @@ def compile_latex(invoice_table, template, output_folder, output_pdf_name):
     try:
         subprocess.run(
             [
+                # "xelatex",
+                # "-interaction=batchmode",
                 "pdflatex",
                 "-interaction=nonstopmode",
                 "-output-directory",

requirements.txt

@@ -2,4 +2,5 @@ streamlit
 pillow
 tqdm
 PyMuPDF
-openai
+openai
+texlive-full