Update app.py

app.py

@@ -498,34 +498,24 @@ def show_tabular_data(df, co):
         score = calculate_relevance_score(content, row['query'], co)
         return score
 
-    # Function to update the relevancy score
-    def update_relevancy(row_index):
-        score = calculate_single_relevancy(df.iloc[row_index])
-        df.at[row_index, 'relevancy_score'] = score
-        df_display.at[row_index, 'relevancy_score'] = f"{score:.4f}"
-
-    # Create a column for the calculate button
-    df_display['calculate'] = ['Calculate' if pd.isna(score) or score == 0 else f"{score:.4f}" 
-                               for score in df['relevancy_score']]
-
     # Display the dataframe
-    edited_df = st.data_editor(
-        df_display,
-        column_config={
-            "calculate": st.column_config.ButtonColumn(
-                "Relevancy Score",
-                help="Click to calculate relevancy score"
-            )
-        },
-        disabled=["page", "query", "clicks", "impressions", "ctr", "position"],
-        hide_index=True,
-    )
-
-    # Check if any button was clicked
-    for index, row in edited_df.iterrows():
-        if row['calculate'] == True:  # Button was clicked
-            update_relevancy(index)
-            st.experimental_rerun()
+    st.dataframe(df_display)
+
+    # Create buttons for each row
+    for index, row in df_display.iterrows():
+        col1, col2 = st.columns([3, 1])
+        with col1:
+            st.write(f"Query: {row['query']} | Page: {row['page']}")
+        with col2:
+            if pd.isna(row['relevancy_score']) or row['relevancy_score'] == 0:
+                if st.button(f"Calculate Score", key=f"calc_{index}"):
+                    score = calculate_single_relevancy(row)
+                    df.at[index, 'relevancy_score'] = score
+                    df_display.at[index, 'relevancy_score'] = score
+                    st.success(f"Relevancy Score: {score:.4f}")
+                    st.experimental_rerun()
+            else:
+                st.write(f"Relevancy Score: {row['relevancy_score']:.4f}")
 
     # Add an expander for each row to show competitor analysis
     for index, row in df.iterrows():
@@ -538,6 +528,8 @@ def show_tabular_data(df, co):
                     our_rank = results_df.index[results_df['url'] == row['page']].tolist()[0] + 1
                     st.write(f"Our page ranks {our_rank} out of {len(results_df)} in terms of relevancy score.")
 
+    return df  # Return the updated dataframe
+
 def show_date_range_selector():
   #  logging.info("Showing date range selector")
     return st.selectbox(
@@ -680,7 +672,7 @@ def main():
             if st.session_state.report_data is not None and not st.session_state.report_data.empty:
                 st.write("Data fetched successfully.")
                 
-                show_tabular_data(st.session_state.report_data, co)
+                st.session_state.report_data = show_tabular_data(st.session_state.report_data, co)
                 
                 download_csv_link(st.session_state.report_data)
             elif st.session_state.report_data is not None: