kh-CHEUNG commited on
Commit
b75316c
·
verified ·
1 Parent(s): a489bfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -138,15 +138,13 @@ def lookup_profile(assessment_id_input):
138
  assessment_id = assessment_id_input.strip()
139
  if not assessment_id:
140
  #state.update("Please enter an Assessment ID", color="red")
141
- return "No matched profile found!"
142
 
143
- results = df[df["Assessment_ID"].str.contains(assessment_id, case=False)]
144
- print(results)
145
- if results.empty:
146
  #state.update("No profiles found for this ID", color="red")
147
  return "No matched profile found!"
148
- else:
149
- return results.to_markdown(index=False)
150
  #profile_preview.update(value=results.to_markdown(index=False))
151
  #state.update("Profile(s) found!", color="green")
152
 
 
138
  assessment_id = assessment_id_input.strip()
139
  if not assessment_id:
140
  #state.update("Please enter an Assessment ID", color="red")
141
+ return "No Assessment Object/Session ID provided!"
142
 
143
+ if assessment_id in df.index:
144
+ return df.[assessment_id].to_markdown(index=False)
145
+ else:
146
  #state.update("No profiles found for this ID", color="red")
147
  return "No matched profile found!"
 
 
148
  #profile_preview.update(value=results.to_markdown(index=False))
149
  #state.update("Profile(s) found!", color="green")
150