Update app.py

app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-import gradio as gr
 from PyPDF2 import PdfReader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 import os
@@ -41,13 +40,10 @@ def get_conversational_chain():
     prompt_template = """
     Responde la pregunta en español de la manera más detallada posible a partir del contexto proporcionado. Si la respuesta no está en
     el contexto proporcionado, simplemente di, "la respuesta no está disponible en el contexto." No proporciones respuestas incorrectas.
-
     Contexto:
     {context}?
-
     Pregunta:
     {question}
-
     Respuesta:
     """
 
@@ -161,34 +157,21 @@ def main():
                 get_vector_store(text_chunks)
                 st.success("¡PDF procesado exitosamente!")
 
-        # 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")
-            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?")
-        
+    # Botones para preguntas predefinidas con estilo personalizado
+    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")
+    st.markdown('</div>', unsafe_allow_html=True)
+
+    st.markdown('<div class="custom-button">', unsafe_allow_html=True)
+    if st.button("Entidad", key="entidad_button"):
+        user_input("A qué entidad pertenece el contenido del documento?")
+    st.markdown('</div>', unsafe_allow_html=True)
+
+    st.markdown('<div class="custom-button">', unsafe_allow_html=True)
+    if st.button("Fecha implantación", key="fecha_button"):
+        user_input("En qué fecha se implantará el contenido del documento?")
+    st.markdown('</div>', unsafe_allow_html=True)
 
     user_question = st.text_input("Introduce tu pregunta", placeholder="¿Qué quieres saber?")