Andrei Shadrikov
commited on
Commit
·
9fc3c5a
1
Parent(s):
7379a7f
themes
Browse files
app.py
CHANGED
@@ -75,41 +75,39 @@ def interact_with_pdf(doc, question):
|
|
75 |
return sorted(outputs, key=lambda x: x["score"], reverse=True)[0]['answer']
|
76 |
|
77 |
|
78 |
-
|
79 |
-
|
80 |
-
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
-
|
85 |
-
)
|
86 |
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
|
93 |
-
|
94 |
|
95 |
-
|
96 |
-
|
97 |
-
|
98 |
-
|
99 |
-
|
100 |
-
|
101 |
-
|
102 |
-
|
103 |
-
|
104 |
-
|
105 |
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
|
110 |
-
|
111 |
-
|
112 |
-
|
113 |
|
114 |
-
|
115 |
-
demo.launch()
|
|
|
75 |
return sorted(outputs, key=lambda x: x["score"], reverse=True)[0]['answer']
|
76 |
|
77 |
|
78 |
+
text_interface = gr.Interface(
|
79 |
+
fn=extract_text,
|
80 |
+
inputs=PDF(label="Загрузить PDF", button_color="#99FF00"),
|
81 |
+
outputs=gr.Textbox(label="Полученный текст"),
|
82 |
+
title="PDF в текст",
|
83 |
+
description="Сервис вынимает содержащийся в PDF контейнере текст."
|
84 |
+
)
|
|
|
85 |
|
86 |
+
pdf_interface = gr.Interface(
|
87 |
+
fn=process_pdf,
|
88 |
+
inputs=[PDF(label="Загрузить PDF", button_color="#99FF00"), out_files],
|
89 |
+
outputs=[gr.File(label="Скачать ZIP архив"), out_files],
|
90 |
+
title="PDF в картинки",
|
91 |
+
description="Переводит постранично файл в изображения, и предлагает бскачать в виде ZIP архива."
|
92 |
+
)
|
93 |
|
94 |
+
image_interface = gr.Interface(
|
95 |
+
fn=interact_with_pdf,
|
96 |
+
inputs=[
|
97 |
+
PDF(label="Загрузить PDF"),
|
98 |
+
gr.Textbox(label="Что хотите найти?")
|
99 |
+
],
|
100 |
+
outputs=gr.Textbox(label="Возможный ответ"),
|
101 |
+
title="Спроси PDF",
|
102 |
+
description="Сервис через методы OCR сканирует загруженный файл для нахождения ответа на заданный вопрос."
|
103 |
+
)
|
104 |
|
105 |
+
# Create a tabbed interface
|
106 |
+
tabbed_interface = gr.TabbedInterface(
|
107 |
+
[text_interface, pdf_interface, image_interface],
|
108 |
+
title="Взаимодействие с PDF",
|
109 |
+
tab_names=["В текст", "В картинки", "Задай вопрос"],
|
110 |
+
# description="Choose a tab to perform the desired task."
|
111 |
+
)
|
112 |
|
113 |
+
tabbed_interface.launch()
|
|