Update app.py

app.py

@@ -489,8 +489,43 @@ def show_model_type_selector():
 def show_tabular_data(df, co):
     st.write("Data Table with Relevancy Scores")
     
-    # Display the dataframe as a table
-    st.dataframe(df)
+    # Create a copy of the dataframe to modify
+    df_display = df.copy()
+    
+    # Function to calculate relevancy score for a single row
+    def calculate_single_relevancy(row):
+        content = fetch_content(row['page'])
+        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()
 
     # Add an expander for each row to show competitor analysis
     for index, row in df.iterrows():
@@ -643,11 +678,7 @@ def main():
                     st.session_state.report_data = fetch_gsc_data(webproperty, search_type, start_date, end_date, selected_dimensions)
 
             if st.session_state.report_data is not None and not st.session_state.report_data.empty:
-                st.write("Data fetched successfully. Click the button below to calculate relevancy scores.")
-                
-                if st.button("Calculate Relevancy Scores"):
-                  #  logger.info("Calculating relevancy scores for all rows")
-                    st.session_state.report_data = calculate_relevancy_scores(st.session_state.report_data, model_type)
+                st.write("Data fetched successfully.")
                 
                 show_tabular_data(st.session_state.report_data, co)