themes

app.py

@@ -8,7 +8,6 @@ import tempfile
 from transformers import pipeline
 import subprocess as sp
 
-out_files = gr.State([])
 FILE_TIMEOUT = 10 ** 3
 MAX_FILES = 10
 
@@ -74,10 +73,10 @@ def interact_with_pdf(doc, question):
             outputs += p(img, question)
     return sorted(outputs, key=lambda x: x["score"], reverse=True)[0]['answer']
 
-
+'''
 text_interface = gr.Interface(
     fn=extract_text,
-    inputs=PDF(label="Загрузить PDF", button_color="#99FF00"),
+    inputs=PDF(label="Загрузить PDF"),
     outputs=gr.Textbox(label="Полученный текст"),
     title="PDF в текст",
     description="Сервис вынимает содержащийся в PDF контейнере текст."
@@ -85,7 +84,7 @@ text_interface = gr.Interface(
 
 pdf_interface = gr.Interface(
     fn=process_pdf,
-    inputs=[PDF(label="Загрузить PDF", button_color="#99FF00"), out_files],
+    inputs=[PDF(label="Загрузить PDF"), out_files],
     outputs=[gr.File(label="Скачать ZIP архив"), out_files],
     title="PDF в картинки",
     description="Переводит постранично файл в изображения, и предлагает бскачать в виде ZIP архива."
@@ -109,5 +108,18 @@ tabbed_interface = gr.TabbedInterface(
     tab_names=["В текст", "В картинки", "Задай вопрос"],
     # description="Choose a tab to perform the desired task."
 )
+'''
+
+with gr.Blocks() as tabbed_interface:
+    out_files = gr.State([])
+    with gr.Tab("В текст"):
+        inp1 = PDF(label="Загрузить PDF")
+        out1 = gr.Textbox(label="Полученный текст")
+        btn1 = gr.Button("Запустить")
+        btn1.click(fn=extract_text, inputs=inp1, outputs=out1)
+    with gt.Tab("В картинки"):
+        pass
+    with gr.Tab("Задай вопрос"):
+        pass
 
-tabbed_interface.launch(show_api=False, max_threads=8, custom_css=None)
+tabbed_interface.launch(show_api=False, max_threads=8)