1.0

.history/app_20220620163100.py

@@ -0,0 +1,51 @@
+#-*- coding : utf-8-*-
+import os,subprocess,base64
+from subprocess import STDOUT #os process manipuation
+os.system("apt-get update")
+os.system("apt-get install sudo")
+os.system("sudo apt update")
+os.system("apt-get install -y libgl1-mesa-glx")
+os.system("apt install ghostscript python3-tk")
+
+import streamlit as st
+# @st.cache
+# def gh():
+#     """install ghostscript on the linux machine"""
+    
+#     proc = subprocess.Popen('apt-get update', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
+#     proc = subprocess.Popen('apt-get install sudo', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
+#     proc = subprocess.Popen('sudo apt update', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
+#     proc = subprocess.Popen('apt install ghostscript python3-tk', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
+#     proc = subprocess.Popen('apt-get install -y libgl1-mesa-glx', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
+#     proc.wait()
+# gh()
+import pandas as pd 
+import camelot as cam # extracting tables from PDFs 
+
+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")
+       
+       
+

app.py

@@ -8,17 +8,17 @@ os.system("apt-get install -y libgl1-mesa-glx")
 os.system("apt install ghostscript python3-tk")
 
 import streamlit as st
-@st.cache
-def gh():
-    """install ghostscript on the linux machine"""
+# @st.cache
+# def gh():
+#     """install ghostscript on the linux machine"""
     
-    proc = subprocess.Popen('apt-get update', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
-    proc = subprocess.Popen('apt-get install sudo', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
-    proc = subprocess.Popen('sudo apt update', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
-    proc = subprocess.Popen('apt install ghostscript python3-tk', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
-    proc = subprocess.Popen('apt-get install -y libgl1-mesa-glx', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
-    proc.wait()
-gh()
+#     proc = subprocess.Popen('apt-get update', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
+#     proc = subprocess.Popen('apt-get install sudo', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
+#     proc = subprocess.Popen('sudo apt update', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
+#     proc = subprocess.Popen('apt install ghostscript python3-tk', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
+#     proc = subprocess.Popen('apt-get install -y libgl1-mesa-glx', shell=True, stdin=None, stdout=open(os.devnull,"wb"), stderr=STDOUT, executable="/bin/bash")
+#     proc.wait()
+# gh()
 import pandas as pd 
 import camelot as cam # extracting tables from PDFs