Spaces:
Sleeping
Sleeping
Alexander Sanchez commited on
Commit Β·
563ceb8
1
Parent(s): 7e00c02
name updated
Browse files
app.py
CHANGED
|
@@ -41,7 +41,7 @@ vs.index(all_pairs)
|
|
| 41 |
corrector = RAGCorrector(vs)
|
| 42 |
evaluator = Evaluator()
|
| 43 |
|
| 44 |
-
print(f"
|
| 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 "", "", "", "", "
|
| 65 |
|
| 66 |
try:
|
| 67 |
result = corrector.correct(htr_text, top_k=int(top_k))
|
| 68 |
except Exception as e:
|
| 69 |
-
return "", "", "", "", f"
|
| 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"
|
| 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"
|
| 181 |
else:
|
| 182 |
return f"βΉ Par ya existΓa en el corpus (id: `{pair_id}`)."
|
| 183 |
except Exception as e:
|
| 184 |
-
return f"
|
| 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
|
| 209 |
-
<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 |
|