GhostNetworkUser commited on
Commit
fd52d45
·
verified ·
1 Parent(s): b84ed06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -3
app.py CHANGED
@@ -39,6 +39,11 @@ context_map = {
39
  "wikitext": "Dom: Enciclopedia | Estilo: Conocimiento general",
40
  "math": "Dom: Matemáticas | Estilo: Problema matemático", # Agregar problemas de matemáticas
41
  "empathetic_dialogues": "Dom: Psicología | Estilo: Apoyo emocional", # Para el comportamiento emocional
 
 
 
 
 
42
  }
43
 
44
  # Detectar idioma automáticamente
@@ -49,7 +54,7 @@ def detect_language(text):
49
  return "en" # Si no se puede detectar, se asume inglés
50
 
51
  # Generación de texto (si se solicita un cuento, por ejemplo)
52
- def generate_text(prompt, max_length=2048):
53
  output = generator(prompt, max_length=int(max_length), num_return_sequences=1)[0]["generated_text"]
54
  return output
55
 
@@ -75,6 +80,18 @@ def process_input(user_input):
75
  # Si la entrada es una pregunta, usaremos un modelo de Preguntas y Respuestas
76
  if '?' in user_input:
77
  context = "Este es un contexto general. Puedo responder preguntas específicas sobre cine, ciencia, o emociones."
 
 
 
 
 
 
 
 
 
 
 
 
78
  answer = answer_question(user_input, context)
79
  return answer
80
  elif "math" in user_input.lower():
@@ -83,11 +100,11 @@ def process_input(user_input):
83
  elif "cuento" in user_input.lower():
84
  # Si el usuario solicita un cuento
85
  prompt = "Había una vez, en un reino lejano..."
86
- return generate_text(prompt, max_length=500)
87
  elif "emoción" in user_input.lower():
88
  # Si es una solicitud de apoyo emocional
89
  context = "Dom: Psicología | Estilo: Apoyo emocional. ¿Cómo te sientes hoy?"
90
- return generate_text(context + " Estoy aquí para apoyarte", max_length=2048)
91
  else:
92
  return "No entiendo la solicitud, por favor intenta preguntar algo más específico."
93
 
 
39
  "wikitext": "Dom: Enciclopedia | Estilo: Conocimiento general",
40
  "math": "Dom: Matemáticas | Estilo: Problema matemático", # Agregar problemas de matemáticas
41
  "empathetic_dialogues": "Dom: Psicología | Estilo: Apoyo emocional", # Para el comportamiento emocional
42
+ "math_advanced": "Dom: Matemáticas Avanzadas | Estilo: Trigonometría y Cálculo", # Dataset de matemáticas avanzadas
43
+ "programming": "Dom: Programación | Estilo: Resolución de código", # Dataset para programación
44
+ "social_sciences": "Dom: Ciencias Sociales | Estilo: Preguntas sobre historia y geografía", # Dataset de sociales
45
+ "economics": "Dom: Economía | Estilo: Problemas económicos y teorías", # Dataset de economía
46
+ "health": "Dom: Salud | Estilo: Información médica y de salud", # Dataset de salud
47
  }
48
 
49
  # Detectar idioma automáticamente
 
54
  return "en" # Si no se puede detectar, se asume inglés
55
 
56
  # Generación de texto (si se solicita un cuento, por ejemplo)
57
+ def generate_text(prompt, max_length=100):
58
  output = generator(prompt, max_length=int(max_length), num_return_sequences=1)[0]["generated_text"]
59
  return output
60
 
 
80
  # Si la entrada es una pregunta, usaremos un modelo de Preguntas y Respuestas
81
  if '?' in user_input:
82
  context = "Este es un contexto general. Puedo responder preguntas específicas sobre cine, ciencia, o emociones."
83
+
84
+ # Identificar tipo de pregunta (matemáticas, programación, etc.)
85
+ if "matemática" in user_input.lower() or "cálculo" in user_input.lower() or "trigonometría" in user_input.lower():
86
+ context = "Dom: Matemáticas Avanzadas | Estilo: Trigonometría y Cálculo. Ejemplo: ¿Cuál es la derivada de x^2?"
87
+ elif "programación" in user_input.lower():
88
+ context = "Dom: Programación | Estilo: Resolución de código. Ejemplo: ¿Cómo se resuelve un bucle en Python?"
89
+ elif "historia" in user_input.lower() or "geografía" in user_input.lower():
90
+ context = "Dom: Ciencias Sociales | Estilo: Preguntas sobre historia y geografía. Ejemplo: ¿Quién fue Sócrates?"
91
+ elif "economía" in user_input.lower():
92
+ context = "Dom: Economía | Estilo: Problemas económicos y teorías. Ejemplo: ¿Qué es la inflación?"
93
+ elif "salud" in user_input.lower():
94
+ context = "Dom: Salud | Estilo: Información médica y de salud. Ejemplo: ¿Qué es la diabetes?"
95
  answer = answer_question(user_input, context)
96
  return answer
97
  elif "math" in user_input.lower():
 
100
  elif "cuento" in user_input.lower():
101
  # Si el usuario solicita un cuento
102
  prompt = "Había una vez, en un reino lejano..."
103
+ return generate_text(prompt, max_length=200)
104
  elif "emoción" in user_input.lower():
105
  # Si es una solicitud de apoyo emocional
106
  context = "Dom: Psicología | Estilo: Apoyo emocional. ¿Cómo te sientes hoy?"
107
+ return generate_text(context + " Estoy aquí para apoyarte", max_length=100)
108
  else:
109
  return "No entiendo la solicitud, por favor intenta preguntar algo más específico."
110