File size: 11,865 Bytes
b276dbe 159fe5e b276dbe 87f954d 2d6d126 c683bff 9e91c10 b276dbe 2d6d126 cf45911 a098337 2d6d126 cf45911 b276dbe c9a645f 0640991 b276dbe 0640991 b276dbe c9a645f 1c1d54b b276dbe 0640991 b276dbe cf45911 b276dbe cf45911 b276dbe 0640991 cf45911 b276dbe 9e91c10 b276dbe 9d9f444 c9a645f b276dbe 0640991 9d9f444 542468d 9d9f444 b276dbe 542468d c9a645f b276dbe 2d6d126 c9a645f a098337 c9a645f a098337 c9a645f a098337 c9a645f a098337 c9a645f a098337 c9a645f a098337 c9a645f 0640991 c9a645f 0640991 c9a645f a098337 c9a645f a098337 0640991 379044a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 |
import streamlit as st
import datetime
import pickle
import numpy as np
import rdflib
import torch
import os
import requests
from rdflib import Graph as RDFGraph, Namespace
from sentence_transformers import SentenceTransformer
# === STREAMLIT UI CONFIG ===
st.set_page_config(
page_title="Atlas de Lenguas: Lenguas Indígenas Sudamericanas",
page_icon="🌍",
layout="wide",
initial_sidebar_state="expanded",
menu_items={
'About': "## Análisis con IA de lenguas indígenas en peligro\n"
"Esta aplicación integra grafos de conocimiento de Glottolog, Wikipedia y Wikidata."
}
)
# === CONFIGURATION ===
ENDPOINT_URL = "https://lxgkooi70bj7diiu.us-east-1.aws.endpoints.huggingface.cloud"
HF_API_TOKEN = os.getenv("HF_API_TOKEN")
EMBEDDING_MODEL = "intfloat/multilingual-e5-base"
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
EX = Namespace("http://example.org/lang/")
# === CUSTOM CSS ===
st.markdown("""
<style>
.header {
color: #2c3e50;
border-bottom: 2px solid #4f46e5;
padding-bottom: 0.5rem;
margin-bottom: 1.5rem;
}
.feature-card {
background-color: #f8fafc;
border-radius: 8px;
padding: 1rem;
margin: 0.5rem 0;
border-left: 3px solid #4f46e5;
}
.response-card {
background-color: #fdfdfd;
color: #1f2937;
border-radius: 8px;
padding: 1.5rem;
box-shadow: 0 2px 6px rgba(0,0,0,0.08);
margin: 1rem 0;
font-size: 1rem;
line-height: 1.5;
}
.language-card {
background-color: #f9fafb;
border-radius: 8px;
padding: 1rem;
margin: 0.5rem 0;
border: 1px solid #e5e7eb;
}
.sidebar-section {
margin-bottom: 1.5rem;
}
.sidebar-title {
font-weight: 600;
color: #4f46e5;
}
.suggested-question {
padding: 0.5rem;
margin: 0.25rem 0;
border-radius: 4px;
cursor: pointer;
transition: all 0.2s;
}
.suggested-question:hover {
background-color: #f1f5f9;
}
.metric-badge {
display: inline-block;
background-color: #e8f4fc;
padding: 0.25rem 0.5rem;
border-radius: 4px;
font-size: 0.85rem;
margin-right: 0.5rem;
}
.tech-badge {
background-color: #ecfdf5;
color: #065f46;
padding: 0.25rem 0.5rem;
border-radius: 4px;
font-size: 0.75rem;
font-weight: 500;
}
</style>
""", unsafe_allow_html=True)
# === CARGA COMPONENTES ===
@st.cache_resource(show_spinner="Cargando modelos de IA y grafos de conocimiento...")
def load_all_components():
embedder = SentenceTransformer(EMBEDDING_MODEL, device=DEVICE)
methods = {}
label, suffix, ttl, matrix_path = ("LinkGraph", "_hybrid_graphsage", "grafo_ttl_hibrido_graphsage.ttl", "embed_matrix_hybrid_graphsage.npy")
with open(f"id_map{suffix}.pkl", "rb") as f:
id_map = pickle.load(f)
with open(f"grafo_embed{suffix}.pickle", "rb") as f:
G = pickle.load(f)
matrix = np.load(matrix_path)
rdf = RDFGraph()
rdf.parse(ttl, format="ttl")
methods[label] = (matrix, id_map, G, rdf)
return methods, embedder
# === FUNCIONES BASE ===
def get_top_k(matrix, id_map, query, k, embedder):
vec = embedder.encode(f"query: {query}", convert_to_tensor=True, device=DEVICE)
vec = vec.cpu().numpy().astype("float32")
sims = np.dot(matrix, vec) / (np.linalg.norm(matrix, axis=1) * np.linalg.norm(vec) + 1e-10)
top_k_idx = np.argsort(sims)[-k:][::-1]
return [id_map[i] for i in top_k_idx]
def get_context(G, lang_id):
node = G.nodes.get(lang_id, {})
lines = [f"**Lengua:** {node.get('label', lang_id)}"]
if node.get("wikipedia_summary"):
lines.append(f"**Wikipedia:** {node['wikipedia_summary']}")
if node.get("wikidata_description"):
lines.append(f"**Wikidata:** {node['wikidata_description']}")
if node.get("wikidata_countries"):
lines.append(f"**Países:** {node['wikidata_countries']}")
return "\n\n".join(lines)
def query_rdf(rdf, lang_id):
q = f"""
PREFIX ex: <http://example.org/lang/>
SELECT ?property ?value WHERE {{ ex:{lang_id} ?property ?value }}
"""
try:
return [(str(row[0]).split("/")[-1], str(row[1])) for row in rdf.query(q)]
except Exception as e:
return [("error", str(e))]
# === PROMPT PARA MODELO MISTRAL ===
def generate_response(matrix, id_map, G, rdf, user_question, k, embedder):
ids = get_top_k(matrix, id_map, user_question, k, embedder)
context = [get_context(G, i) for i in ids]
rdf_facts = []
for i in ids:
rdf_facts.extend([f"{p}: {v}" for p, v in query_rdf(rdf, i)])
prompt_es = f"""<s>[INST]
Eres un experto en lenguas indígenas sudamericanas. Utiliza estricta y únicamente la información a continuación para responder la pregunta del usuario en **español**.
- No infieras ni asumas hechos que no estén explícitamente establecidos.
- Si la respuesta es desconocida o insuficiente, di \"No puedo responder con los datos disponibles.\"
- Limita tu respuesta a 100 palabras.
### CONTEXTO:
{chr(10).join(context)}
### RELACIONES RDF:
{chr(10).join(rdf_facts)}
### PREGUNTA:
{user_question}
Respuesta:
[/INST]"""
prompt_en = f"""<s>[INST]
You are an expert in South American indigenous languages.
Use strictly and only the information below to answer the user question in **English**.
- Do not infer or assume facts that are not explicitly stated.
- If the answer is unknown or insufficient, say \"I cannot answer with the available data.\"
- Limit your answer to 100 words.
### CONTEXT:
{chr(10).join(context)}
### RDF RELATIONS:
{chr(10).join(rdf_facts)}
### QUESTION:
{user_question}
Answer:
[/INST]"""
try:
res_es = requests.post(
ENDPOINT_URL,
headers={"Authorization": f"Bearer {HF_API_TOKEN}", "Content-Type": "application/json"},
json={"inputs": prompt_es}, timeout=60
)
res_en = requests.post(
ENDPOINT_URL,
headers={"Authorization": f"Bearer {HF_API_TOKEN}", "Content-Type": "application/json"},
json={"inputs": prompt_en}, timeout=60
)
out_es = res_es.json()
out_en = res_en.json()
if isinstance(out_es, dict) and "generated_text" in out_es:
response_es = out_es["generated_text"].strip()
elif isinstance(out_es, list) and "generated_text" in out_es[0]:
response_es = out_es[0]["generated_text"].replace(prompt_es.strip(), "").strip()
else:
response_es = "Error en respuesta en español."
if isinstance(out_en, dict) and "generated_text" in out_en:
response_en = out_en["generated_text"].strip()
elif isinstance(out_en, list) and "generated_text" in out_en[0]:
response_en = out_en[0]["generated_text"].replace(prompt_en.strip(), "").strip()
else:
response_en = "Error in English response."
combined = f"<b>Respuesta en español:</b><br>{response_es}<br><br><b>Answer in English:</b><br>{response_en}"
return combined, ids, context, rdf_facts
except Exception as e:
return f"Error al consultar el modelo: {str(e)}", ids, context, rdf_facts
# === MAIN ===
def main():
methods, embedder = load_all_components()
st.markdown("""
<div class="header">
<h1>🌍 Atlas de Lenguas: Lenguas Indígenas Sudamericanas</h1>
</div>
""", unsafe_allow_html=True)
with st.expander("📌 **Resumen General**", expanded=True):
st.markdown("""
Esta aplicación ofrece **análisis impulsado por IA, Grafos y RAGs (GraphRAGs)** de lenguas indígenas de América del Sur,
integrando información de **Glottolog, Wikipedia y Wikidata**.
""")
with st.sidebar:
st.markdown("### 📚 Información de Contacto")
st.markdown("""
- <span class="tech-badge">Correo: jxvera@gmail.com</span>
""", unsafe_allow_html=True)
st.markdown("---")
st.markdown("### 🚀 Inicio Rápido")
st.markdown("""
1. **Escribe una pregunta** en el cuadro de entrada
2. **Haz clic en 'Analizar'** para obtener la respuesta
3. **Explora los resultados** con los detalles expandibles
""")
st.markdown("---")
st.markdown("### 🔍 Preguntas de Ejemplo")
questions = [
"¿Qué idiomas están en peligro en Brasil?",
"¿Qué idiomas se hablan en Perú?",
"¿Cuáles idiomas están relacionados con el Quechua?",
"¿Dónde se habla el Mapudungun?"
]
for q in questions:
if st.button(q, key=f"suggested_{q}", use_container_width=True):
st.session_state.query = q
st.markdown("---")
st.markdown("### ⚙️ Detalles Técnicos")
st.markdown("""
- <span class="tech-badge">Embeddings</span> GraphSAGE
- <span class="tech-badge">Modelo de Lenguaje</span> Mistral (Inference Endpoint)
- <span class="tech-badge">Grafo de Conocimiento</span> Integración basada en RDF
""", unsafe_allow_html=True)
st.markdown("---")
st.markdown("### 📂 Fuentes de Datos")
st.markdown("""
- **Glottolog** (Clasificación de idiomas)
- **Wikipedia** (Resúmenes textuales)
- **Wikidata** (Hechos estructurados)
""")
st.markdown("---")
st.markdown("### 📊 Parámetros de Análisis")
k = st.slider("Número de idiomas a analizar", 1, 10, 3)
st.markdown("---")
st.markdown("### 🔧 Opciones Avanzadas")
show_ctx = st.checkbox("Mostrar información de contexto", False)
show_rdf = st.checkbox("Mostrar hechos estructurados", False)
st.markdown("### 📝 Haz una pregunta sobre lenguas indígenas")
st.markdown("*(Puedes preguntar en español o inglés, y el modelo responderá en **ambos idiomas**.)*")
query = st.text_input(
"Ingresa tu pregunta:",
value=st.session_state.get("query", ""),
label_visibility="collapsed",
placeholder="Ej. ¿Qué lenguas se hablan en Perú?"
)
if st.button("Analizar", type="primary", use_container_width=True):
if not query:
st.warning("Por favor, ingresa una pregunta")
return
label = "LinkGraph"
method = methods[label]
start = datetime.datetime.now()
response, lang_ids, context, rdf_data = generate_response(*method, query, k, embedder)
duration = (datetime.datetime.now() - start).total_seconds()
st.markdown(f"""
<div class="response-card">
{response}
<div style="margin-top: 1rem;">
<span class="metric-badge">⏱️ {duration:.2f}s</span>
<span class="metric-badge">🌐 {len(lang_ids)} idiomas</span>
</div>
</div>
""", unsafe_allow_html=True)
if show_ctx:
with st.expander(f"📖 Contexto de {len(lang_ids)} idiomas"):
for lang_id, ctx in zip(lang_ids, context):
st.markdown(f"<div class='language-card'>{ctx}</div>", unsafe_allow_html=True)
if show_rdf:
with st.expander("🔗 Hechos estructurados (RDF)"):
st.code("\n".join(rdf_data))
st.markdown("---")
st.markdown("""
<div style="font-size: 0.8rem; color: #64748b; text-align: center;">
<b>📌 Nota:</b> Esta herramienta está diseñada para investigadores, lingüistas y preservacionistas culturales.
Para mejores resultados, usa preguntas específicas sobre idiomas, familias o regiones.
</div>
""", unsafe_allow_html=True)
if __name__ == "__main__":
main()
|