Update resultados_consulta.py

resultados_consulta.py

@@ -1,6 +1,7 @@
 import pandas as pd
 import numpy as np
 from preprocesamiento_articulos import remove_URL, remove_html_markup, eliminar_puntuacion, eliminar_stopwords, obtener_raices
+import streamlit as st
 
 
 # La idea aqui es colocar el query en minusculas y limpiar el query de palabras vacias y algunos caracteres
@@ -61,8 +62,9 @@ def retrieve_index(data,cosine_scores, document_index, topn=10):
     data['scores'] = cosine_scores
     df_top_scores=data.reset_index().sort_values('scores',ascending=False).head(topn)
     df_top_scores=df_top_scores[df_top_scores['scores'] > 0]
+    st.write(df_top_scores)
     return df_top_scores.index # Dataframe original con los scores 
-                                                                                   # del coseno en una columna nueva 
+                               # del coseno en una columna nueva 
 
 
 def resultados_consulta(df,articulos_indexados, query):
@@ -71,7 +73,7 @@ def resultados_consulta(df,articulos_indexados, query):
     qs=query_score(articulos_indexados,query)
     if 'query_tf_idf' in qs.columns:
         cosenos = cosine_similarity(qs, df['ID'].values, 'query_tf_idf')
-        indices = retrieve_index(df, cosenos, 'ID', 100) # LUEGO PARAMETRIZAR ESTE NUMERO
+        indices = retrieve_index(df, cosenos, 'ID', len(df))
     return indices
     
 def detalles_resultados(df,indices):