Update app.py

app.py

@@ -9,12 +9,12 @@ from backend.llm_local import generate
 from backend.summarizer import extractive_sents, map_reduce
 
 
-settings.RAW_DIR.mkdir(parents=True, exist_ok=True)
-settings.INDEX_DIR.mkdir(parents=True, exist_ok=True)
-
+for p in [settings.RAW_DIR, settings.INDEX_DIR, settings.MODEL_DIR]:
+    p.mkdir(parents=True, exist_ok=True)
 
 STATE_INDEXED = False
 
+
 def list_uploaded_files():
     files = []
     for f in settings.RAW_DIR.glob("**/*"):
@@ -22,6 +22,60 @@ def list_uploaded_files():
             files.append(str(f.relative_to(settings.RAW_DIR)))
     return sorted(files)
 
+def _save_one(file_obj: Any) -> str:
+    """
+    Salva um item vindo do componente gr.File, qualquer que seja o formato:
+    - str/Path (caminho no disco)
+    - objeto com .path
+    - objeto com .save()
+    - objeto com .name / .orig_name e .read()
+    Retorna o nome salvo.
+    """
+    dest_dir = settings.RAW_DIR
+    dest_dir.mkdir(parents=True, exist_ok=True)
+
+    if isinstance(file_obj, (str, Path)):
+        src = Path(file_obj)
+        dest = dest_dir / src.name
+        shutil.copy(src, dest)
+        return dest.name
+
+    path = getattr(file_obj, "path", None)
+    if path:
+        src = Path(path)
+        name = getattr(file_obj, "orig_name", None) or getattr(file_obj, "name", None) or src.name
+        dest = dest_dir / name
+        shutil.copy(src, dest)
+        return dest.name
+
+    saver = getattr(file_obj, "save", None)
+    if callable(saver):
+        name = getattr(file_obj, "orig_name", None) or getattr(file_obj, "name", None) or "arquivo_subido"
+        dest = dest_dir / name
+        saver(str(dest))
+        return dest.name
+
+    reader = getattr(file_obj, "read", None)
+    if callable(reader):
+        data = reader()
+        name = getattr(file_obj, "orig_name", None) or getattr(file_obj, "name", None) or "arquivo_subido"
+        dest = dest_dir / name
+        with open(dest, "wb") as f:
+            f.write(data)
+        return dest.name
+
+    raise ValueError("Formato de arquivo não reconhecido pelo upload.")
+
+def upload_files(files):
+    try:
+        saved = []
+        for item in (files or []):
+            saved.append(_save_one(item))
+        msg = "Upload concluído: " + (", ".join(saved) if saved else "nenhum arquivo")
+        return gr.update(choices=list_uploaded_files(), value=None), msg + ". Agora clique em **Indexar**."
+    except Exception as e:
+        return gr.update(choices=list_uploaded_files(), value=None), f"Falha no upload: {type(e).__name__}: {e}"
+
 
 def build_index():
     """Ingestão (parse) + construção e carga do índice. Mostra erro detalhado no UI."""
@@ -37,7 +91,6 @@ def build_index():
         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:
@@ -50,7 +103,6 @@ def chat_answer(history, message):
         return history + [("system", f"Índice não está pronto: **{type(e).__name__}** — {e}\n"
                                      "Vá na aba **Upload & Indexar** e clique em **Indexar**.")], ""
 
-
     try:
         hits = retriever.search(message, top_k=settings.TOP_K)
         if not hits:
@@ -63,7 +115,6 @@ def chat_answer(history, message):
     except Exception as e:
         return history + [("system", f"Falha na busca/resposta: **{type(e).__name__}** — {e}")], ""
 
-
 def summarize_run(filename, pages, chapter, query, style, length):
     if not STATE_INDEXED:
         try:
@@ -95,6 +146,7 @@ def summarize_run(filename, pages, chapter, query, style, length):
         final = map_reduce(hits, focus=query or "resumo", style=style, length=length)
     return final
 
+
 with gr.Blocks(title="Agente de Estudos IA") as demo:
     gr.Markdown("# Agente de Estudos IA \nCarregue seu **livro** e **slides** e estude com o bot.")
 
@@ -107,7 +159,6 @@ with gr.Blocks(title="Agente de Estudos IA") as demo:
         upload_btn.click(upload_files, inputs=files_in, outputs=[uploaded_list, status])
         index_btn.click(build_index, outputs=status)
 
-
     with gr.Tab("Conversar"):
         chatbot = gr.Chatbot(height=420)
         msg = gr.Textbox(placeholder="Pergunte algo")
@@ -125,72 +176,8 @@ with gr.Blocks(title="Agente de Estudos IA") as demo:
         length_dd = gr.Dropdown(choices=["curto", "médio", "longo"], value="médio", label="Tamanho")
         run_btn = gr.Button("Gerar resumo", variant="primary")
         out_md = gr.Markdown()
-
         run_btn.click(summarize_run,
                       inputs=[file_dd, pages_tb, chapter_tb, query_tb, style_dd, length_dd],
                       outputs=out_md)
 
-
-
-
-def _save_one(file_obj: Any) -> str:
-    """
-    Salva um item vindo do componente gr.File, qualquer que seja o formato:
-    - str/Path (caminho no disco)
-    - objeto com .path
-    - objeto com .save()
-    - objeto com .name / .orig_name e .read()
-    Retorna o nome salvo.
-    """
-    dest_dir = settings.RAW_DIR
-    dest_dir.mkdir(parents=True, exist_ok=True)
-
-
-    if isinstance(file_obj, (str, Path)):
-        src = Path(file_obj)
-        dest = dest_dir / src.name
-        shutil.copy(src, dest)
-        return dest.name
-
-
-    path = getattr(file_obj, "path", None)
-    if path:
-        src = Path(path)
-        name = getattr(file_obj, "orig_name", None) or getattr(file_obj, "name", None) or src.name
-        dest = dest_dir / name
-        shutil.copy(src, dest)
-        return dest.name
-
-
-    saver = getattr(file_obj, "save", None)
-    if callable(saver):
-        name = getattr(file_obj, "orig_name", None) or getattr(file_obj, "name", None) or "arquivo_subido"
-        dest = dest_dir / name
-        saver(str(dest))
-        return dest.name
-
-
-    reader = getattr(file_obj, "read", None)
-    if callable(reader):
-        data = reader()
-        name = getattr(file_obj, "orig_name", None) or getattr(file_obj, "name", None) or "arquivo_subido"
-        dest = dest_dir / name
-        with open(dest, "wb") as f:
-            f.write(data)
-        return dest.name
-
-    raise ValueError("Formato de arquivo não reconhecido pelo upload.")
-
-def upload_files(files):
-    try:
-        saved = []
-        for item in (files or []):
-            saved.append(_save_one(item))
-        msg = "Upload concluído: " + (", ".join(saved) if saved else "nenhum arquivo")
-        return gr.update(choices=list_uploaded_files(), value=None), msg + ". Agora clique em **Indexar**."
-    except Exception as e:
-
-        return gr.update(choices=list_uploaded_files(), value=None), f"Falha no upload: {type(e).__name__}: {e}"
-
-
 demo.queue().launch()