Update app.py

app.py

@@ -90,27 +90,14 @@ def generate_bullets(number_of_bullets, target_audience, product, call_to_action
         generation_config=generation_config,
     )
 
-# Mostrar los beneficios generados con formato correcto en HTML
-if "submit" in locals() and submit:
-    if target_audience and product and call_to_action:
-        try:
-            # Obtener la respuesta del modelo
-            generated_bullets = generate_bullets(number_of_bullets, target_audience, product, call_to_action, temperature)
-            
-            # Dividir los bullets por líneas y generar el HTML sin formato markdown
-            bullets_list = generated_bullets.split("*")
-            bullets_html = "".join([f"<li>{bullet.strip()}</li>" for bullet in bullets_list if bullet.strip()])
+    # Obtener respuesta generada
+    response = model.generate(
+        instruction=system_instruction,
+        examples=benefit_types,
+    )
 
-            col2.markdown(f"""
-                <div class="generated-bullets">
-                    <h4>Mira los bullets generados:</h4>
-                    <ul>{bullets_html}</ul>
-                </div>
-            """, unsafe_allow_html=True)
-        except Exception as e:
-            st.error(f"Error al generar los bullets: {str(e)}")
-    else:
-        st.error("Por favor, completa todos los campos.")
+    # Retornar los bullets generados
+    return response.generations[0].text
 
 # Configurar la interfaz de usuario con Streamlit
 st.set_page_config(page_title="Quick Prompt", layout="wide")
@@ -158,7 +145,7 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # Crear columnas
-col1, col2 = st.columns([1, 2])  
+col1, col2 = st.columns([1, 2])
 
 # Columnas de entrada
 with col1:
@@ -177,10 +164,15 @@ if submit:
         try:
             # Obtener la respuesta del modelo
             generated_bullets = generate_bullets(number_of_bullets, target_audience, product, call_to_action, temperature)
+            
+            # Dividir los bullets por líneas y generar el HTML sin formato markdown
+            bullets_list = generated_bullets.split("*")
+            bullets_html = "".join([f"<li>{bullet.strip()}</li>" for bullet in bullets_list if bullet.strip()])
+
             col2.markdown(f"""
                 <div class="generated-bullets">
                     <h4>Mira los bullets generados:</h4>
-                    <p>{generated_bullets}</p>
+                    <ul>{bullets_html}</ul>
                 </div>
             """, unsafe_allow_html=True)
         except Exception as e: