Update app.py

app.py

@@ -1,3 +1,9 @@
+#!/usr/bin/env python3
+"""
+Финальная векторная RAG система для HuggingFace Spaces
+Адаптированная версия с поддержкой векторного поиска и резервным режимом
+"""
+
 import os
 import json
 import pickle
@@ -201,20 +207,20 @@ class VectorRAGSystem:
 
 📊 **Технические характеристики:**
 - 📦 Векторных эмбеддингов: {total_chunks}
-- 🔍 Режим поиска: {mode}
+- 🔍 Режим поиска: {mode} (только векторный)
 - 🧠 Модель генерации: {self.generation_model}
 - 🎯 LLM реранкинг: {self.reranking_model}
 - 📄 Parent-page enrichment: {pdf_enrichment}
 - 📋 Структурированный вывод: {structured_output}
 
 🚀 **Архитектурные особенности:**
-- 🔎 **Векторный поиск** с text-embedding-3-large
+- 🔎 **Векторный поиск** с text-embedding-3-large (только)
 - 📄 **Parent-page enrichment** через PyMuPDF
 - 🧠 **LLM реранкинг** для повышения релевантности
 - 🤔 **Chain-of-Thought** рассуждения
 - 📋 **JSON Schema** для структурированных ответов
 - 📊 **Confidence scoring** и детальная аналитика
-- 📚 **Предобработка** PDF файла (текст и таблицы) через pdfplumber
+- 📚 **Предобработка** с pdfplumber + таблицы
 
 💡 **Готова к интеллектуальному анализу отчета ПАО Сбербанк 2023!**"""
         
@@ -512,10 +518,15 @@ class VectorRAGSystem:
             
             # Подготовка источников
             sources = []
+            # Создаем словарь для быстрого поиска search_score по chunk_index
+            search_scores = {}
+            for chunk, score in search_results:
+                search_scores[chunk.get("chunk_index", -1)] = score
+            
             for chunk, score in reranked_results[:self.final_chunks_count]:
                 sources.append({
                     "page": chunk["page"],
-                    "search_score": search_results[0][1] if search_results else 0,
+                    "search_score": search_scores.get(chunk.get("chunk_index", -1), 0),
                     "rerank_score": score,
                     "preview": chunk["text"][:200] + "..." if len(chunk["text"]) > 200 else chunk["text"]
                 })