Sheshera Mysore commited on
Commit
0cdb40c
·
1 Parent(s): ab5080f

Add a spinner for generating the profile and recs.

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -416,11 +416,13 @@ if uploaded_file is not None:
416
  kp2val_vectors = st.session_state.kp2val_vectors
417
  user_tplan = st.session_state.user_tplan
418
  else: # This happens on the first run.
419
- kp2val_vectors, user_tplan = generate_profile_values(profile_keyphrases=input_user_kps)
 
420
  st.session_state['kp2val_vectors'] = kp2val_vectors
421
  st.session_state['user_tplan'] = user_tplan
422
  else:
423
- kp2val_vectors, user_tplan = generate_profile_values(profile_keyphrases=input_user_kps)
 
424
  st.session_state['kp2val_vectors'] = kp2val_vectors
425
  st.session_state['user_tplan'] = user_tplan
426
  st.session_state.run_user_kps.append(copy.copy(input_user_kps))
@@ -449,11 +451,12 @@ if uploaded_file is not None:
449
  # st.write('Generating recs...')
450
  st.session_state.tuning_i += 1
451
  st.session_state.i_selections.append(copy.deepcopy(profile_selections))
452
- top_papers = second_stage_ranked_docs(first_stage_pids=first_stage_ret_pids,
453
- selected_query_kps=profile_selections,
454
- pid2abstract=pid2abstract_cands,
455
- pid2sent_reps_cand=pid2sent_vectors_cands,
456
- to_rank=30)
 
457
  st.session_state.i_resultps[st.session_state.tuning_i] = copy.deepcopy(top_papers)
458
 
459
  # Read off from the result cache and allow users to save some papers.
 
416
  kp2val_vectors = st.session_state.kp2val_vectors
417
  user_tplan = st.session_state.user_tplan
418
  else: # This happens on the first run.
419
+ with st.spinner(text="Generating profile..."):
420
+ kp2val_vectors, user_tplan = generate_profile_values(profile_keyphrases=input_user_kps)
421
  st.session_state['kp2val_vectors'] = kp2val_vectors
422
  st.session_state['user_tplan'] = user_tplan
423
  else:
424
+ with st.spinner(text="Generating profile..."):
425
+ kp2val_vectors, user_tplan = generate_profile_values(profile_keyphrases=input_user_kps)
426
  st.session_state['kp2val_vectors'] = kp2val_vectors
427
  st.session_state['user_tplan'] = user_tplan
428
  st.session_state.run_user_kps.append(copy.copy(input_user_kps))
 
451
  # st.write('Generating recs...')
452
  st.session_state.tuning_i += 1
453
  st.session_state.i_selections.append(copy.deepcopy(profile_selections))
454
+ with st.spinner(text="Recommending papers..."):
455
+ top_papers = second_stage_ranked_docs(first_stage_pids=first_stage_ret_pids,
456
+ selected_query_kps=profile_selections,
457
+ pid2abstract=pid2abstract_cands,
458
+ pid2sent_reps_cand=pid2sent_vectors_cands,
459
+ to_rank=30)
460
  st.session_state.i_resultps[st.session_state.tuning_i] = copy.deepcopy(top_papers)
461
 
462
  # Read off from the result cache and allow users to save some papers.