themes

app.py

@@ -8,6 +8,7 @@ import tempfile
 from transformers import pipeline
 import subprocess as sp
 
+out_files = gr.State([])
 FILE_TIMEOUT = 10 ** 3
 MAX_FILES = 10
 
@@ -16,6 +17,14 @@ p = pipeline(
     model="impira/layoutlm-document-qa",
 )
 
+custom_css = """
+<style>
+button.gradio-pdf-button {
+    background-color: #99FF00 !important;
+}
+</style>
+"""
+
 def handle_files(cur_files):
     cur_time = cur_files[-1][0]
     deleted_indices = set()
@@ -73,7 +82,7 @@ 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"),
@@ -106,12 +115,12 @@ tabbed_interface = gr.TabbedInterface(
     [text_interface, pdf_interface, image_interface],
     title="Взаимодействие с PDF",
     tab_names=["В текст", "В картинки", "Задай вопрос"],
-    # description="Choose a tab to perform the desired task."
+    # description="Choose a tab to perform the desired task.",
+    css=custom_css
 )
 '''
 
 with gr.Blocks() as tabbed_interface:
-    out_files = gr.State([])
     with gr.Tab("В текст"):
         inp1 = PDF(label="Загрузить PDF")
         out1 = gr.Textbox(label="Полученный текст")
@@ -128,5 +137,6 @@ with gr.Blocks() as tabbed_interface:
         out3 = gr.Textbox(label="Возможный ответ")
         btn3 = gr.Button("Запустить")
         btn3.click(fn=interact_with_pdf, inputs=[inp3, inp4], outputs=out3)
+'''
 
 tabbed_interface.launch(show_api=False, max_threads=8)