Update app.py

app.py

@@ -114,14 +114,24 @@ def qa_process(selected_queries, qa_llm_model, aggregator,
         genra.retrieval(batch_number, queries_df, topk, summarize_batch) 
     
     gr.Info("Processed all batches!")
-    # result ------ genra.answers_store
-
+    
+    gr.Info("Getting summary...")
     summary = genra.summarize_history(queries_df)
     
-    analysis_results = [f"Word Count: {len(text.split())}" for text in selected_queries]
+    gr.Info("Preparing results...")
+    results = genra.answers_store
+    final_answers, q_a = [], []
+    for q, g_answers in results.items():
+        for answer in g_answers:
+            final_answers.append({'question':q, "tweets":answer['tweets'], "batch":answer['batch_number'], "summary":answer['summary'] })
+            for t in answer['tweets']:
+                q_a.append((q,t))
+    answers_df = pd.DataFrame.from_dict(final_answers)
+    q_a = list(set(q_a))
+    q_a_df = pd.DataFrame(q_a, columns =['question', 'tweet'])
+    q_a_df = q_a_df.sort_values(by=["question"], ascending=False)
     
-    result_df = pd.DataFrame({"Selected Text": selected_queries, "Analysis": analysis_results})
-    return result_df, summary
+    return q_a_df, answers_df, summary
 
             
 with gr.Blocks() as demo:
@@ -234,13 +244,13 @@ with gr.Blocks() as demo:
         qa_button = gr.Button("Start QA")
         hsummary = gr.Textbox(label="Historical Summary")
         
-        analysis_output = gr.DataFrame(headers=["Selected Text", "Analysis"])
-
+        qa_df = gr.DataFrame()
+        answers_df = gr.DataFrame()
 
         addqry_button.click(add_query, inputs=[query_inp, queries_state], outputs=[selected_queries, queries_state])
         qa_button.click(qa_process, 
                         inputs=[selected_queries, qa_llm_model, aggregator, batch_size, topk, text_field, data], 
-                        outputs=[analysis_output, hsummary])
+                        outputs=[qa_df, answers_df, hsummary])
 
 
 demo.launch()