Update app.py

app.py

@@ -124,32 +124,37 @@ class VectorRAGSystem:
             print("🔄 Попытка загрузки векторных данных...")
             
             # Файлы векторных данных
-            faiss_file = "chunks_flatip.faiss"
-            metadata_file = "metadata.json"
+            faiss_file     = "chunks_flatip.faiss"
+            metadata_file  = "metadata.json"
             
             if not all(os.path.exists(f) for f in [faiss_file, metadata_file]):
                 print("📁 Файлы векторных данных не найдены")
                 return False
             
-            # Загружаем метаданные (список с page и chunk_id)
+            # Загружаем метаданные
             with open(metadata_file, 'r', encoding='utf-8') as f:
                 metadata_list = json.load(f)
             
-            # Создаем структуру чанков без текстов (получим из PDF по требованию)
+            # Собираем self.chunks для всех типов элементов
             self.chunks = []
             for i, item in enumerate(metadata_list):
+                # унифицируем идентификатор чанка
+                chunk_id = item.get("chunk_id",
+                            item.get("table_id",
+                            item.get("img_id", None)))
+                
                 self.chunks.append({
-                    "page": item["page"],
-                    "chunk_id": item["chunk_id"],
+                    "page":        item["page"],
+                    "chunk_id":    chunk_id,
                     "chunk_index": i,
-                    "text": "",  # Получим из PDF по требованию
-                    "metadata": {}
+                    "text":        "",   # подгрузим из PDF при выдаче
+                    "metadata":    {}
                 })
             
-            # Сохраняем метаданные
+            # Сохраняем общую статистику
             self.metadata = {"total_chunks": len(self.chunks)}
             
-            # Загружаем FAISS индекс
+            # Загружаем FAISS-индекс
             if HAS_FAISS:
                 self.faiss_index = faiss.read_index(faiss_file)
             
@@ -169,7 +174,7 @@ class VectorRAGSystem:
         except Exception as e:
             print(f"❌ Ошибка загрузки векторных данных: {e}")
             return False
-    
+ 
     def get_page_text(self, page_num: int) -> str:
         """Получение полного текста страницы с кешированием"""
         if page_num in self.page_texts: