blessedpug commited on
Commit
8c35be2
·
1 Parent(s): e0d5a1d

pdf viewer update

Browse files
Files changed (1) hide show
  1. pipeline.py +6 -5
pipeline.py CHANGED
@@ -1,7 +1,7 @@
1
  import openai
2
  from dotenv import load_dotenv
3
  from io import BytesIO
4
- import os
5
  from PIL import Image
6
  import base64
7
  import json
@@ -137,12 +137,13 @@ def extract_child_fee_info(img_input, emp_name, emp_code, department):
137
  if items and "bill_month" in items[0]:
138
  bill_month = items[0]["bill_month"]
139
 
140
- # Use a temp file for output so Gradio can return it
141
- import tempfile
142
- temp = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
 
143
  filled_pdf_path = fill_child_fee_pdf(
144
  template_pdf_path="CHILD FEE REIMBURSEMENT FORM.pdf",
145
- output_pdf_path=temp.name,
146
  emp_name=emp_name,
147
  emp_code=emp_code,
148
  department=department,
 
1
  import openai
2
  from dotenv import load_dotenv
3
  from io import BytesIO
4
+ import os, uuid
5
  from PIL import Image
6
  import base64
7
  import json
 
137
  if items and "bill_month" in items[0]:
138
  bill_month = items[0]["bill_month"]
139
 
140
+ os.makedirs("outputs", exist_ok=True)
141
+ output_pdf_path = f"outputs/filled_child_fee_form_{uuid.uuid4().hex}.pdf"
142
+
143
+
144
  filled_pdf_path = fill_child_fee_pdf(
145
  template_pdf_path="CHILD FEE REIMBURSEMENT FORM.pdf",
146
+ output_pdf_path=output_pdf_path,
147
  emp_name=emp_name,
148
  emp_code=emp_code,
149
  department=department,