Alexander Sanchez commited on
Commit
563ceb8
Β·
1 Parent(s): 7e00c02

name updated

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -41,7 +41,7 @@ vs.index(all_pairs)
41
  corrector = RAGCorrector(vs)
42
  evaluator = Evaluator()
43
 
44
- print(f"βœ… Sistema listo. Documentos en vector store: {vs.count()}")
45
 
46
  # ── Ejemplos de demostraciΓ³n ──────────────────────────────────────────────────
47
 
@@ -61,12 +61,12 @@ def corregir(htr_text: str, top_k: int, mostrar_prompt: bool):
61
  return "", "", "", "", "⚠ Introduce un texto HTR para corregir."
62
 
63
  if not os.getenv("OPENAI_API_KEY"):
64
- return "", "", "", "", "❌ Falta OPENAI_API_KEY en el fichero .env"
65
 
66
  try:
67
  result = corrector.correct(htr_text, top_k=int(top_k))
68
  except Exception as e:
69
- return "", "", "", "", f"❌ Error al llamar a la API: {e}"
70
 
71
  corrected = result["corrected"]
72
  retrieved = result["retrieved"]
@@ -127,7 +127,7 @@ def corregir(htr_text: str, top_k: int, mostrar_prompt: bool):
127
  if mostrar_prompt:
128
  prompt_md = f"```\nSYSTEM:\n{result.get('_system', '(ver rag_corrector.py)')}\n\nUSER:\n{result['prompt']}\n```"
129
 
130
- status = f"βœ… CorrecciΓ³n completada con **{result['model']}** Β· {vs.count()} docs en Γ­ndice"
131
 
132
  return corrected, docs_md, analysis_md, diff_md, status
133
 
@@ -177,11 +177,11 @@ def add_to_corpus(htr_text: str, gt_text: str, doc_type: str, region: str, date:
177
  }
178
  added = vs.index([new_pair])
179
  if added:
180
- return f"βœ… Par aΓ±adido al corpus con id `{pair_id}`. Total: {vs.count()} docs."
181
  else:
182
  return f"β„Ή Par ya existΓ­a en el corpus (id: `{pair_id}`)."
183
  except Exception as e:
184
- return f"❌ Error: {e}"
185
 
186
 
187
  # ── Interfaz Gradio ───────────────────────────────────────────────────────────
@@ -205,8 +205,8 @@ with gr.Blocks(
205
  # ── Header ────────────────────────────────────────────────────────────────
206
  gr.HTML("""
207
  <div class="header">
208
- <h1>RAG CODEX for NSF project</h1>
209
- <p>Sistema RAG de correcciΓ³n de castellano del siglo XVI</p>
210
  </div>
211
  """)
212
 
 
41
  corrector = RAGCorrector(vs)
42
  evaluator = Evaluator()
43
 
44
+ print(f" Sistema listo. Documentos en vector store: {vs.count()}")
45
 
46
  # ── Ejemplos de demostraciΓ³n ──────────────────────────────────────────────────
47
 
 
61
  return "", "", "", "", "⚠ Introduce un texto HTR para corregir."
62
 
63
  if not os.getenv("OPENAI_API_KEY"):
64
+ return "", "", "", "", " Falta OPENAI_API_KEY en el fichero .env"
65
 
66
  try:
67
  result = corrector.correct(htr_text, top_k=int(top_k))
68
  except Exception as e:
69
+ return "", "", "", "", f" Error al llamar a la API: {e}"
70
 
71
  corrected = result["corrected"]
72
  retrieved = result["retrieved"]
 
127
  if mostrar_prompt:
128
  prompt_md = f"```\nSYSTEM:\n{result.get('_system', '(ver rag_corrector.py)')}\n\nUSER:\n{result['prompt']}\n```"
129
 
130
+ status = f" CorrecciΓ³n completada con **{result['model']}** Β· {vs.count()} docs en Γ­ndice"
131
 
132
  return corrected, docs_md, analysis_md, diff_md, status
133
 
 
177
  }
178
  added = vs.index([new_pair])
179
  if added:
180
+ return f" Par aΓ±adido al corpus con id `{pair_id}`. Total: {vs.count()} docs."
181
  else:
182
  return f"β„Ή Par ya existΓ­a en el corpus (id: `{pair_id}`)."
183
  except Exception as e:
184
+ return f" Error: {e}"
185
 
186
 
187
  # ── Interfaz Gradio ───────────────────────────────────────────────────────────
 
205
  # ── Header ────────────────────────────────────────────────────────────────
206
  gr.HTML("""
207
  <div class="header">
208
+ <h1>RAG CODEX for Historical Spanish'</h1>
209
+ <p>RAG system of Spanish correction from the 16th century</p>
210
  </div>
211
  """)
212