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