Update app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ for archivo_index in archivos_indices:
|
|
| 28 |
# Funci贸n para predecir la similitud con cada 铆ndice
|
| 29 |
def predecir_similitud(texto):
|
| 30 |
resultados = []
|
| 31 |
-
umbral_similitud =
|
| 32 |
|
| 33 |
# Vectorizar el texto (usando la misma funci贸n que antes)
|
| 34 |
texto_vectorizado = vectorize_text(texto)
|
|
@@ -42,16 +42,18 @@ def predecir_similitud(texto):
|
|
| 42 |
# Comparar con el umbral y proporcionar el resultado
|
| 43 |
if similitud >= umbral_similitud:
|
| 44 |
resultado = (nombre_archivo, similitud)
|
| 45 |
-
#else:
|
| 46 |
-
# resultado = (nombre_archivo, -1) #usamos -1 para indicar que no hay coincidencia
|
| 47 |
|
| 48 |
resultados.append(resultado)
|
| 49 |
|
| 50 |
-
|
| 51 |
-
|
| 52 |
|
| 53 |
-
|
| 54 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 55 |
|
| 56 |
return resultados_texto
|
| 57 |
|
|
|
|
| 28 |
# Funci贸n para predecir la similitud con cada 铆ndice
|
| 29 |
def predecir_similitud(texto):
|
| 30 |
resultados = []
|
| 31 |
+
umbral_similitud = 0.65
|
| 32 |
|
| 33 |
# Vectorizar el texto (usando la misma funci贸n que antes)
|
| 34 |
texto_vectorizado = vectorize_text(texto)
|
|
|
|
| 42 |
# Comparar con el umbral y proporcionar el resultado
|
| 43 |
if similitud >= umbral_similitud:
|
| 44 |
resultado = (nombre_archivo, similitud)
|
|
|
|
|
|
|
| 45 |
|
| 46 |
resultados.append(resultado)
|
| 47 |
|
| 48 |
+
# Ordenar la lista de resultados por similitud
|
| 49 |
+
resultados_ordenados = sorted(resultados, key=lambda x: x[1], reverse=True)
|
| 50 |
|
| 51 |
+
# Construir la cadena de resultados
|
| 52 |
+
resultados_texto = '\n'.join([f"{nombre} (similitud:{sim:.2f})" if sim != -1 else f"{nombre} (similitud:{sim:.2f})" for nombre, sim in resultados_ordenados])
|
| 53 |
+
|
| 54 |
+
# Verificar si no hay resultados por encima del umbral
|
| 55 |
+
if all(sim <= umbral_similitud for _, sim in resultados_ordenados):
|
| 56 |
+
resultados_texto = "No se ha encontrado ning煤n argumento de los predeterminados"
|
| 57 |
|
| 58 |
return resultados_texto
|
| 59 |
|