Update app.py

app.py

@@ -52,41 +52,31 @@ def explain_recommendation(model_name):
     response = openai.ChatCompletion.create(model="gpt-4", messages=messages)
     return response['choices'][0]['message']['content'].strip()
 
-# 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
-
+# Streamlit UI
 st.image("./A8title.png")
 st.title('Discover the best model for your GenAI App')
 
 st.write("")
 
-st.markdown("<h4 style='font-size:20px;'>Outline Functionality of Your Application:</h4>", unsafe_allow_html=True)
+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:
-    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
+# Hardcoded 'Next' button that is always visible
+if st.button("Next", key="next_to_dataset"):
+    st.session_state.show_dataset_description = True
 
-if st.session_state['show_dataset_description']:
+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)
-            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("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!")