Update app.py

app.py

@@ -45,20 +45,34 @@ if st.button('Buscar'):
         df = df[['text', 'volume', 'Nivel de Competencia', 'competition_index', 'Oferta Baja', 'Oferta Alta', 'Tendencia', 'Ads']]
         df.columns = ['Texto', 'Volumen', 'Nivel de Competencia', 'Índice de Competencia', 'Oferta Baja', 'Oferta Alta', 'Tendencia', 'Ads']
         df.sort_values(by='Volumen', ascending=False, inplace=True)
-        st.table(df)
         
-        # Convert DataFrame to CSV
-        csv = df.to_csv(index=False)
+        # Control the number of rows displayed
+        rows_to_display = 20
+        if 'show_all' not in st.session_state:
+            st.session_state.show_all = False
         
-        st.download_button(
-            label="Descargar resultados como CSV",
-            data=csv,
-            file_name='resultados.csv',
-            mime='text/csv',
-        )
+        if st.session_state.show_all:
+            rows_to_display = len(df)
         
-        st.code(df.to_string(index=False), language='text')
+        st.table(df.head(rows_to_display))
         
-        st.write('Selecciona el texto arriba y cópialo manualmente si es necesario.')
+        col1, col2 = st.columns([1, 1])
+        
+        with col1:
+            if st.session_state.show_all:
+                if st.button('Ver menos'):
+                    st.session_state.show_all = False
+            else:
+                if st.button('Ver más'):
+                    st.session_state.show_all = True
+
+        with col2:
+            csv = df.to_csv(index=False)
+            st.download_button(
+                label="Descargar resultados como CSV",
+                data=csv,
+                file_name='resultados.csv',
+                mime='text/csv',
+            )
 
 # Run the app using the command: streamlit run app.py