added modfi

app.py

@@ -6,6 +6,7 @@ from utils import (
 )
 
 import streamlit as st
+import json
 import textwrap as tw
 import pandas as pd
 from streamlit_extras.stylable_container import stylable_container
@@ -63,7 +64,6 @@ if uploaded_file is not None:
 
             df.insert(0, 'Select', select_values)
             df.at[st.session_state.current_row_index, 'Select'] = True
-            st.session_state.comments_add = dict((idx, t) for idx, t in enumerate(df["Comment"]))
             st.session_state['df'] = df
         
         df_col, = st.columns(1)
@@ -92,6 +92,12 @@ if uploaded_file is not None:
     question = row_data['question']
     pred_relevant_choice = row_data['pred_relevant_choice']
     correct_response = row_data['correct_response']
+    relative_relevancy = json.loads(row_data['relative_relevancy'])
+    reason = relative_relevancy['reason']
+    most_relevant = relative_relevancy['most_relevant']
+    options=['first', 'second', 'both']
+    default_index = options.index(most_relevant)
+    default_comment = row_data['Comment']
     first_css = container_regular_css
     second_css = container_regular_css
     first_color = ':black'
@@ -102,7 +108,7 @@ if uploaded_file is not None:
     with st.container(border=True):
         # st.write(f"Question: {question}")
         st.markdown(f'<p class="question-font">[Q]: {question}</p>', unsafe_allow_html=True)
-        select_by = st.selectbox("Mark By: ", ["pred_relevant_choice", "correct_response"])
+        select_by = st.selectbox("Mark By: ", ["pred_relevant_choice", "correct_response"], key='mark_by_dropdown')
         col1_res, col2_res, _ = st.columns([0.2, 0.2, 0.6])
         if select_by == 'pred_relevant_choice':
             col1_res.write(f":blue[pred_relevant_choice: {pred_relevant_choice}]")
@@ -162,7 +168,12 @@ if uploaded_file is not None:
                         predicted_sql_text,
                         language="sql"
                     )
-        comment = st.text_area("Comment", value=row_data['Comment'])
+        col1_comment, col2_comment = st.columns([1, 1])
+        with col1_comment:
+            select_most_relevant = st.selectbox("most_relevant", options=options, index=default_index, key='most_relevant_drop_down')
+            reason_box = st.text_area("reason", value=reason, key='reason_text')
+        with col2_comment:
+            comment = st.text_area("Comment", value=default_comment, key='comment_text')
     
     with st.container():
         bcol1, bcol2, _, _, _, _, _, _, _, _ = st.columns(10)
@@ -172,12 +183,14 @@ if uploaded_file is not None:
             download = st.button("Download")
 
     if save:
-        st.session_state.comments_add[row_index] = comment
+        updated_relative_relevancy = json.dumps({'most_relevant': select_most_relevant, "reason": reason_box})
         st.session_state.df.at[row_index, 'Comment'] = comment
+        st.session_state.df.at[row_index, 'relative_relevancy'] = updated_relative_relevancy
+        st.session_state.df.at[row_index, 'correct_response'] = select_most_relevant
         df = st.session_state.df.drop('Select', axis=1)
         df.to_csv('modified.csv', index=False)
-        # st.rerun()
-    
+        st.rerun()
+
     if download:
         with open('modified.csv') as f:
             st.download_button('Download modified CSV', f, file_name='modified.csv')