Update app.py

app.py

@@ -161,6 +161,7 @@ def qa_summarise(selected_queries, qa_llm_model, text_field, data_df):
     doc_df = pd.DataFrame()
     doc_df["number"] = [i+1 for i in range(len(texts))]
     doc_df["text"] = texts
+    doc_df["IDs"] = qa_input_df["tweet_id"].to_list()
     
     return summary, doc_df
 
@@ -274,10 +275,10 @@ with gr.Blocks(fill_width=True) as demo:
             selected_queries = gr.CheckboxGroup(label="Select at least one query using the checkboxes", interactive=True)
             queries_state = gr.State()
             qa_tab.select(init_queries, inputs=queries_state, outputs=[selected_queries, queries_state])
-    
             query_inp = gr.Textbox(label="Add custom queries like the one above, one at a time")
             QA_addqry_button = gr.Button("Add to queries", interactive=False)
-            QA_run_button = gr.Button("Start QA", interactive=False)
+            
+        QA_run_button = gr.Button("Start QA", interactive=False)
         hsummary = gr.Textbox(label="Summary")
         
         qa_df = gr.DataFrame()
@@ -403,7 +404,10 @@ with gr.Blocks(fill_width=True) as demo:
 
     QA_run_button.click(qa_summarise, 
                     inputs=[selected_queries, qa_llm_model, T_text_field, T_data_ss_state],
-                    outputs=[hsummary, qa_df])      
+                    outputs=[hsummary, qa_df])   
+
+    qa_df.select(T_on_select, None, T_tweetID)
+    T_tweetID.change(fn=None, inputs=T_tweetID, outputs=None, js=m_js)
     
     
     # Event listener for single text classification