Update app.py

app.py

@@ -11,7 +11,6 @@ sum_model = AutoModelForSeq2SeqLM.from_pretrained("LaciaStudio/Lacia_sum_small_v
 def summarize_document(file):
     if file is None:
         return "Файл не загружен."
-    # Открываем файл и читаем его содержимое
     with open(file, "r", encoding="utf-8") as f:
         text = f.read()
     input_text = "summarize: " + text
@@ -49,35 +48,121 @@ def answer_question_text(context, question):
     return answer_question(context, question)
 
 # -------------------------------
-# Интерфейс Gradio
+# Функция установки языка и обновления интерфейса
 # -------------------------------
-with gr.Blocks() as demo:
-    gr.Markdown("# Интерфейс для суммаризации и вопросов-ответов")
-    with gr.Row():
-        # Левая колонка – суммаризация
-        with gr.Column():
-            gr.Markdown("## Суммаризация документа")
-            file_input_sum = gr.File(label="Прикрепить файл для суммаризации", file_count="single", type="filepath")
-            summarize_button = gr.Button("Суммаризировать")
-            summary_output = gr.Textbox(label="Суммаризация", lines=10)
-            summarize_button.click(fn=summarize_document, inputs=file_input_sum, outputs=summary_output)
-        
-        # Правая колонка – Q&A с двумя вкладками
-        with gr.Column():
-            gr.Markdown("## Вопрос-ответ по документу")
-            with gr.Tabs():
-                with gr.Tab("Загрузить файл"):
-                    file_input_qa = gr.File(label="Прикрепить файл с документом", file_count="single", type="filepath")
-                    question_input_file = gr.Textbox(label="Введите вопрос", placeholder="Ваш вопрос здесь")
-                    answer_button_file = gr.Button("Получить ответ")
-                    answer_output_file = gr.Textbox(label="Ответ", lines=5)
-                    answer_button_file.click(fn=answer_question_file, inputs=[file_input_qa, question_input_file], outputs=answer_output_file)
-                with gr.Tab("Ввести текст"):
-                    context_input = gr.Textbox(label="Введите текст документа", lines=10, placeholder="Текст документа здесь")
-                    question_input_text = gr.Textbox(label="Введите вопрос", placeholder="Ваш вопрос здесь")
-                    answer_button_text = gr.Button("Получить ответ")
-                    answer_output_text = gr.Textbox(label="Ответ", lines=5)
-                    answer_button_text.click(fn=answer_question_text, inputs=[context_input, question_input_text], outputs=answer_output_text)
-                    
+def set_language(lang):
+    if lang == "English":
+        texts = {
+            "app_title": "# Interface for Summarization and Q&A",
+            "sum_section": "Document Summarization",
+            "sum_file_label": "Attach file for summarization",
+            "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_text_label": "Enter document text",
+        }
+    else:  # Русский
+        texts = {
+            "app_title": "# Интерфейс для суммаризации и вопросов-ответов",
+            "sum_section": "Суммаризация документа",
+            "sum_file_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_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"])      # метка поля с ответом (текст)
+    )
+
+# -------------------------------
+# Интерфейс Gradio с улучшенным дизайном и выбором языка
+# -------------------------------
+with gr.Blocks(css="""
+    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; }
+""") 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 / Начать")
+    
+    # Основной интерфейс (изначально скрыт)
+    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.Column():
+                 qa_section_md = gr.Markdown("Placeholder Q&A Section")
+                 with gr.Tabs() as 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")
+                         answer_button_file = gr.Button("Placeholder QA Answer Button")
+                         answer_output_file = gr.Textbox(label="Placeholder QA Answer Output", lines=5)
+                     with gr.Tab("Placeholder Tab 2") as text_tab:
+                         context_input = gr.Textbox(label="Placeholder Context Input", lines=10, placeholder="Document text here")
+                         question_input_text = gr.Textbox(label="Placeholder QA Question", placeholder="Your question here")
+                         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
+         ]
+    )
+    
+    # Связываем функционал нейросетей
+    summarize_button.click(fn=summarize_document, inputs=file_input_sum, outputs=summary_output)
+    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)
+    
 if __name__ == "__main__":
     demo.launch()