Update app.py

app.py

@@ -86,3 +86,37 @@ if st.button("Recommend AI Model"):
             st.success("Thank you for your feedback!")
     else:
         st.warning("Please provide a description.")
+
+# Streamlit UI
+st.title('AI Model Recommender')
+
+description = st.text_area("Describe your application:", "")
+
+if "rec_model_pressed" not in st.session_state:
+    st.session_state.rec_model_pressed = False
+
+if "feedback_submitted" not in st.session_state:
+    st.session_state.feedback_submitted = False
+
+if st.button("Recommend AI Model"):
+    st.session_state.rec_model_pressed = True
+
+if st.session_state.rec_model_pressed:
+    if description:
+        recommended_model = recommend_ai_model_via_gpt(description)
+        
+        st.subheader(f"Recommended AI Model: {recommended_model}")
+        explanation = explain_recommendation(recommended_model)
+        st.write("Reason:", explanation)
+
+        # Collecting rating and feedback through Streamlit
+        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"):
+            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.")