Update app.py

app.py

@@ -61,19 +61,18 @@ st.write("")
 st.markdown("<h4 style='font-size:20px;'>Outline Your Application's Functionality:</h4>", unsafe_allow_html=True)
 description = st.text_area("", key="app_description")
 
-# Using a placeholder for the "Next" button to always display it
 if description or 'next_clicked' in st.session_state:
     if st.button("Next", key="next_to_dataset"):
         st.session_state['next_clicked'] = True
         st.session_state.show_dataset_description = True
 
-
-
 if 'show_dataset_description' in st.session_state and st.session_state.show_dataset_description:
     st.markdown("<h4 style='font-size:20px;'>Detail Your Model's Training Dataset:</h4>", unsafe_allow_html=True)
-    dataset_description = st.text_area("", key="dataset_description", on_change=lambda: st.session_state.__setitem__('show_next_button', True))
-    if dataset_description:
+    dataset_description = st.text_area("", key="dataset_description")
+    
+    if dataset_description or 'recommend_clicked' in st.session_state:
         if st.button("Recommend AI Model", key="recommend_model_button"):
+            st.session_state['recommend_clicked'] = True
             recommended_model = recommend_ai_model_via_gpt(description)
             st.subheader(f"Recommended: {recommended_model}")
             explanation = explain_recommendation(recommended_model)