Update app.py

app.py

@@ -4,7 +4,6 @@ import os
 import google.generativeai as genai
 import random
 import datetime                    
-import base64
 from streamlit import session_state as state
 from formulas.webinar_formulas import webinar_formulas
 from formulas.webinar_name_formulas import webinar_name_formulas
@@ -19,58 +18,37 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 # Función auxiliar para mostrar el contenido generado y los botones de descarga
 # In the display_generated_content function, modify the names section:
 
-# Modificar la función display_generated_content para usar session_state
-# Inicializar variables de estado si no existen
-if 'last_generated_script' not in st.session_state:
-    st.session_state.last_generated_script = None
-if 'last_generated_names' not in st.session_state:
-    st.session_state.last_generated_names = None
-
-# Modify the display_generated_content function
 def display_generated_content(col, generated_content, content_type):
     timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
     
-    # Guardar el contenido generado en el estado de la sesión
+    # Determinar el tipo de contenido para personalizar los botones y títulos
     if content_type == "script":
-        st.session_state.last_generated_script = generated_content
         download_label = "DESCARGAR GUIÓN DE WEBINAR ▶▶"
         file_name = f"guion_webinar_{timestamp}.txt"
         subheader_text = "Tu guión de webinar:"
         
-        # Usar un enfoque diferente para la descarga
-        b64 = base64.b64encode(generated_content.encode()).decode()
-        
-        # Crear botón de descarga superior con HTML mejorado
-        top_button_html = f"""
-        <a href="data:text/plain;base64,{b64}" download="{file_name}" 
-           style="display: inline-block; padding: 0.5em 1em; color: white; background-color: #4CAF50; 
-                  text-align: center; text-decoration: none; font-size: 16px; margin: 4px 2px; 
-                  cursor: pointer; border-radius: 4px; font-weight: bold;">
-            {download_label}
-        </a>
-        """
-        
-        # Crear botón de descarga inferior con HTML mejorado
-        bottom_button_html = f"""
-        <a href="data:text/plain;base64,{b64}" download="{file_name}" 
-           style="display: inline-block; padding: 0.5em 1em; color: white; background-color: #4CAF50; 
-                  text-align: center; text-decoration: none; font-size: 16px; margin: 4px 2px; 
-                  cursor: pointer; border-radius: 4px; font-weight: bold;">
-            {download_label}
-        </a>
-        """
-        
-        # Mostrar botón de descarga superior
-        col.markdown(top_button_html, unsafe_allow_html=True)
+        # Mostrar botón de descarga superior para guiones
+        col.download_button(
+            label=download_label,
+            data=generated_content,
+            file_name=file_name,
+            mime="text/plain",
+            key=f"download_top_{content_type}"
+        )
         
         # Mostrar el contenido generado
         col.subheader(subheader_text)
         col.markdown(generated_content)
         
         # Mostrar botón de descarga inferior
-        col.markdown(bottom_button_html, unsafe_allow_html=True)
+        col.download_button(
+            label=download_label,
+            data=generated_content,
+            file_name=file_name,
+            mime="text/plain",
+            key=f"download_bottom_{content_type}"
+        )
     else:  # nombres
-        st.session_state.last_generated_names = generated_content
         subheader_text = "Tus nombres de webinar:"
         file_name = f"nombres_webinar_{timestamp}.txt"
         download_label = "DESCARGAR NOMBRES DE WEBINAR ▶▶"
@@ -78,27 +56,28 @@ def display_generated_content(col, generated_content, content_type):
         # Contar el número de nombres generados
         num_names = len([line for line in generated_content.split('\n') if line.strip().startswith(('1.', '2.', '3.', '4.', '5.', '6.', '7.', '8.', '9.', '10.', '11.', '12.', '13.', '14.', '15.'))])
         
+        # Para los nombres, usar HTML personalizado en lugar de st.download_button
+        import base64
+        
         # Codificar el contenido para descarga
         b64 = base64.b64encode(generated_content.encode()).decode()
         
-        # Crear botón de descarga superior con HTML mejorado
+        # Crear botón de descarga superior con clase específica
         top_button_html = f"""
-        <a href="data:text/plain;base64,{b64}" download="{file_name}" 
-           style="display: inline-block; padding: 0.5em 1em; color: white; background-color: #4CAF50; 
-                  text-align: center; text-decoration: none; font-size: 16px; margin: 4px 2px; 
-                  cursor: pointer; border-radius: 4px; font-weight: bold;">
-            {download_label}
-        </a>
+        <div class="custom-download-button names-download-button top-button">
+            <a href="data:text/plain;base64,{b64}" download="{file_name}">
+                {download_label}
+            </a>
+        </div>
         """
         
-        # Crear botón de descarga inferior con HTML mejorado
+        # Crear botón de descarga inferior con clase específica
         bottom_button_html = f"""
-        <a href="data:text/plain;base64,{b64}" download="{file_name}" 
-           style="display: inline-block; padding: 0.5em 1em; color: white; background-color: #4CAF50; 
-                  text-align: center; text-decoration: none; font-size: 16px; margin: 4px 2px; 
-                  cursor: pointer; border-radius: 4px; font-weight: bold;">
-            {download_label}
-        </a>
+        <div class="custom-download-button names-download-button bottom-button">
+            <a href="data:text/plain;base64,{b64}" download="{file_name}">
+                {download_label}
+            </a>
+        </div>
         """
         
         # Mostrar botón de descarga superior
@@ -116,30 +95,18 @@ def generate_and_display(col, generator_func, audience, product, temperature, se
     if validate_inputs(audience, product):
         try:
             with col:
-                # Verificar si ya hay contenido generado en la sesión
-                existing_content = None
-                if content_type == "script" and st.session_state.last_generated_script:
-                    existing_content = st.session_state.last_generated_script
-                elif content_type == "names" and st.session_state.last_generated_names:
-                    existing_content = st.session_state.last_generated_names
+                with st.spinner(f"Generando {'guión' if content_type == 'script' else 'nombres'} de webinar...", show_time=True):
+                    # Llamar a la función generadora con los parámetros adecuados
+                    generated_content = generator_func(
+                        audience=audience,
+                        topic=product,
+                        temperature=temperature,
+                        selected_formula=selected_formula,
+                        **kwargs
+                    )
                 
-                # Si no hay contenido existente, generar nuevo contenido
-                if not existing_content:
-                    with st.spinner(f"Generando {'guión' if content_type == 'script' else 'nombres'} de webinar...", show_time=True):
-                        # Llamar a la función generadora con los parámetros adecuados
-                        generated_content = generator_func(
-                            audience=audience,
-                            topic=product,
-                            temperature=temperature,
-                            selected_formula=selected_formula,
-                            **kwargs
-                        )
-                    
-                    # Mostrar el contenido generado usando la función auxiliar
-                    display_generated_content(col, generated_content, content_type)
-                else:
-                    # Mostrar el contenido existente
-                    display_generated_content(col, existing_content, content_type)
+                # Mostrar el contenido generado usando la función auxiliar
+                display_generated_content(col, generated_content, content_type)
                 
         except ValueError as e:
             col.error(f"Error: {str(e)}")