From 3be7393900793646540fb3c497a528d86679bcc2 Mon Sep 17 00:00:00 2001
From: fengyu6623 <fengyu6623@devide>
Date: Wed, 31 Jan 2024 15:57:34 +0800
Subject: [PATCH] Auto Commit

---
 main.py | 80 +++++++++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 64 insertions(+), 16 deletions(-)

diff --git a/main.py b/main.py
index fb9ab94..9e72a54 100644
--- a/main.py
+++ b/main.py
@@ -1,24 +1,72 @@
 import streamlit as st
-from streamlit_option_menu import option_menu
+import inscode
 
-# 设置Streamlit应用程序的标题
-st.set_page_config(page_title="app name", layout="wide")
+st.set_page_config(page_title="PDF Extractor", page_icon=":books:")
 
-menu1="菜单1"
-menu2="菜单2"
+st.title("PDF Extractor")
 
-with st.sidebar:
-    menu = option_menu("菜单", [menu1, menu2], 
-        icons=['house', "list-task"], 
-        menu_icon="cast", default_index=0)
+st.sidebar.title("功能列表")
 
-def main():
+menu = ['PDF文件上传', '文字提取与内容总结', '问答式提取文档信息']
+choice = st.sidebar.selectbox('选择一个功能', menu)
 
-    if menu == menu1:
-        st.subheader(f"{menu1}")
+if choice == 'PDF文件上传':
 
-    if menu == menu2:
-        st.subheader(f"{menu2}")
+    st.subheader('上传PDF文件')
 
-if __name__ == '__main__':
-    main()
+    uploaded_file = st.file_uploader("选择一个PDF文件",type=['pdf'])
+    
+    if uploaded_file is not None:
+        
+        st.success("PDF文件上传成功！")
+        
+        show_pdf = st.checkbox('显示PDF文件内容')
+        
+        if show_pdf:
+            st.write(uploaded_file)
+            
+elif choice == '文字提取与内容总结':
+
+    st.subheader('文字提取与内容总结')
+
+    uploaded_file = st.file_uploader("选择一个PDF文件",type=['pdf'])
+    
+    if uploaded_file is not None:
+        
+        st.success("PDF文件上传成功！")
+        
+        text = inscode.ocr('pdf_file', uploaded_file)
+        
+        st.write(text)
+
+        summarize = inscode.ai('summarize', text)
+        
+        st.write(summarize)
+        
+elif choice == '问答式提取文档信息':
+    
+    st.subheader('问答式提取文档信息')
+    
+    uploaded_file = st.file_uploader("选择一个PDF文件",type=['pdf'])
+
+    if uploaded_file is not None:
+        
+        st.success("PDF文件上传成功！")
+        
+        text = inscode.ocr('pdf_file', uploaded_file)
+        
+        summarize = inscode.ai('summarize', text)
+        
+        st.write('文章内容总结：')
+        
+        st.write(summarize)
+        
+        prompt = st.text_input('请提出一个问题')
+        
+        if prompt:
+            
+            answer = inscode.ai(prompt, text)
+            
+            st.write(f'问题： {prompt}')
+            
+            st.write(f'回答： {answer}')
\ No newline at end of file
-- 
GitLab