Upload 6 files

app.py

@@ -0,0 +1,125 @@
+# -*- coding: utf-8 -*-
+"""app.ipynb
+
+Automatically generated by Colab.
+
+Original file is located at
+    https://colab.research.google.com/drive/1e3y3OTjQnNFZ7FLgs2elviYcG7ptj2fI
+"""
+
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+from sentence_transformers import SentenceTransformer, util
+import numpy as np
+import joblib
+
+# === Cargar modelos previos ===
+modelo_riesgo = joblib.load("modelo_riesgo.pkl")
+modelo_violencia = joblib.load("modelo_tipo_violencia.pkl")
+modelo_medida = joblib.load("modelo_tipo_medida.pkl")
+codificadores = joblib.load("codificadores_label_encoder.pkl")
+modelo_vector = SentenceTransformer("Snowflake/snowflake-arctic-embed-xs")
+
+# === Cargar modelo de lenguaje OpenHermes 2.5 ===
+modelo_llm = pipeline(
+    "text-generation",
+    model="teknium/OpenHermes-2.5-Mistral-7B",
+    tokenizer="teknium/OpenHermes-2.5-Mistral-7B",
+    max_new_tokens=300,
+    temperature=0.5,
+    do_sample=True
+)
+
+# === Frases prototipo para verificador semántico ===
+frases_fisica = [
+    "Me golpeó con el puño cerrado", "Me pateó", "Me lanzó contra la pared",
+    "Me estranguló", "Me fracturó una costilla", "Me tiró al piso violentamente"
+]
+frases_sexual = [
+    "Me obligó a tener relaciones sexuales", "Me tocó sin consentimiento",
+    "Me violó", "Me forzó a tener sexo", "Me agredió sexualmente"
+]
+embeds_fisica = modelo_vector.encode(frases_fisica)
+embeds_sexual = modelo_vector.encode(frases_sexual)
+
+# === FUNCIÓN ACTUALIZADA ===
+def predecir_con_recomendacion(edad, genero, hijos, convivencia_agresor, consumo_sustancias, apoyo_familiar, descripcion):
+    # Vector tabular
+    vector_tabular = np.array([
+        int(edad),
+        int(hijos),
+        codificadores["genero"].transform([genero])[0],
+        0, 0, 0,
+        codificadores["convivencia_agresor"].transform([convivencia_agresor])[0],
+        codificadores["consumo_sustancias"].transform([consumo_sustancias])[0],
+        codificadores["apoyo_familiar"].transform([apoyo_familiar])[0]
+    ])
+
+    # Vectorizar la descripción
+    vector_desc = modelo_vector.encode([descripcion])[0]
+    entrada = np.concatenate([vector_tabular, vector_desc])
+
+    # Predicciones
+    riesgo_cod = modelo_riesgo.predict([entrada])[0]
+    tipo_violencia_cod = modelo_violencia.predict([entrada])[0]
+    tipo_medida_cod = modelo_medida.predict([entrada])[0]
+
+    # Decodificación
+    riesgo = codificadores["riesgo"].inverse_transform([riesgo_cod])[0]
+    tipo_violencia = codificadores["tipo_violencia"].inverse_transform([tipo_violencia_cod])[0]
+    tipo_medida = codificadores["tipo_medida"].inverse_transform([tipo_medida_cod])[0]
+
+    # Verificador semántico
+    emb_desc = modelo_vector.encode(descripcion)
+    sim_fisica = max(util.cos_sim(emb_desc, embeds_fisica)[0])
+    sim_sexual = max(util.cos_sim(emb_desc, embeds_sexual)[0])
+
+    if sim_fisica > 0.7:
+        tipo_violencia = "física"
+    elif sim_sexual > 0.7:
+        tipo_violencia = "sexual"
+
+    # PROMPT optimizado
+    prompt = f"""[INSTRUCCIÓN]
+Como asistente legal, tu tarea es determinar qué medida de protección debe aplicarse en un caso de violencia intrafamiliar, de acuerdo al Artículo 5 de la Ley 575 de 2000 (modificado por la Ley 1257 de 2008 y la Ley 2126 de 2021).
+
+[CASO]
+Tipo de violencia: {tipo_violencia}
+Nivel de riesgo: {riesgo}
+Tipo de medida cautelar: {tipo_medida}
+Descripción del caso: {descripcion}
+
+[OBJETIVO]
+Con base en la descripción del caso y la ley, indica:
+- Cuál(es) literal(es) del Artículo 5 son aplicables (a-n)
+- Justifica legalmente tu elección en lenguaje claro y formal.
+
+[RESPUESTA]
+"""
+
+    salida = modelo_llm(prompt)[0]["generated_text"].split("[RESPUESTA]")[-1].strip()
+
+    return tipo_violencia, riesgo, tipo_medida, salida
+
+import gradio as gr
+
+gr.Interface(
+    fn=predecir_con_recomendacion,
+    inputs=[
+        gr.Slider(18, 65, value=30, label="Edad"),
+        gr.Radio(["F", "M"], label="Género"),
+        gr.Slider(0, 5, step=1, value=1, label="Número de hijos"),
+        gr.Radio(["sí", "no"], label="¿Convive con el agresor?"),
+        gr.Radio(["sí", "no"], label="¿Hay consumo de sustancias?"),
+        gr.Radio(["sí", "no"], label="¿Tiene apoyo familiar?"),
+        gr.Textbox(lines=4, placeholder="Describa aquí el caso de violencia", label="Descripción del caso")
+    ],
+    outputs=[
+        gr.Textbox(label="🛑 Tipo de violencia predicho"),
+        gr.Textbox(label="⚠️ Nivel de riesgo"),
+        gr.Textbox(label="🧾 Tipo de medida cautelar"),
+        gr.Textbox(label="📜 Recomendación legal razonada")
+    ],
+    title="LEGALFAMI – Asistente Legal con Razonamiento Jurídico",
+    description="Este asistente predice el tipo de violencia, nivel de riesgo y tipo de medida cautelar, y sugiere una medida específica del Art. 5 con justificación legal automatizada.",
+    theme="default"
+).launch(debug=True)

codificadores_label_encoder.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ad512b489aab1c643491545cf4c83c1d70fca1046be77d14f6e086311be9fe4e
+size 2675

modelo_riesgo.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a77a4d0e276b74988acfe154cfafa4c134cd4d0bcb8d899b990b7394d01c5a63
+size 833729

modelo_tipo_medida.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:660b0a1006f814e7b791653ab6da0a658e914eda1be4b229c1137b794aa8053e
+size 5779049

modelo_tipo_violencia.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f7fff6310bd6f3fa5b8666f0fad9bab7afa631e6ca49feda214b64c053d04088
+size 1350153

requirements.txt

@@ -0,0 +1,7 @@
+gradio
+transformers
+sentence-transformers
+scikit-learn
+pandas
+numpy
+joblib