Spaces:
Sleeping
Sleeping
Update session_state.py
Browse files- 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=
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 198 |
)
|
| 199 |
-
|
|
|
|
|
|
|
|
|
|
| 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() != ""
|