Update app.py

app.py

@@ -3,10 +3,10 @@ import pickle
 import pandas as pd
 import torch
 import numpy as np
-from streamlit_card import card
 
 cosine_scores = pickle.load(open('cosine_scores.pkl','rb'))
-coursedf = pd.read_pickle('course_df.pkl') 
+coursedf = pd.read_pickle('course_df.pkl')          # course_df uses titles to generate course recommendations
+course_df_new = pd.read_pickle('course_df_new.pkl') #course_df_new makes recommendations using the entire description
 
 course_title_list = [i + ": " + j for i, j in zip(coursedf['ref'].to_list(), coursedf['title'].to_list())]
 
@@ -29,19 +29,26 @@ def recommend(index):
 st.set_page_config(page_title='DiscoverCourses', page_icon=':book:')
 st.title('DiscoverCourses')
 
-selected_course = st.selectbox('Please select a course',course_title_list)
+selected_course = st.selectbox('Pick a course',course_title_list)
 
-if st.button('I want more like this!'):
+if st.button('Recommend by title'):
     output=recommend(np.where((coursedf['ref']+": "+coursedf['title']) == selected_course)[0][0])
     for result in output:
         index=np.where(coursedf['title'] == result)[0][0]
         course_id=coursedf.iloc[index,0]
         st.subheader(course_id+": "+result)
-        link = "[ExploreCourses](https://explorecourses.stanford.edu/search?q="+course_id+")"
+        st.text(course_df_new.iloc[index,3]) #Using the new coursedf because it has proper descriptions for each course
+        link = "[ExploreCourses](https://explorecourses.stanford.edu/search?q="+course_id+" "+result+")"
         st.markdown(link, unsafe_allow_html=True)
         st.divider()
-        # card(
-        #     title=course_id,
-        #     text=result,
-        #     url="https://explorecourses.stanford.edu/search?q="+course_id+": "+result
-        # )
+        
+if st.button('Recommend by description'):
+    index_new=np.where((coursedf['ref']+": "+coursedf['title']) == selected_course)[0][0]
+    rec_list=course_df_new.iloc[index_new,2]
+    for result in rec_list:
+        course_id=coursedf.iloc[index,0]
+        st.subheader(course_id+": "+result)
+        st.text(course_df_new.iloc[index,3]) #Using the new coursedf because it has proper descriptions for each course
+        link = "[ExploreCourses](https://explorecourses.stanford.edu/search?q="+course_id+" "+result+")"
+        st.markdown(link, unsafe_allow_html=True)
+        st.divider()