#-*- coding : utf-8-*-

import pandas as pd 
import streamlit as st
import base64
import subprocess # process in the os
from subprocess import STDOUT #os process manipuation
import os
import camelot as cam # extracting tables from PDFs 

os.system('apt-get install -y ghostscript')
st.title("PDF Table Extractor")

input_pdf = st.file_uploader(label = "", type = 'pdf')

page_number = st.text_input("请填写表格所在PDF页码，eg: 3", value = 1)

if input_pdf is not None:
    # byte object into a PDF file 
    with open("input.pdf", "wb") as f:
        base64_pdf = base64.b64encode(input_pdf.read()).decode('utf-8')
        f.write(base64.b64decode(base64_pdf))
    f.close()

    # read the pdf and parse it using stream
    tables = cam.read_pdf("input.pdf", pages=page_number)
    result = pd.ExcelWriter('result.xlsx', engine='xlsxwriter') 
    tables[0].to_excel(result,index=False) 
    # for i in range(0,len(tables)):
    #     table = tables[i].df
    #     sheetname = str(i)
    #     table.to_excel(result, sheetname,index=False) 

    with open('result.xlsx','rb') as f:
       st.download_button('提取完成，点击下载！', f,file_name='result.xlsx',mime="application/vnd.ms-excel")