본문 바로가기
IT 프로그래밍 관련/streamlit 대시보드

streamlit pdf file을 화면 출력하도록 변환하는 함수

by 지나는행인 2021. 3. 8.
728x90

이 함수는 pdf파일을 가져와서 화면에 출력 될 수 있도록 PdfFileReader를 통하여 읽어들여,

 

변환하여 출력하는 함수이다.

import streamlit as st
from PIL import Image    
import pandas as pd       
from PyPDF2 import PdfFileReader
import os



def read_pdf(pdf_file):
    pdfreader = PdfFileReader(pdf_file)
    count = pdfreader.numPages
    text = ''
    for i in range(count) :
        page = pdfreader.getPage(i)
        text = text+page.extractText()
    return text

 라이브러리로 from PyPDF2 import PdfFileReader를 잊지말고 인스톨하여 하도록한다.

 

상기 예제의 호출로는 

 

text = read_pdf(doc_file)                   

 

## doc_file = doc_file = st.file_uploader('Upload pdf of txt', type=['pdf', 'txt'])

 

st.write(text)                  ##화면에 표기

 

를 사용하였다.

댓글