crismunoz commited on
Commit
1de592f
·
1 Parent(s): 15d3c80
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -48,16 +48,18 @@ def program():
48
  if selected_task:
49
  st.header(selected_task)
50
  results = data[selected_task]
 
51
 
52
- recommendations = pd.DataFrame(results['recommendations'])
53
- recommendations['date'] = pd.to_datetime(recommendations.apply(lambda x:x['metadata']['date'], axis=1))
54
- recommendations = recommendations.sort_values(by='date')
55
-
56
-
57
- rec = {str(r['id']): r for r in recommendations.to_dict('records')}
58
  for group in results['groups']:
59
  ids = [i.strip() for i in group['ids'].split(",")]
60
- rec2html = ''.join([f"""<tr><td style="border: 1px solid #ddd; padding: 8px;">{i+1}</td><td style="border: 1px solid #ddd; padding: 8px;"><a href="{rec[idx]['metadata']['id']}" target="_blank">{rec[idx]['title']}</a></td><td style="border: 1px solid #ddd; padding: 8px;">{rec[idx]['metadata']['date']}</td></tr>""" for i, idx in enumerate(ids)])
 
 
 
 
 
 
 
61
  title = group['title'].split(':', 1)[1].strip()
62
  st.markdown(f"""
63
  <div style="border: 1px solid #ccc; padding: 10px; margin: 10px 0; border-radius: 5px; width: 100%;">
 
48
  if selected_task:
49
  st.header(selected_task)
50
  results = data[selected_task]
51
+ rec = {str(r['id']): r for r in results['recommendations']}
52
 
 
 
 
 
 
 
53
  for group in results['groups']:
54
  ids = [i.strip() for i in group['ids'].split(",")]
55
+
56
+ selected_rec = [rec[i] for i in ids]
57
+
58
+ selected_rec = pd.DataFrame(selected_rec)
59
+ selected_rec['date'] = pd.to_datetime(selected_rec.apply(lambda x:x['metadata']['date'], axis=1))
60
+ selected_rec = selected_rec.sort_values(by='date').to_dict('records')
61
+
62
+ rec2html = ''.join([f"""<tr><td style="border: 1px solid #ddd; padding: 8px;">{i+1}</td><td style="border: 1px solid #ddd; padding: 8px;"><a href="{rec['metadata']['id']}" target="_blank">{rec['title']}</a></td><td style="border: 1px solid #ddd; padding: 8px;">{rec['metadata']['date']}</td></tr>""" for i, rec in enumerate(selected_rec)])
63
  title = group['title'].split(':', 1)[1].strip()
64
  st.markdown(f"""
65
  <div style="border: 1px solid #ccc; padding: 10px; margin: 10px 0; border-radius: 5px; width: 100%;">