Andrei Shadrikov commited on
Commit
9fc3c5a
·
1 Parent(s): 7379a7f
Files changed (1) hide show
  1. app.py +32 -34
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
- with gr.Blocks(theme=gr.themes.Default(primary_hue=gr.themes.colors.red, secondary_hue=gr.themes.colors.pink)) as demo:
79
- text_interface = gr.Interface(
80
- fn=extract_text,
81
- inputs=PDF(label="Загрузить PDF"),
82
- outputs=gr.Textbox(label="Полученный текст"),
83
- title="PDF в текст",
84
- description="Сервис вынимает содержащийся в PDF контейнере текст."
85
- )
86
 
87
- pdf_interface = gr.Interface(
88
- fn=process_pdf,
89
- inputs=[PDF(label="Загрузить PDF"), out_files],
90
- outputs=[gr.File(label="Скачать ZIP архив"), out_files],
91
- title="PDF в картинки",
92
- description="Переводит постранично файл в изображения, и предлагает бскачать в виде ZIP архива."
93
- )
94
 
95
- image_interface = gr.Interface(
96
- fn=interact_with_pdf,
97
- inputs=[
98
- PDF(label="Загрузить PDF"),
99
- gr.Textbox(label="Что хотите найти?")
100
- ],
101
- outputs=gr.Textbox(label="Возможный ответ"),
102
- title="Спроси PDF",
103
- description="Сервис через методы OCR сканирует загруженный файл для нахождения ответа на заданный вопрос."
104
- )
105
 
106
- # Create a tabbed interface
107
- tabbed_interface = gr.TabbedInterface(
108
- [text_interface, pdf_interface, image_interface],
109
- title="Взаимодействие с PDF",
110
- tab_names=["В текст", "В картинки", "Задай вопрос"],
111
- # description="Choose a tab to perform the desired task."
112
- )
113
 
114
- # Launch the app
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()