Update app.py

app.py

@@ -10,12 +10,11 @@ from transformers import T5ForConditionalGeneration, T5Tokenizer
 def get_blocks_from_docx():
     docx_list = glob.glob("*.docx")
     if not docx_list:
-        return ["Файл .docx не найден!"]
+        return []
     doc = Document(docx_list[0])
     blocks = []
     for p in doc.paragraphs:
         txt = p.text.strip()
-        # Исключаем очень короткие и похожие на заголовки блоки
         if (
             txt
             and not (len(txt) <= 3 and txt.isdigit())
@@ -27,7 +26,6 @@ def get_blocks_from_docx():
             and len(txt.split()) > 3
         ):
             blocks.append(txt)
-    # Таблицы
     for table in doc.tables:
         for row in table.rows:
             row_text = " | ".join(cell.text.strip() for cell in row.cells if cell.text.strip())
@@ -42,6 +40,10 @@ def get_blocks_from_docx():
     return uniq_blocks
 
 blocks = get_blocks_from_docx()
+
+if len(blocks) < 1:
+    blocks = ["База знаний пуста: проверьте содержание и формат вашего .docx!"]
+
 vectorizer = TfidfVectorizer().fit(blocks)
 matrix = vectorizer.transform(blocks)
 
@@ -57,26 +59,30 @@ def rut5_answer(question, context):
         output_ids = model.generate(
             input_ids,
             max_length=250, num_beams=4, min_length=40,
-            no_repeat_ngram_size=3,
-            do_sample=False
+            no_repeat_ngram_size=3, do_sample=False
         )
     return tokenizer.decode(output_ids[0], skip_special_tokens=True)
 
 def ask_chatbot(question):
     if not question.strip():
         return "Пожалуйста, введите вопрос."
-    if len(blocks) < 2:
-        return "Ошибка: база знаний пуста или слишком мала. Проверьте .docx."
+    if not blocks or blocks == ["База знаний пуста: проверьте содержание и формат вашего .docx!"]:
+        return "Ошибка: база знаний пуста. Проверьте .docx и перезапустите Space."
     user_vec = vectorizer.transform([question])
     sims = cosine_similarity(user_vec, matrix)
     n_blocks = min(3, len(blocks))
-    top_idxs = sims.argsort()[-n_blocks:][::-1] if n_blocks > 0 else []
-    context_blocks = [
-        blocks[i] for i in top_idxs if sims[i] > 0.08 and len(blocks[i].split()) > 3 and len(blocks[i]) > 35
-    ]
+    # Корректно работают даже при len(blocks) == 1
+    top_idxs = list(reversed(sims.argsort()[-n_blocks:]))
+    context_blocks = []
+    for idx in top_idxs:
+        try:
+            if sims[idx] > 0.08 and len(blocks[idx].split()) > 3 and len(blocks[idx]) > 35:
+                context_blocks.append(blocks[idx])
+        except IndexError:
+            continue
     context = " ".join(context_blocks)
     if not context:
-        return "Не найден релевантный фрагмент в документе. Попробуйте иначе сформулировать вопрос."
+        return "Не найден релевантный фрагмент в документе. Попробуйте иначе сформулировать вопрос или добавьте больше содержательных абзацев в .docx."
     answer = rut5_answer(question, context)
     if len(answer.strip().split()) < 8 or len(answer.split('.')) < 2:
         answer += "\n\n" + context
@@ -93,30 +99,22 @@ EXAMPLES = [
 
 with gr.Blocks() as demo:
     gr.Markdown(
-        """
-        # Русскоязычный FAQ-чат-бот на базе вашей методички и нейросетевой модели
-
-        Задайте вопрос — получите развернутый AI-ответ (бот анализирует ваш документ, подборка абзацев и генерация выполняются автоматически).
-        """
+        "# Русскоязычный FAQ-чат-бот по методичке (AI+документ)\nЗадайте вопрос — получите развернутый ответ на основании вашего документа!"
     )
     question = gr.Textbox(label="Ваш вопрос", lines=2)
     ask_btn = gr.Button("Получить ответ")
     answer = gr.Markdown(label="Ответ", visible=True)
 
-    # Активация спиннера "Чат-бот думает..."
     def with_spinner(q):
         yield "Чат-бот думает..."
         yield ask_chatbot(q)
 
     ask_btn.click(with_spinner, question, answer)
     question.submit(with_spinner, question, answer)
-    
     gr.Markdown("#### Примеры вопросов:")
     gr.Examples(EXAMPLES, inputs=question)
-
     gr.Markdown("""
     ---
-
     ### Контакты (укажите свои)
     Преподаватель: ___________________
     Email: ___________________________
@@ -124,3 +122,4 @@ with gr.Blocks() as demo:
     """)
 
 demo.launch()
+