jamalinu commited on
Commit
5545fe9
·
verified ·
1 Parent(s): f65d3f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -14
app.py CHANGED
@@ -1,11 +1,21 @@
1
  import os
 
2
  import gradio as gr
3
 
4
  from src.engine import DeepPragmaEngine
5
  from src.llm_client import GroqClient
6
 
 
 
 
 
 
 
 
 
7
  engine = DeepPragmaEngine()
8
 
 
9
  llm = None
10
  if os.environ.get("GROQ_API_KEY"):
11
  llm = GroqClient()
@@ -18,41 +28,40 @@ def analizar(frase):
18
 
19
  clasificacion = engine.analizar(frase)
20
 
21
- validacion = ""
22
 
23
  if "Hate" in clasificacion:
24
-
25
  if llm:
26
- validacion = llm.razonar(frase).strip()
27
  else:
28
- validacion = "GROQ_API_KEY no configurada."
29
 
30
- return clasificacion, validacion
31
 
32
 
33
- with gr.Blocks(title="DeepPragma AI") as demo:
34
 
35
  gr.Markdown("# 🛡️ DeepPragma Classifier")
36
 
37
- frase = gr.Textbox(
38
- label="Introduce una frase",
39
- placeholder="Write something..."
40
  )
41
 
42
  boton = gr.Button("Analizar")
43
 
44
- clasificacion = gr.Textbox(
45
  label="Clasificación"
46
  )
47
 
48
- razonamiento = gr.Textbox(
49
- label="Validación experta (LLM)"
50
  )
51
 
52
  boton.click(
53
  analizar,
54
- inputs=frase,
55
- outputs=[clasificacion, razonamiento]
56
  )
57
 
 
58
  demo.launch()
 
1
  import os
2
+ import nltk
3
  import gradio as gr
4
 
5
  from src.engine import DeepPragmaEngine
6
  from src.llm_client import GroqClient
7
 
8
+
9
+ # Descargar recursos necesarios si no existen
10
+ try:
11
+ nltk.data.find("sentiment/vader_lexicon")
12
+ except LookupError:
13
+ nltk.download("vader_lexicon")
14
+
15
+
16
  engine = DeepPragmaEngine()
17
 
18
+
19
  llm = None
20
  if os.environ.get("GROQ_API_KEY"):
21
  llm = GroqClient()
 
28
 
29
  clasificacion = engine.analizar(frase)
30
 
31
+ explicacion = ""
32
 
33
  if "Hate" in clasificacion:
 
34
  if llm:
35
+ explicacion = llm.razonar(frase)
36
  else:
37
+ explicacion = "GROQ_API_KEY no configurada"
38
 
39
+ return clasificacion, explicacion
40
 
41
 
42
+ with gr.Blocks() as demo:
43
 
44
  gr.Markdown("# 🛡️ DeepPragma Classifier")
45
 
46
+ entrada = gr.Textbox(
47
+ label="Introduce una frase"
 
48
  )
49
 
50
  boton = gr.Button("Analizar")
51
 
52
+ salida1 = gr.Textbox(
53
  label="Clasificación"
54
  )
55
 
56
+ salida2 = gr.Textbox(
57
+ label="Validación experta"
58
  )
59
 
60
  boton.click(
61
  analizar,
62
+ entrada,
63
+ [salida1, salida2]
64
  )
65
 
66
+
67
  demo.launch()