Update app.py

app.py

@@ -9,9 +9,47 @@ from backend.llm_local import generate
 from backend.summarizer import extractive_sents, map_reduce
 
 
+
 for p in [settings.RAW_DIR, settings.INDEX_DIR, settings.MODEL_DIR]:
     p.mkdir(parents=True, exist_ok=True)
 
+def _index_files_exist() -> bool:
+    """Checa se o índice FAISS e o cache existem."""
+    from backend.rag import retriever
+    return (retriever.index_path.exists() if hasattr(retriever, "index_path") else False) or \
+           (settings.INDEX_DIR / "faiss.index").exists()
+
+def _raw_has_files() -> bool:
+    return any(settings.RAW_DIR.rglob("*.*"))
+
+def try_autoload_index() -> str | None:
+    """
+    Se já houver índice, carrega. Se houver arquivos em RAW mas não índice,
+    tenta construir automaticamente.
+    """
+    global STATE_INDEXED
+    try:
+        if _index_files_exist():
+            retriever.load()         
+            STATE_INDEXED = True
+            return None
+        elif _raw_has_files():
+            
+            ingest_all()              
+            retriever.build()       
+            retriever.load()
+            STATE_INDEXED = True
+            return None
+        else:
+            STATE_INDEXED = False
+            return "Nenhum arquivo em backend/data/raw/. Envie arquivos (ou comite nessa pasta) e clique em Indexar."
+    except Exception as e:
+        STATE_INDEXED = False
+        return f"Falha ao preparar índice automaticamente: {type(e).__name__} — {e}"
+
+
+
+
 STATE_INDEXED = False
 
 
@@ -81,27 +119,32 @@ def build_index():
     """Ingestão (parse) + construção e carga do índice. Mostra erro detalhado no UI."""
     global STATE_INDEXED
     try:
+        if not _raw_has_files():
+            STATE_INDEXED = False
+            return "Nenhum arquivo encontrado em backend/data/raw/. Adicione arquivos ou faça upload."
+
+
         if not (settings.INDEX_DIR / "meta.jsonl").exists():
-            ingest_all()  # extrai textos e cria meta.jsonl
-        retriever.build()  # cria/atualiza FAISS
-        retriever.load()   # carrega o índice em memória
+            ingest_all()
+
+        retriever.build()
+        retriever.load()
         STATE_INDEXED = True
         return "Índice criado/carregado. Vá para **Conversar** ou **Resumir**."
     except Exception as e:
         STATE_INDEXED = False
         return f"Falha ao indexar/carregar: **{type(e).__name__}** — {e}"
 
+
 def chat_answer(history, message):
-    """Carrega o índice se preciso e responde; se falhar, mostra o motivo em vez de 'ERRO!'."""
     if not history:
         history = []
 
-    try:
-        if not STATE_INDEXED:
-            retriever.load()
-    except Exception as e:
-        return history + [("system", f"Índice não está pronto: **{type(e).__name__}** — {e}\n"
-                                     "Vá na aba **Upload & Indexar** e clique em **Indexar**.")], ""
+
+    if not STATE_INDEXED:
+        warn = try_autoload_index()
+        if warn:
+            return history + [("system", warn + "\nVá na aba **Upload & Indexar** e clique em **Indexar**.")], ""
 
     try:
         hits = retriever.search(message, top_k=settings.TOP_K)