Romanes commited on
Commit
30611c0
·
verified ·
1 Parent(s): 5edddfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -18,6 +18,7 @@ ART = ROOT / "artifacts"
18
  VEC_PATH = ART / "tfidf_vectorizer.joblib"
19
  MAT_PATH = ART / "tfidf_matrix.npz"
20
  IDX_PATH = ART / "doc_index.csv"
 
21
 
22
  # =========================
23
  # Utils de texto
@@ -846,10 +847,16 @@ with gr.Blocks(title="Recomendador de Códigos (CICP / CPC / UNSPSC)") as demo:
846
  with gr.Row():
847
  btn = gr.Button("Buscar", variant="primary")
848
  btn_xlsx = gr.Button("Descargar búsqueda")
 
849
  out = gr.Dataframe(headers=["Catálogo","Código","Nombre","Similaridad"], label="Resultados", wrap=True)
850
  status = gr.Markdown()
851
  file_out = gr.File(label="Archivo generado", interactive=False)
852
 
 
 
 
 
 
853
  def _on_search(q):
854
  df, msg = recomendar(q)
855
  return df, (f"**Estado:** {msg}" if msg else "")
 
18
  VEC_PATH = ART / "tfidf_vectorizer.joblib"
19
  MAT_PATH = ART / "tfidf_matrix.npz"
20
  IDX_PATH = ART / "doc_index.csv"
21
+ CATALOGOS_PATH = ROOT / "CATALOGOS.xlsx"
22
 
23
  # =========================
24
  # Utils de texto
 
847
  with gr.Row():
848
  btn = gr.Button("Buscar", variant="primary")
849
  btn_xlsx = gr.Button("Descargar búsqueda")
850
+
851
  out = gr.Dataframe(headers=["Catálogo","Código","Nombre","Similaridad"], label="Resultados", wrap=True)
852
  status = gr.Markdown()
853
  file_out = gr.File(label="Archivo generado", interactive=False)
854
 
855
+ # --- NUEVO: descarga del catálogo oficial + texto descriptivo ---
856
+ gr.Markdown("**Para más información, consultar los catálogos.**")
857
+ gr.DownloadButton(label="📥 Descargar CATALOGOS.xlsx", value=str(CATALOGOS_PATH), variant="secondary")
858
+ # ----------------------------------------------------------------
859
+
860
  def _on_search(q):
861
  df, msg = recomendar(q)
862
  return df, (f"**Estado:** {msg}" if msg else "")