updated app

Dockerfile

@@ -1,10 +1,8 @@
 FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
 
-ENV DEBIAN_FRONTEND=noninteractive
 
 RUN apt-get update && apt-get install -y \
-    python3 python3-pip python3-venv build-essential curl git libvips-dev \
-    weasyprint tzdata \
+    python3 python3-pip python3-venv build-essential curl git \
     && rm -rf /var/lib/apt/lists/*
 
 RUN pip install --no-cache-dir --upgrade pip \
@@ -20,7 +18,7 @@ COPY pyproject.toml ./
 
 RUN poetry install --no-interaction --no-ansi \
     && pip install --no-build-isolation flash-attn \
-    && pip install "rerankers[all]" \
+    && pip install "rerankers[all]"
 
 RUN useradd -m -u 1000 user
 USER user

app.py

@@ -7,7 +7,6 @@ import fitz
 import json
 from src.pipelines import InvoiceGenerator
 
-print("HERE")
 
 st.set_page_config(page_title="Invoice generator", layout="wide")
 output_folder = "output"
@@ -39,26 +38,30 @@ def get_image_from_pdf(pdf_path):
     img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
     return img
 
+def read_markdown_file(file_path):
+    with open(file_path, "r") as f:
+        return f.read()
+
 
 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}"
+    output_md = "invoice_" + os.path.basename(image_path).split(".")[0] + ".md"
+    path_to_output_md = f"{output_folder}/{output_md}"
     try:
         result = pipeline.generate_invoice(
-            image_path=image_path, output_path=path_to_output_pdf, car_parts=car_parts
+            image_path=image_path, output_path=path_to_output_md, 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
+        print(f"Generated invoice: {path_to_output_md}")
+        st.session_state["generated_md"] = path_to_output_md
         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)
+        return read_markdown_file(path_to_output_md)
     except Exception as e:
         st.write("Could not generate invoice, please try again")
         print(e)
@@ -103,7 +106,8 @@ with col2:
 
 with col3:
     if st.session_state.get("status") == "loaded":
-        st.image(st.session_state["invoice"], caption="Generated invoice", use_container_width=True)
+        # st.image(st.session_state["invoice"], caption="Generated invoice", use_container_width=True)
+        st.markdown(st.session_state["invoice"])
         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']}")

src/pipelines.py

@@ -2,7 +2,6 @@ from src.RAG import RAG
 from src.model import Pixtral
 from src.prompts import GENERATE_INVOICE_PROMPT, GENERATE_BRIEF_DAMAGE_DESCRIPTION_PROMPT, \
     GENERATE_DETAILED_DAMAGE_DESCRIPTION_PROMPT
-from md2pdf.core import md2pdf
 from rerankers import Reranker
 import re
 from fuzzywuzzy import fuzz
@@ -39,7 +38,10 @@ class InvoiceGenerator:
         with open(template_path, "r") as f:
             md_text = f.read()
         md_text = md_text.replace(r"<<table>>", generated_invoice)
-        md2pdf(output_path, md_content=md_text)
+        with open(output_path, "w") as f:
+            f.write(md_text)
+        #  md2pdf(output_path, md_content=md_text)
+        return md_text
 
     @staticmethod
     def check_within_range(generated_invoice, car_parts):
@@ -143,7 +145,6 @@ class InvoiceGenerator:
             print(comparing_results)
 
         if output_path:
-            self.format_invoice(generated_invoice=generated_invoice, output_path=output_path,
-                                template_path=template_path)
+            self.format_invoice(generated_invoice=generated_invoice, output_path=output_path, template_path=template_path)
 
         return result