Update app.py

app.py

@@ -66,6 +66,7 @@ def explain_recommendation(model_name):
 st.title('AI Model Recommender')
 
 description = st.text_area("Describe your application:", "")
+recommendation_type = st.radio("What type of recommendation are you looking for?", ["Recommend Open-Source Model", "Recommend API Service"])
 
 if "rec_model_pressed" not in st.session_state:
     st.session_state.rec_model_pressed = False
@@ -78,17 +79,23 @@ if st.button("Recommend AI Model"):
 
 if st.session_state.rec_model_pressed:
     if description:
-        recommended_model = recommend_ai_model_via_gpt(description)
-        
+        # Modified query based on recommendation type
+        if recommendation_type == "Recommend Open-Source Model":
+            query = f"Given the application described as: '{description}', which open-source AI model would be most suitable?"
+        else:  # Recommend API Service
+            query = f"Given the application described as: '{description}', which AI service API would be best?"
+
+        recommended_model = recommend_ai_model_via_gpt(query)  # Updated function call
+
         # Validate recommended model
         # Commenting out model validation for the example
         # if recommended_model not in KNOWN_MODELS:
         #     st.warning("The recommendation is ambiguous. Please refine your description or consult an expert.")
         # else:
-        st.subheader(f"Recommended AI Model: {recommended_model}")
+        st.subheader(f"Recommended: {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?")
@@ -98,6 +105,8 @@ if st.session_state.rec_model_pressed:
 
         if st.session_state.feedback_submitted:
             st.success("Thank you for your feedback!")
-            st.success("Contact team@autumn8.ai or call (857) 600-0180 to learn how we can fine-tune and host this app for you.")
+
+            st.write("Contact team@autumn8.ai or call (857) 600-0180 to learn how we can fine-tune and host this app for you.")
+            
     else:
-        st.warning("Please provide a description.")
+        st.warning("Please provide a description.")