Update app.py

app.py

@@ -52,7 +52,13 @@ def explain_recommendation(model_name):
     response = openai.ChatCompletion.create(model="gpt-4", messages=messages)
     return response['choices'][0]['message']['content'].strip()
 
-# Streamlit UI
+# Streamlit UI Initialization
+if 'next_clicked' not in st.session_state:
+    st.session_state['next_clicked'] = False
+
+if 'show_dataset_description' not in st.session_state:
+    st.session_state['show_dataset_description'] = False
+
 st.image("./A8title.png")
 st.title('Discover the best model for your GenAI App')
 
@@ -61,19 +67,19 @@ 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 description or 'next_clicked' in st.session_state:
-    if st.button("Next step: Dataset", key="next_to_dataset"):
-        st.session_state['next_clicked'] = True
+if description:
+    st.session_state['next_clicked'] = True
+
+if st.session_state['next_clicked']:
+    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.write("")
-    st.markdown("<h4 style='font-size:20px;'>Detail Your Training Dataset:</h4>", unsafe_allow_html=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 dataset_description or 'recommend_clicked' in st.session_state:
+    if dataset_description:
         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)