Update app.py

app.py

@@ -68,12 +68,12 @@ 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")
 
-if st.button("Next", key="next_to_dataset"):
-    st.session_state.show_dataset_description = True
+if description:
+    if st.button("Next", key="next_to_dataset"):
+        st.session_state.show_dataset_description = True
 
-if 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")
+if 'show_dataset_description' in st.session_state and st.session_state.show_dataset_description:
+    dataset_description = st.text_area("Detail your model's training dataset:")
     if dataset_description:
         if st.button("Recommend AI Model", key="recommend_model_button"):
             recommended_model = recommend_ai_model_via_gpt(description, dataset_description)
@@ -85,4 +85,7 @@ if st.session_state.show_dataset_description:
             if st.button("Submit Feedback", key="submit_feedback_key"):
                 st.session_state.feedback_submitted = True
                 if st.session_state.feedback_submitted:
-                    st.success("Thank you for your feedback!")
+                    st.success("Thank you for your feedback!")
+
+else:
+    st.warning("Please provide a description of your application.")