Update app.py

app.py

@@ -266,7 +266,8 @@ def main():
                     help=f"Haz clic para usar esta pregunta",
                     use_container_width=True
                 ):
-                    procesar_consulta(pregunta)
+                    # Guardar la pregunta seleccionada en el estado de sesión
+                    st.session_state.pregunta_actual = pregunta
             
             st.markdown("---")
 
@@ -278,16 +279,18 @@ def main():
                 user_question = st.text_input(
                     "Escribe tu pregunta:",
                     placeholder="Ej: ¿Qué normativa regula este proceso?",
-                    label_visibility="collapsed"
+                    value=st.session_state.pregunta_actual,  # Mostrar la pregunta seleccionada si existe
+                    key="input_pregunta"
                 )
             with col2:
-                st.markdown("<br>", unsafe_allow_html=True)
                 enviar = st.form_submit_button("Enviar ▶")
-    
-        if user_question and enviar:
-            procesar_consulta(user_question)
+
+        # Procesar la pregunta seleccionada o introducida manualmente
+        if enviar or (st.session_state.pregunta_actual and not enviar):
+            procesar_consulta(user_question or st.session_state.pregunta_actual)
+            # Limpiar la pregunta actual después de procesarla
+            if "pregunta_actual" in st.session_state:
+                del st.session_state.pregunta_actual
+
     else:
         st.write("Por favor, sube un documento para continuar.")
-
-if __name__ == "__main__":
-    main()