Update app.py

app.py

@@ -19,6 +19,15 @@ def summarize_document(file):
     summary = sum_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
     return summary
 
+def summarize_text(text):
+    if not text:
+        return "Текст не предоставлен."
+    input_text = "summarize: " + text
+    inputs = sum_tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True)
+    summary_ids = sum_model.generate(inputs["input_ids"], max_length=150, num_beams=4, early_stopping=True)
+    summary = sum_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+    return summary
+
 # -------------------------------
 # Модель вопросов-ответов (Q&A)
 # -------------------------------
@@ -55,57 +64,70 @@ def set_language(lang):
         texts = {
             "app_title": "# Interface for Summarization and Q&A",
             "sum_section": "Document Summarization",
+            "sum_tab_file": "Upload File",
             "sum_file_label": "Attach file for summarization",
+            "sum_tab_text": "Enter Text",
+            "sum_text_label": "Enter document text",
             "sum_button": "Summarize",
             "sum_output": "Summarization",
             "qa_section": "Document Q&A",
             "qa_tab_file": "Upload File",
-            "qa_tab_text": "Enter Text",
             "qa_file_label": "Attach file with document",
             "qa_question_label": "Enter your question",
             "qa_answer_button": "Get Answer",
             "qa_answer_label": "Answer",
+            "qa_tab_text": "Enter Text",
             "qa_text_label": "Enter document text",
         }
     else:  # Русский
         texts = {
             "app_title": "# Интерфейс для суммаризации и вопросов-ответов",
             "sum_section": "Суммаризация документа",
+            "sum_tab_file": "Загрузить файл",
             "sum_file_label": "Прикрепить файл для суммаризации",
+            "sum_tab_text": "Ввести текст",
+            "sum_text_label": "Введите текст документа",
             "sum_button": "Суммаризировать",
             "sum_output": "Суммаризация",
             "qa_section": "Вопрос-ответ по документу",
             "qa_tab_file": "Загрузить файл",
-            "qa_tab_text": "Ввести текст",
             "qa_file_label": "Прикрепить файл с документом",
             "qa_question_label": "Введите вопрос",
             "qa_answer_button": "Получить ответ",
             "qa_answer_label": "Ответ",
+            "qa_tab_text": "Ввести текст",
             "qa_text_label": "Введите текст документа",
         }
     return (
-       gr.update(visible=False),                  # скрыть выбор языка
-       gr.update(visible=True),                   # показать основной интерфейс
-       gr.update(value=texts["app_title"]),         # заголовок приложения
-       gr.update(value="### " + texts["sum_section"]),  # секция суммаризации
-       gr.update(label=texts["sum_file_label"]),    # метка загрузки файла для суммаризации
-       gr.update(value=texts["sum_button"]),        # текст кнопки суммаризации
-       gr.update(label=texts["sum_output"]),        # метка поля суммаризации
-       gr.update(value="### " + texts["qa_section"]),   # секция Q&A
-       gr.update(label=texts["qa_tab_file"]),         # название вкладки «Загрузить файл»
-       gr.update(label=texts["qa_tab_text"]),         # название вкладки «Ввести текст»
-       gr.update(label=texts["qa_file_label"]),       # метка загрузки файла для Q&A
-       gr.update(label=texts["qa_question_label"]),   # метка поля ввода вопроса (файл)
-       gr.update(value=texts["qa_answer_button"]),    # текст кнопки ответа (файл)
-       gr.update(label=texts["qa_answer_label"]),     # метка поля с ответом (файл)
-       gr.update(label=texts["qa_text_label"]),       # метка поля ввода текста для Q&A
-       gr.update(label=texts["qa_question_label"]),   # метка поля ввода вопроса (текст)
-       gr.update(value=texts["qa_answer_button"]),    # текст кнопки ответа (текст)
-       gr.update(label=texts["qa_answer_label"])      # метка поля с ответом (текст)
+       gr.update(visible=False),               # Скрыть выбор языка
+       gr.update(visible=True),                # Показать основной интерфейс
+       gr.update(value=texts["app_title"]),      # Заголовок приложения
+       # --- Обновление комп��нентов для суммаризации ---
+       gr.update(value="### " + texts["sum_section"]),  # Заголовок секции суммаризации
+       gr.update(label=texts["sum_tab_file"]),          # Название вкладки (файл)
+       gr.update(label=texts["sum_file_label"]),        # Метка компонента File (суммаризация, файл)
+       gr.update(value=texts["sum_button"]),            # Кнопка суммаризации (файл)
+       gr.update(label=texts["sum_output"]),            # Вывод суммаризации (файл)
+       gr.update(label=texts["sum_tab_text"]),          # Название вкладки (текст)
+       gr.update(label=texts["sum_text_label"]),        # Метка компонента Text (суммаризация, текст)
+       gr.update(value=texts["sum_button"]),            # Кнопка суммаризации (текст)
+       gr.update(label=texts["sum_output"]),            # Вывод суммаризации (текст)
+       # --- Обновление компонентов для Q&A ---
+       gr.update(value="### " + texts["qa_section"]),   # Заголовок секции Q&A
+       gr.update(label=texts["qa_tab_file"]),           # Название вкладки Q&A (файл)
+       gr.update(label=texts["qa_tab_text"]),           # Название вкладки Q&A (текст)
+       gr.update(label=texts["qa_file_label"]),         # Метка компонента File (Q&A, файл)
+       gr.update(label=texts["qa_question_label"]),     # Метка поля вопроса (Q&A, файл)
+       gr.update(value=texts["qa_answer_button"]),      # Кнопка получения ответа (Q&A, файл)
+       gr.update(label=texts["qa_answer_label"]),       # Вывод ответа (Q&A, файл)
+       gr.update(label=texts["qa_text_label"]),         # Метка компонента Text (Q&A, текст)
+       gr.update(label=texts["qa_question_label"]),     # Метка поля вопроса (Q&A, текст)
+       gr.update(value=texts["qa_answer_button"]),      # Кнопка получения ответа (Q&A, текст)
+       gr.update(label=texts["qa_answer_label"])        # Вывод ответа (Q&A, текст)
     )
 
 # -------------------------------
-# Интерфейс Gradio с улучшенным дизайном и выбором языка
+# Интерфейс Gradio с улучшенным дизайном, выбором языка и вкладками для ввода
 # -------------------------------
 with gr.Blocks(css="""
     body { background-color: #f0f2f5; }
@@ -115,7 +137,7 @@ with gr.Blocks(css="""
 """) as demo:
     gr.HTML("<style>body { background-color: #f0f2f5; } .gradio-container { border-radius: 10px; box-shadow: 0 0 15px rgba(0,0,0,0.1); padding: 20px; background-color: white; } h1, h2, h3 { color: #333; } .gr-button { background-color: #4CAF50; color: white; }</style>")
     
-    # Контейнер выбора языка (показывается первым)
+    # Контейнер выбора языка (первоначально показывается)
     with gr.Column(visible=True) as lang_container:
          lang_radio = gr.Radio(choices=["English", "Русский"], label="Select Language / Выберите язык", value="English")
          start_button = gr.Button("Start / Начать")
@@ -124,16 +146,22 @@ with gr.Blocks(css="""
     with gr.Column(visible=False) as main_container:
          header_markdown = gr.Markdown("Placeholder Title")
          with gr.Row():
-             # Левая колонка – суммаризация
+             # Левая колонка – Суммаризация с двумя вкладками
              with gr.Column():
                  sum_section_md = gr.Markdown("Placeholder Summarization Section")
-                 file_input_sum = gr.File(label="Placeholder File Input", file_count="single", type="filepath")
-                 summarize_button = gr.Button("Placeholder Summarize Button")
-                 summary_output = gr.Textbox(label="Placeholder Summarization Output", lines=10)
-             # Правая колонка – Q&A
+                 with gr.Tabs() as sum_tabs:
+                     with gr.Tab("Placeholder Tab 1") as sum_file_tab:
+                         file_input_sum = gr.File(label="Placeholder File Input", file_count="single", type="filepath")
+                         summarize_button_file = gr.Button("Placeholder Summarize Button")
+                         summary_output_file = gr.Textbox(label="Placeholder Summarization Output", lines=10)
+                     with gr.Tab("Placeholder Tab 2") as sum_text_tab:
+                         text_input_sum = gr.Textbox(label="Placeholder Text Input", lines=10, placeholder="Enter document text here")
+                         summarize_button_text = gr.Button("Placeholder Summarize Button")
+                         summary_output_text = gr.Textbox(label="Placeholder Summarization Output", lines=10)
+             # Правая колонка – Q&A с двумя вкладками
              with gr.Column():
                  qa_section_md = gr.Markdown("Placeholder Q&A Section")
-                 with gr.Tabs() as tabs:
+                 with gr.Tabs() as qa_tabs:
                      with gr.Tab("Placeholder Tab 1") as file_tab:
                          file_input_qa = gr.File(label="Placeholder QA File Input", file_count="single", type="filepath")
                          question_input_file = gr.Textbox(label="Placeholder QA Question", placeholder="Your question here")
@@ -145,22 +173,30 @@ with gr.Blocks(css="""
                          answer_button_text = gr.Button("Placeholder QA Answer Button")
                          answer_output_text = gr.Textbox(label="Placeholder QA Answer Output", lines=5)
     
-    # При нажатии на кнопку выбора языка обновляем метки и показываем основной интерфейс
+    # При выборе языка – обновляем все метки и показываем основной интерфейс
     start_button.click(
          set_language,
          inputs=[lang_radio],
          outputs=[
              lang_container, main_container,
-             header_markdown, sum_section_md, file_input_sum,
-             summarize_button, summary_output, qa_section_md,
-             file_tab, text_tab, file_input_qa, question_input_file,
-             answer_button_file, answer_output_file, context_input,
-             question_input_text, answer_button_text, answer_output_text
+             header_markdown, 
+             # Компоненты для суммаризации (файл)
+             sum_section_md,
+             sum_file_tab, file_input_sum, summarize_button_file, summary_output_file,
+             # Компоненты для суммаризации (текст)
+             sum_text_tab, text_input_sum, summarize_button_text, summary_output_text,
+             # Компоненты для Q&A
+             qa_section_md,
+             file_tab, text_tab, file_input_qa, question_input_file, answer_button_file, answer_output_file,
+             context_input, question_input_text, answer_button_text, answer_output_text
          ]
     )
     
     # Связываем функционал нейросетей
-    summarize_button.click(fn=summarize_document, inputs=file_input_sum, outputs=summary_output)
+    # Суммаризация
+    summarize_button_file.click(fn=summarize_document, inputs=file_input_sum, outputs=summary_output_file)
+    summarize_button_text.click(fn=summarize_text, inputs=text_input_sum, outputs=summary_output_text)
+    # Q&A
     answer_button_file.click(fn=answer_question_file, inputs=[file_input_qa, question_input_file], outputs=answer_output_file)
     answer_button_text.click(fn=answer_question_text, inputs=[context_input, question_input_text], outputs=answer_output_text)