Update app.py

app.py

@@ -69,6 +69,25 @@ st.set_page_config(page_title="Poetic Vision", page_icon=":pencil:", layout="wid
 st.title("Story Maker App :pencil:")
 st.markdown("### Upload your image or provide a prompt to create a Vision around that: ")
 
+# Add custom CSS for the button
+st.markdown("""
+    <style>
+    div.stButton > button {
+        background-color: #FFCC00; /* Color llamativo */
+        color: white;
+        width: 90%;  /* 90% del ancho */
+        height: 50px; /* Altura del botón */
+        font-size: 18px;  /* Tamaño de la fuente */
+        border: none;
+        border-radius: 8px;
+    }
+    div.stButton > button:hover {
+        background-color: #FFD700; /* Color más claro al pasar el mouse */
+        color: black;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
 # Create two columns for layout (40% and 60%)
 col1, col2 = st.columns([2, 3])  # 2 + 3 = 5 parts total
 
@@ -84,23 +103,8 @@ with col1:
         language = st.selectbox("Idioma del texto:", ["Inglés", "Español"])
         mood = st.selectbox("Estado de ánimo:", ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
 
-    # Botón "Escribir historia" con estilo CSS
-    st.markdown("""
-        <style>
-            .big-button {
-                background-color: #FFCC00; /* Color llamativo */
-                color: white;
-                width: 90%;
-                height: 50px; /* Altura del botón */
-                font-size: 20px; /* Tamaño de fuente */
-                border: none;
-                border-radius: 5px; /* Bordes redondeados */
-                cursor: pointer;
-            }
-        </style>
-        <button class="big-button" onclick="document.getElementById('generate_button').click();">Escribir historia</button>
-        <input type="submit" id="generate_button" style="display:none;">
-    """, unsafe_allow_html=True)
+    # Generate button con el nuevo estilo y ancho ajustado
+    submit = st.button("Escribir historia")
 
 # Display uploaded image
 if uploaded_file is not None:
@@ -108,7 +112,7 @@ if uploaded_file is not None:
     col1.image(image, caption="Imagen subida", use_column_width=True)
 
 # Handle submit action
-if st.button("Escribir historia", key="generate_button"):
+if submit:
     if uploaded_file or input_prompt:
         try:
             image_data = input_image_setup(uploaded_file)