Delete app.py
Browse files
app.py
DELETED
@@ -1,215 +0,0 @@
|
|
1 |
-
# -*- coding: utf-8 -*-
|
2 |
-
"""app.ipynb
|
3 |
-
|
4 |
-
Automatically generated by Colab.
|
5 |
-
|
6 |
-
Original file is located at
|
7 |
-
https://colab.research.google.com/drive/1WQ2YqAcXFp-14DglIYkV4rawVDSbyhCe
|
8 |
-
"""
|
9 |
-
|
10 |
-
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
11 |
-
from sentence_transformers import SentenceTransformer, util
|
12 |
-
import numpy as np
|
13 |
-
import joblib
|
14 |
-
import torch
|
15 |
-
import gradio as gr
|
16 |
-
from transformers import TextGenerationPipeline
|
17 |
-
|
18 |
-
# === Cargar modelos entrenados ===
|
19 |
-
modelo_riesgo = joblib.load("modelo_riesgo.pkl")
|
20 |
-
modelo_violencia = joblib.load("modelo_tipo_violencia.pkl")
|
21 |
-
modelo_medida = joblib.load("modelo_tipo_medida.pkl")
|
22 |
-
codificadores = joblib.load("codificadores_label_encoder.pkl")
|
23 |
-
modelo_vector = SentenceTransformer("Snowflake/snowflake-arctic-embed-xs")
|
24 |
-
|
25 |
-
# === Cargar modelo de lenguaje OpenHermes-2.5 ===
|
26 |
-
model_id = "teknium/OpenHermes-2.5-Mistral-7B"
|
27 |
-
|
28 |
-
tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
|
29 |
-
model = AutoModelForCausalLM.from_pretrained(
|
30 |
-
model_id,
|
31 |
-
trust_remote_code=True,
|
32 |
-
torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
|
33 |
-
device_map="auto"
|
34 |
-
)
|
35 |
-
|
36 |
-
modelo_llm = TextGenerationPipeline(
|
37 |
-
model=model,
|
38 |
-
tokenizer=tokenizer,
|
39 |
-
max_new_tokens=1000,
|
40 |
-
temperature=0.5,
|
41 |
-
do_sample=True,
|
42 |
-
)
|
43 |
-
|
44 |
-
# === Frases prototipo para verificador semántico ===
|
45 |
-
frases_fisica = [
|
46 |
-
"Me golpeó con el puño cerrado", "Me pateó", "Me lanzó contra la pared",
|
47 |
-
"Me estranguló", "Me fracturó una costilla", "Me tiró al piso violentamente"
|
48 |
-
]
|
49 |
-
frases_sexual = [
|
50 |
-
"Me obligó a tener relaciones sexuales", "Me tocó sin consentimiento",
|
51 |
-
"Me violó", "Me forzó a tener sexo", "Me agredió sexualmente"
|
52 |
-
]
|
53 |
-
embeds_fisica = modelo_vector.encode(frases_fisica)
|
54 |
-
embeds_sexual = modelo_vector.encode(frases_sexual)
|
55 |
-
|
56 |
-
# === Verificador semántico ===
|
57 |
-
def verificar_semantico(descripcion):
|
58 |
-
emb_desc = modelo_vector.encode(descripcion)
|
59 |
-
tipos_detectados = []
|
60 |
-
if max(util.cos_sim(emb_desc, embeds_fisica)[0]) > 0.7:
|
61 |
-
tipos_detectados.append("física")
|
62 |
-
if max(util.cos_sim(emb_desc, embeds_sexual)[0]) > 0.7:
|
63 |
-
tipos_detectados.append("sexual")
|
64 |
-
return tipos_detectados
|
65 |
-
|
66 |
-
# === FUNCIÓN PRINCIPAL MODIFICADA ===
|
67 |
-
def predecir_con_recomendacion(edad, genero, hijos, convivencia_agresor, consumo_sustancias, apoyo_familiar, descripcion):
|
68 |
-
# Codificar variables tabulares
|
69 |
-
vector_tabular = np.array([
|
70 |
-
int(edad),
|
71 |
-
int(hijos),
|
72 |
-
codificadores["genero"].transform([genero])[0],
|
73 |
-
0, 0, 0,
|
74 |
-
codificadores["convivencia_agresor"].transform([convivencia_agresor])[0],
|
75 |
-
codificadores["consumo_sustancias"].transform([consumo_sustancias])[0],
|
76 |
-
codificadores["apoyo_familiar"].transform([apoyo_familiar])[0]
|
77 |
-
])
|
78 |
-
|
79 |
-
# Vectorizar descripción
|
80 |
-
vector_desc = modelo_vector.encode([descripcion])[0]
|
81 |
-
entrada = np.concatenate([vector_tabular, vector_desc])
|
82 |
-
|
83 |
-
# Modelos clásicos
|
84 |
-
riesgo_cod = modelo_riesgo.predict([entrada])[0]
|
85 |
-
tipo_violencia_cod = modelo_violencia.predict([entrada])[0]
|
86 |
-
tipo_medida_cod = modelo_medida.predict([entrada])[0]
|
87 |
-
|
88 |
-
# Decodificación
|
89 |
-
riesgo = codificadores["riesgo"].inverse_transform([riesgo_cod])[0]
|
90 |
-
tipo_violencia_pred = codificadores["tipo_violencia"].inverse_transform([tipo_violencia_cod])[0]
|
91 |
-
tipo_medida = codificadores["tipo_medida"].inverse_transform([tipo_medida_cod])[0]
|
92 |
-
|
93 |
-
# Verificador semántico
|
94 |
-
tipos_semantico = verificar_semantico(descripcion)
|
95 |
-
|
96 |
-
# Unir modelo + semántico sin duplicados
|
97 |
-
tipos_combinados = list(set([tipo_violencia_pred] + tipos_semantico))
|
98 |
-
tipos_str = ", ".join(tipos_combinados)
|
99 |
-
|
100 |
-
# PROMPT legal ajustado con decisión motivada para Comisario
|
101 |
-
prompt = f"""[INSTRUCCIÓN]
|
102 |
-
Eres un Comisario/a de Familia en Colombia. Con base en los hechos relatados, el nivel de riesgo y el tipo de violencia identificado, debes emitir un Auto de Medida Provisional claro, completo y con redacción profesional jurídica.
|
103 |
-
|
104 |
-
Utiliza la siguiente estructura detallada:
|
105 |
-
|
106 |
-
1. **CONSIDERACIONES:**
|
107 |
-
- Realiza un resumen claro y detallado de los hechos narrados por la parte denunciante, destacando las circunstancias de tiempo, modo y lugar.
|
108 |
-
- Analiza el riesgo actual y el contexto familiar con referencias a la Ley 575 de 2000, Ley 1257 de 2008 y Ley 2126 de 2021.
|
109 |
-
- Cita expresamente los literales del Artículo 5 de la Ley 575 de 2000 (letras a-n) que resultan aplicables al caso.
|
110 |
-
- Justifica cada literal aplicado explicando detalladamente las razones jurídicas, la finalidad protectora de la medida y su proporcionalidad frente a los hechos denunciados.
|
111 |
-
|
112 |
-
2. **RESUELVE:**
|
113 |
-
- **PRIMERO:** Admitir la solicitud y avocar conocimiento, con mención expresa de la normativa que habilita esta decisión.
|
114 |
-
- **SEGUNDO:** Adoptar las medidas de protección, enumerando cada una de manera individual, describiendo su contenido y fundamento legal.
|
115 |
-
- **TERCERO:** Advertir al presunto agresor sobre las consecuencias legales de incumplir estas medidas, citando expresamente el Artículo 5 parágrafo 3 de la Ley 575 de 2000 y el Artículo 38 de la Ley 1257 de 2008.
|
116 |
-
- **CUARTO:** Disponer que cualquier cambio de domicilio deberá ser informado oportunamente al Juzgado competente, conforme a lo dispuesto en el Artículo 5 de la Ley 575 de 2000.
|
117 |
-
- **QUINTO:** Ordenar, en caso necesario, acompañamiento policial para hacer efectivas las medidas de protección y prevenir nuevos actos de violencia.
|
118 |
-
- **SEXTO:** Citar al presunto agresor a audiencia pública con el fin de tramitar el proceso y resolver sobre la confirmación, modificación o levantamiento de las medidas adoptadas, indicando que su inasistencia no suspende el procedimiento.
|
119 |
-
|
120 |
-
[DATOS DEL CASO]
|
121 |
-
- Tipos de violencia detectados: {tipos_str}
|
122 |
-
- Nivel de riesgo: {riesgo}
|
123 |
-
- Medida cautelar sugerida: {tipo_medida}
|
124 |
-
- Descripción del caso: {descripcion}
|
125 |
-
|
126 |
-
[OBJETIVO]
|
127 |
-
Redacta el Auto de Medida Provisional completo siguiendo estrictamente la estructura detallada, con lenguaje jurídico profesional y argumentos amplios que fundamenten cada decisión. Indica expresamente el año de cada ley citada. Limita la respuesta a máximo 600 tokens, priorizando claridad y rigor legal.
|
128 |
-
|
129 |
-
[RESPUESTA]
|
130 |
-
"""
|
131 |
-
|
132 |
-
respuesta = modelo_llm(prompt)[0]["generated_text"]
|
133 |
-
razonamiento = respuesta.split("[RESPUESTA]")[-1].strip()
|
134 |
-
|
135 |
-
return tipos_str, riesgo, tipo_medida, razonamiento
|
136 |
-
|
137 |
-
# === Interfaz Gradio (sin cambios) ===
|
138 |
-
with gr.Blocks(
|
139 |
-
theme=gr.themes.Base(
|
140 |
-
primary_hue="green",
|
141 |
-
neutral_hue="slate",
|
142 |
-
font=[gr.themes.GoogleFont("Roboto")],
|
143 |
-
),
|
144 |
-
css="""
|
145 |
-
body {
|
146 |
-
background-color: #004225;
|
147 |
-
font-family: 'Roboto', sans-serif;
|
148 |
-
}
|
149 |
-
.gr-block.gr-markdown h1 {
|
150 |
-
display: flex;
|
151 |
-
align-items: center;
|
152 |
-
gap: 16px;
|
153 |
-
color: #ffffff;
|
154 |
-
}
|
155 |
-
.gr-block.gr-markdown h1 img {
|
156 |
-
height: 60px;
|
157 |
-
border-radius: 4px;
|
158 |
-
}
|
159 |
-
.gr-block.gr-markdown p {
|
160 |
-
color: #d6f5e3;
|
161 |
-
}
|
162 |
-
.gr-input, .gr-textbox, .gr-button {
|
163 |
-
background-color: #ffffff !important;
|
164 |
-
color: #000000 !important;
|
165 |
-
}
|
166 |
-
.gr-box {
|
167 |
-
border: 1px solid #2e8b57;
|
168 |
-
border-radius: 8px;
|
169 |
-
}
|
170 |
-
""",
|
171 |
-
) as interfaz:
|
172 |
-
gr.Markdown(
|
173 |
-
"""
|
174 |
-
<h1><img src="https://i.imgur.com/6m2M1to.png" alt="Logo Externado">LEGALFAMI – Asistente Legal con Razonamiento Jurídico</h1>
|
175 |
-
<p>Predice tipo de violencia, nivel de riesgo, medida cautelar y genera una decisión motivada conforme a la Ley 575 de 2000.</p>
|
176 |
-
"""
|
177 |
-
)
|
178 |
-
|
179 |
-
with gr.Row():
|
180 |
-
with gr.Column():
|
181 |
-
edad = gr.Slider(18, 65, value=30, label="Edad de la Víctima")
|
182 |
-
genero = gr.Radio(["F", "M"], label="Género")
|
183 |
-
hijos = gr.Slider(0, 5, step=1, value=1, label="Número de Hijos")
|
184 |
-
convivencia_agresor = gr.Radio(["sí", "no"], label="¿Convive con el Agresor?")
|
185 |
-
consumo_sustancias = gr.Radio(["sí", "no"], label="¿Hay Consumo de Sustancias?")
|
186 |
-
apoyo_familiar = gr.Radio(["sí", "no"], label="¿Tiene Apoyo Familiar?")
|
187 |
-
descripcion = gr.Textbox(lines=5, placeholder="Describa detalladamente el caso...", label="Descripción del Caso")
|
188 |
-
boton = gr.Button("🔍 Analizar Caso")
|
189 |
-
|
190 |
-
with gr.Column():
|
191 |
-
tipo_violencia_out = gr.Textbox(label="🛑 Tipo de Violencia Detectada", interactive=False)
|
192 |
-
riesgo_out = gr.Textbox(label="⚠️ Nivel de Riesgo Estimado", interactive=False)
|
193 |
-
medida_out = gr.Textbox(label="🧾 Tipo de Medida Cautelar Sugerida", interactive=False)
|
194 |
-
recomendacion_out = gr.Textbox(label="📋 Recomendación Legal Razonada", lines=16, interactive=False)
|
195 |
-
|
196 |
-
boton.click(
|
197 |
-
fn=predecir_con_recomendacion,
|
198 |
-
inputs=[
|
199 |
-
edad,
|
200 |
-
genero,
|
201 |
-
hijos,
|
202 |
-
convivencia_agresor,
|
203 |
-
consumo_sustancias,
|
204 |
-
apoyo_familiar,
|
205 |
-
descripcion,
|
206 |
-
],
|
207 |
-
outputs=[
|
208 |
-
tipo_violencia_out,
|
209 |
-
riesgo_out,
|
210 |
-
medida_out,
|
211 |
-
recomendacion_out,
|
212 |
-
],
|
213 |
-
)
|
214 |
-
|
215 |
-
interfaz.launch()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|