Update app.py

app.py

@@ -161,9 +161,34 @@ def main():
                 get_vector_store(text_chunks)
                 st.success("¡PDF procesado exitosamente!")
 
-    # Botones para preguntas predefinidas con estilo personalizado
-    with gr.Row():
+        # Botones para preguntas predefinidas con estilo personalizado
+    
+        col1, col2, col3 = st.columns(3)
+    
+        with col1:
+            st.markdown('<div class="custom-button">', unsafe_allow_html=True)
+            button1 = st.button("Resumen", key="resumen_button"):
+            user_input("Realiza un resumen sobre los aspectos más relevantes comentados en el documento")
+            st.markdown('</div>', unsafe_allow_html=True)
+        with col2:
+            st.markdown('<div class="custom-button">', unsafe_allow_html=True)
+            button2 = st.button("Entidad", key="entidad_button"):
+            st.markdown('</div>', unsafe_allow_html=True)
+        
+        with col3:
+            st.markdown('<div class="custom-button">', unsafe_allow_html=True)
+            button3 = st.button("Fecha implantación", key="fecha_button"):
+            st.markdown('</div>', unsafe_allow_html=True)
         
+        if button1:
+        # Do something...
+            user_input("Realiza un resumen sobre los aspectos más relevantes comentados en el documento")
+        if button2:
+            # Do something...
+            user_input("A qué entidad pertenece el contenido del documento?")
+        if button3:
+            # Do something...
+            user_input("En qué fecha se implantará el contenido del documento?")
         st.markdown('<div class="custom-button">', unsafe_allow_html=True)
         if st.button("Resumen", key="resumen_button"):
             user_input("Realiza un resumen sobre los aspectos más relevantes comentados en el documento")