Update app.py

app.py

@@ -43,6 +43,7 @@ KNOWN_MODELS = [
 ]
 
 def recommend_ai_model_via_gpt(description):
+    combined_description = f"{description} Dataset: {dataset_description}"
     messages = [
         {"role": "user", "content": description}
     ]
@@ -81,9 +82,16 @@ st.image("./A8title2.png")
 st.title('Find the best AI stack for your app')
 
 description = st.text_area("Describe your application:")
-dataset_description = st.text_area("Describe the dataset you want to use for fine-tuning your model:")
 
-recommendation_type = st.radio("What type of recommendation are you looking for?", ["Recommend Open-Source Model", "Recommend API Service"])
+if description:
+    dataset_description = st.text_area("Describe the dataset you want to use for fine-tuning your model:")
+
+    if dataset_description:
+        if st.button("Recommend AI Model"):
+            recommended_model = recommend_ai_model_via_gpt(description, dataset_description)
+            st.subheader(f"Recommended: {recommended_model}")
+
+            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