Update app.py

app.py

@@ -110,18 +110,19 @@ with col1:
     number_of_headlines = st.selectbox("Número de Titulares", options=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10], index=4)
     temperature = st.slider("Creatividad", min_value=0.0, max_value=1.0, value=0.5, step=0.1)
     
-    # Botón de enviar
-    if st.button("Generar Titulares"):
-        if target_audience and product:
+# Mostrar los titulares generados
+if st.button("Generar Titulares"):
+    if target_audience and product:
+        try:
             # Obtener la respuesta del modelo
             generated_headlines = generate_headlines(number_of_headlines, target_audience, product, temperature)
-            # Mostrar los bullets generados
             col2.markdown(f"""
                 <div style="border: 1px solid #000000; padding: 5px; border-radius: 8px; background-color: #ffffff;">
                     <h4>Aquí están tus titulares:</h4>
                     <p>{generated_headlines}</p>
                 </div>
             """, unsafe_allow_html=True)
-        else:
-            col2.error("Por favor, proporciona el público objetivo y el producto.")
-
+        except ValueError as e:
+            col2.error(f"Error: {str(e)}")
+    else:
+        col2.error("Por favor, proporciona el público objetivo y el producto.")