Update app.py

app.py

@@ -62,14 +62,21 @@ st.markdown("<h4 style='font-size:20px;'>Outline Your Application's Functionalit
 description = st.text_area("", key="app_description")
 
 if description:
-    if st.button("Recommend AI Model", key="recommend_model_button"):
-        recommended_model = recommend_ai_model_via_gpt(description)
-        st.subheader(f"Recommended: {recommended_model}")
-        explanation = explain_recommendation(recommended_model)
-        st.write("Explanation:", explanation)
-        rating = st.slider("Rate the explanation from 1 (worst) to 5 (best):", 1, 5)
-        feedback = st.text_input("Any additional feedback?")
-        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!")
+    if st.button("Next", key="next_to_dataset"):
+        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")
+    if dataset_description:
+        if st.button("Recommend AI Model", key="recommend_model_button"):
+            recommended_model = recommend_ai_model_via_gpt(description, dataset_description)
+            st.subheader(f"Recommended: {recommended_model}")
+            explanation = explain_recommendation(recommended_model)
+            st.write("Explanation:", explanation)
+            rating = st.slider("Rate the explanation from 1 (worst) to 5 (best):", 1, 5)
+            feedback = st.text_input("Any additional feedback?")
+            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!")