JeCabrera commited on
Commit
9759c24
verified
1 Parent(s): a8e76ea

Update session_state.py

Browse files
Files changed (1) hide show
  1. session_state.py +11 -3
session_state.py CHANGED
@@ -194,9 +194,17 @@ class SessionState:
194
  self.client = genai.Client(api_key=os.environ.get('GOOGLE_API_KEY'))
195
  title_response = self.client.models.generate_content(
196
  model=model_name,
197
- contents=f"Genera un t铆tulo corto (m谩ximo 5 palabras) que describa de qu茅 trata esta consulta, sin usar comillas ni puntuaci贸n: '{prompt}'"
 
 
 
 
 
198
  )
199
- return title_response.text.strip()
 
 
 
200
  except Exception as e:
201
  print(f"Error al generar t铆tulo: {e}")
202
  return None
@@ -235,4 +243,4 @@ class SessionState:
235
 
236
  def has_prompt(self):
237
  """Verifica si hay un prompt en el estado de la sesi贸n"""
238
- return self.prompt is not None and self.prompt.strip() != ""
 
194
  self.client = genai.Client(api_key=os.environ.get('GOOGLE_API_KEY'))
195
  title_response = self.client.models.generate_content(
196
  model=model_name,
197
+ contents=(
198
+ "Genera un t铆tulo natural y humano en espa帽ol (3 a 6 palabras) "
199
+ "que resuma esta consulta. No uses separadores tipo '|', no uses etiquetas, "
200
+ "no uses comillas y evita formato rob贸tico. Devuelve solo el t铆tulo final: "
201
+ f"'{prompt}'"
202
+ )
203
  )
204
+ cleaned_title = " ".join(
205
+ title_response.text.strip().replace('"', '').replace('|', ' ').split()
206
+ )
207
+ return " ".join(cleaned_title.split()[:6])
208
  except Exception as e:
209
  print(f"Error al generar t铆tulo: {e}")
210
  return None
 
243
 
244
  def has_prompt(self):
245
  """Verifica si hay un prompt en el estado de la sesi贸n"""
246
+ return self.prompt is not None and self.prompt.strip() != ""