Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
奋斗小菠萝
ChatPDF
提交
3be73939
C
ChatPDF
项目概览
奋斗小菠萝
/
ChatPDF
与 Fork 源项目一致
Fork自
inscode / Streamlit
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
C
ChatPDF
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
3be73939
编写于
1月 31, 2024
作者:
F
fengyu6623
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Auto Commit
上级
0520122c
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
64 addition
and
16 deletion
+64
-16
main.py
main.py
+64
-16
未找到文件。
main.py
浏览文件 @
3be73939
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
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录