pendar02 commited on
Commit
44d47af
·
verified ·
1 Parent(s): dde1577

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -5
app.py CHANGED
@@ -174,6 +174,11 @@ def generate_focused_summary(question, abstracts, model, tokenizer):
174
  def main():
175
  st.title("🔬 Biomedical Papers Analysis")
176
 
 
 
 
 
 
177
  # File upload section
178
  uploaded_file = st.file_uploader(
179
  "Upload Excel file containing papers",
@@ -210,9 +215,12 @@ def main():
210
 
211
  progress_text = st.empty()
212
  progress_bar = st.progress(0)
213
- summary_display = st.container()
214
 
 
 
215
  summaries = []
 
 
216
  for i, (_, row) in enumerate(df.iterrows()):
217
  progress_text.text(f"Processing paper {i+1} of {len(df)}")
218
  progress_bar.progress((i + 1) / len(df))
@@ -220,10 +228,19 @@ def main():
220
  summary = generate_summary(row['Abstract'], model, tokenizer)
221
  summaries.append(summary)
222
 
223
- with summary_display:
224
- st.write(f"**Paper {i+1}:** {row['Article Title']}")
225
- st.write(summary)
226
- st.divider()
 
 
 
 
 
 
 
 
 
227
 
228
  st.session_state.summaries = summaries
229
 
 
174
  def main():
175
  st.title("🔬 Biomedical Papers Analysis")
176
 
177
+ # Initialize text processor if not already done
178
+ if st.session_state.text_processor is None:
179
+ with st.spinner("Loading NLP models..."):
180
+ st.session_state.text_processor = TextProcessor()
181
+
182
  # File upload section
183
  uploaded_file = st.file_uploader(
184
  "Upload Excel file containing papers",
 
215
 
216
  progress_text = st.empty()
217
  progress_bar = st.progress(0)
 
218
 
219
+ # Create a table for live updates
220
+ summary_table = st.table([])
221
  summaries = []
222
+ table_data = []
223
+
224
  for i, (_, row) in enumerate(df.iterrows()):
225
  progress_text.text(f"Processing paper {i+1} of {len(df)}")
226
  progress_bar.progress((i + 1) / len(df))
 
228
  summary = generate_summary(row['Abstract'], model, tokenizer)
229
  summaries.append(summary)
230
 
231
+ # Update table data
232
+ table_data.append({
233
+ "Title": row['Article Title'],
234
+ "Summary": summary
235
+ })
236
+ summary_table.dataframe(
237
+ pd.DataFrame(table_data),
238
+ column_config={
239
+ "Title": st.column_config.TextColumn("Paper Title", width="medium"),
240
+ "Summary": st.column_config.TextColumn("Generated Summary", width="large")
241
+ },
242
+ hide_index=True
243
+ )
244
 
245
  st.session_state.summaries = summaries
246