Update app.py

app.py

@@ -25,6 +25,8 @@ if 'submitted' not in st.session_state:
     st.session_state.submitted = False
 if 'offer_result' not in st.session_state:
     st.session_state.offer_result = ""
+if 'generated' not in st.session_state:
+    st.session_state.generated = False
 
 # Hide Streamlit menu and footer
 st.markdown("""
@@ -141,6 +143,7 @@ with col1:
             return
         
         st.session_state.submitted = True
+        st.session_state.generated = False  # Reset generated flag
         
         # Store inputs based on what's available
         if has_manual_input:
@@ -176,7 +179,7 @@ with col1:
 
 # Results column
 with col2:
-    if st.session_state.submitted:
+    if st.session_state.submitted and not st.session_state.generated:
         with st.spinner('Creando tu oferta perfecta...'):
             base_prompt = f"""You are a professional copywriter specializing in creating irresistible offers. 
             Create a compelling and persuasive offer using the {st.session_state.formula_type} formula.
@@ -250,21 +253,25 @@ with col2:
                     response = model.generate_content(prompt, generation_config=generation_config)
                     
                 st.session_state.offer_result = response.text
+                st.session_state.generated = True  # Mark as generated
                 
-                st.markdown('### Oferta Generada')
-                st.markdown(st.session_state.offer_result)
-                
-                col_download, col_empty = st.columns([8, 2])
-                with col_download:
-                    st.download_button(
-                        label="Descargar Oferta",
-                        data=st.session_state.offer_result,
-                        file_name="oferta_generada.txt",
-                        mime="text/plain"
-                    )
             except Exception as e:
                 st.error(f'Ocurrió un error: {str(e)}')
                 st.session_state.submitted = False
+    
+    # Display results if we have an offer result
+    if st.session_state.generated:
+        st.markdown('### Oferta Generada')
+        st.markdown(st.session_state.offer_result)
+        
+        col_download, col_empty = st.columns([8, 2])
+        with col_download:
+            st.download_button(
+                label="Descargar Oferta",
+                data=st.session_state.offer_result,
+                file_name="oferta_generada.txt",
+                mime="text/plain"
+            )
 
 # Footer
 st.markdown('---')