Update app.py

app.py

@@ -10,15 +10,11 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 # Función para obtener una mención del producto de manera probabilística
 def get_random_product_mention():
-    # Definir las opciones de mención y sus probabilidades
     mentions = ["Directa", "Indirecta", "Metafórica"]
-    probabilities = [0.34, 0.33, 0.33]  # Probabilidades de cada mención (ajusta según desees)
-    
-    # Elegir una opción al azar basándose en las probabilidades
+    probabilities = [0.34, 0.33, 0.33]  # Probabilidades de cada mención
     return random.choices(mentions, probabilities)[0]
 
 def get_gemini_response(target_audience, product, text_type, length, mood, model_choice):
-    # Generar la mención aleatoria
     product_mention = get_random_product_mention()  # Llamada correcta a la función
     model = genai.GenerativeModel(model_choice)
     
@@ -93,12 +89,12 @@ with st.sidebar:
     mood = st.selectbox("Tono de voz", ["Conversacional", "Formal", "Emocional", "Persuasivo"])
     model_choice = st.selectbox("Modelo", ["gemini-1.5-flash", "gemini-1.5-pro"])
 
-    # Botón para generar el texto
-    if st.button("Escribir mi texto"):
-        try:
-            # Obtener la respuesta del modelo
-            generated_text = get_gemini_response(target_audience, product, text_type, length, mood, model_choice)
-            st.write("**Texto generado:**")
-            st.write(generated_text)
-        except Exception as e:
-            st.error(f"Ocurrió un error: {e}")
+# Botón para generar el texto (fuera de la barra lateral)
+if st.button("Escribir mi texto"):
+    try:
+        # Obtener la respuesta del modelo
+        generated_text = get_gemini_response(target_audience, product, text_type, length, mood, model_choice)
+        st.write("**Texto generado:**")
+        st.write(generated_text)
+    except Exception as e:
+        st.error(f"Ocurrió un error: {e}")