Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import os
+import pandas as pd
 from together import Together
 
 
@@ -66,7 +67,8 @@ with st.sidebar:
         st.experimental_rerun()
 
 
-st.title("Meta Llama3 🦙")
+st.set_page_config(layout="wide")
+st.title("Duel Agent Simulation 🦙🦙")
 
 
 # Initialize chat history
@@ -88,6 +90,11 @@ judge = call_llama
 
 # React to user input
 iter = 0
+list_of_iters = []
+list_of_questions = []
+list_of_answers = []
+list_of_judge_comments = []
+list_of_passes = []
 if submit_button:
 
     # Initiatization
@@ -142,9 +149,27 @@ if submit_button:
         )
         
         # Stopping rule
-        if '9' in judge_comments:
+        passed_or_not = 1 if '8' in judge_comments else 0
+        if '8' in judge_comments:
             break
 
+        # Collect all responses
+        list_of_iters.append(iter)
+        list_of_questions.append(question)
+        list_of_answers.append(answer)
+        list_of_judge_comments.append(judge_comments)
+        list_of_passes.append(passed_or_not)
+        results_tab = pd.DataFrame({
+            "Iter.": list_of_iters,
+            "Questions": list_of_questions,
+            "Answers": list_of_answers,
+            "Judge Comments": list_of_judge_comments,
+            "Passed": list_of_passes
+        })
+
+        with st.expander("See explanation"):
+            st.table(results_tab)
+
         # Checkpoint
         iter += 1