km1lo commited on
Commit
b8ea244
·
verified ·
1 Parent(s): 10a75eb

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -215
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()