Update app.py

app.py

@@ -54,12 +54,18 @@ def explain_recommendation(model_name):
     return response['choices'][0]['message']['content'].strip()
 
 # Streamlit UI
+if 'show_dataset_description' not in st.session_state:
+    st.session_state.show_dataset_description = False
+
 st.image("./A8title2.png")
 st.title('Find the best AI stack for your app')
 
 description = st.text_area("Describe your application:")
 
-if description:
+if st.button("Next", key="next_to_dataset"):
+    st.session_state.show_dataset_description = True
+
+if st.session_state.show_dataset_description:
     dataset_description = st.text_area("Describe your dataset:")
     if dataset_description:
         if st.button("Recommend AI Model", key="recommend_model_button"):
@@ -72,7 +78,4 @@ if 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!")
-
-else:
-    st.warning("Please provide a description and dataset details.")
+                    st.success("Thank you for your feedback!")