MMOON commited on
Commit
2bd789e
·
verified ·
1 Parent(s): 20519d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -6,9 +6,9 @@ import fitz # PyMuPDF
6
  import plotly.express as px
7
  import io
8
 
9
- def extract_data_from_pdf(pdf_path):
10
- # Open the PDF file
11
- pdf_document = fitz.open(pdf_path)
12
  extracted_texts = [page.get_text() for page in pdf_document]
13
  pdf_document.close()
14
 
@@ -86,8 +86,11 @@ def main():
86
  uploaded_file = st.file_uploader("Choisir un fichier PDF", type="pdf")
87
 
88
  if uploaded_file is not None:
 
 
 
89
  # Process data
90
- data = extract_data_from_pdf(uploaded_file)
91
  results_df = analyze_sterilization(data)
92
 
93
  # Display results
 
6
  import plotly.express as px
7
  import io
8
 
9
+ def extract_data_from_pdf(pdf_bytes):
10
+ # Open the PDF file from bytes
11
+ pdf_document = fitz.open(stream=pdf_bytes, filetype="pdf")
12
  extracted_texts = [page.get_text() for page in pdf_document]
13
  pdf_document.close()
14
 
 
86
  uploaded_file = st.file_uploader("Choisir un fichier PDF", type="pdf")
87
 
88
  if uploaded_file is not None:
89
+ # Read the uploaded file as bytes
90
+ pdf_bytes = uploaded_file.getvalue()
91
+
92
  # Process data
93
+ data = extract_data_from_pdf(pdf_bytes)
94
  results_df = analyze_sterilization(data)
95
 
96
  # Display results