adrienbrdne commited on
Commit
f222139
·
verified ·
1 Parent(s): dd91c4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -128,6 +128,8 @@ if st.button("Generate Key Issues", key="generate_issues_button"):
128
  if 'key_issues' in result and isinstance(result['key_issues'], list) and result['key_issues']:
129
  st.success("Key Issues received successfully!")
130
  temp_df = pd.DataFrame(result['key_issues'])
 
 
131
  # Ensure 'title' and 'description' columns exist
132
  if 'title' in temp_df.columns and 'description' in temp_df.columns:
133
  # Select and reorder necessary columns, reset index for selection
 
128
  if 'key_issues' in result and isinstance(result['key_issues'], list) and result['key_issues']:
129
  st.success("Key Issues received successfully!")
130
  temp_df = pd.DataFrame(result['key_issues'])
131
+ if 'id' in temp_df.columns:
132
+ temp_df.drop(columns=['id'], inplace=True)
133
  # Ensure 'title' and 'description' columns exist
134
  if 'title' in temp_df.columns and 'description' in temp_df.columns:
135
  # Select and reorder necessary columns, reset index for selection