minor changes, and debate initialization with CoRE

app.py

@@ -225,14 +225,14 @@ def reasoner(query: str, documents: list[str], selected_dreamer: str, selected_r
             else:
                 prompt_template = Template(templates["with_evidence"])
                 prompt = prompt_template.substitute(claim=query, corpus_text=documents)
-        elif selected_dreamer == 'CoRE' or selected_dreamer == 'CoRE-Contrastive':
+        elif (selected_dreamer == 'CoRE' or selected_dreamer == 'CoRE-Contrastive') and selected_reasoner != "Multi Agent Debate":
             conditions = [""]
             prompt_template = Template(templates["generate_condition"])
             prompt = prompt_template.substitute(claim=query)
             prompt = get_examples() + prompt + " answer: " 
             print(prompt)
 
-            message += f"Using {selected_dreamer} to analyze and verify the claim in detail..."
+            message += f"Using {selected_dreamer} to decompose and elaborate the claim..."
             display_to_chat(placeholder, message)
             placeholder = st.empty()
 
@@ -243,7 +243,7 @@ def reasoner(query: str, documents: list[str], selected_dreamer: str, selected_r
             
             condition = conditions[0] if selected_dreamer == 'CoRE' else conditions[1]
 
-            message = "To reason about the claim, CoRE is using the condition: " + condition + "\n\n\n\n"
+            message = "To reason about the claim, CoRE is considering the potential condition: " + condition + "\n\n\n\n"
 
             if not documents or len(documents) == 0:
                 prompt_template = Template(templates["with_condition"])
@@ -277,6 +277,17 @@ def reasoner(query: str, documents: list[str], selected_dreamer: str, selected_r
                 prompt = prompt_template.substitute(claim=query, corpus_text=documents)
 
         if selected_reasoner == "Multi Agent Debate":
+            initial_agent_beliefs=None
+            if selected_dreamer == 'CoRE' or selected_dreamer == 'CoRE-Contrastive':
+                conditions = [""]
+                prompt_template = Template(templates["generate_condition"])
+                prompt = prompt_template.substitute(claim=query)
+                prompt = get_examples() + prompt + " answer: " 
+                llm_response = llm_client.run_inference(prompt)
+                initial_agent_beliefs = llm_response.split('\n\n')
+                print(initial_agent_beliefs)
+
+            
             multi_agent_debate = MultiAgentDebate(client=llm_client.client)
             initial_evidence = "" if (not documents or len(documents)==0) else documents[0]
             def mad_printer(msg, explanation):
@@ -289,6 +300,7 @@ def reasoner(query: str, documents: list[str], selected_dreamer: str, selected_r
                 
             decision = multi_agent_debate(claim=query,
                                           doc=initial_evidence,
+                                          initial_agent_responses=initial_agent_beliefs,
                                           writer=mad_printer)
             reasoning = "of the debate and discussion."
         else:

multi_agent_debate.py

@@ -433,6 +433,7 @@ class MultiAgentDebate:
         print(x)
 
     def __call__(self, doc, claim, initialization=True, model_name='gpt-4o-mini',
+                 initial_agent_responses=None,
                   writer=safe_print):
         # number of simultaneous debates for evaluation
         num_debates = 1
@@ -455,7 +456,19 @@ class MultiAgentDebate:
             ## more utterances
 
             if initialization:
-                agents_responses = ["The claim is not refuted by evidence.", "The claim is refuted by evidence.", "The claim is not refuted by evidence.", "The claim is refuted by evidence."]
+                if initial_agent_responses is None:
+                    agents_responses = ["The claim is not refuted by evidence.", "The claim is refuted by evidence.", "The claim is not refuted by evidence.", "The claim is refuted by evidence."]
+                else:
+                    agents_responses = []
+                    for n in range(4):
+                        if n < len(initial_agent_responses):
+                            agents_responses.append(initial_agent_responses[n])
+                        else:
+                            if n % 2 == 0:
+                                agents_responses.append("The claim is not refuted by evidence.")
+                            else:
+                                agents_responses.append("The claim is refuted by evidence.")
+
             else:
                 agents_responses = ["","","",""]
 
@@ -477,7 +490,8 @@ class MultiAgentDebate:
             if initialization:
                 print("ROUND %s: (This is the initialization round where agents are assigned initial stance as their beliefs.)\n"%str(round_counter+1))
                 for n in range(len(agents_responses)):
-                    writer("Agent %s: "%str(n+1) + agents_responses[n] + "\n", "This is my initial belief.")
+                    writer("Agent %s: "%str(n+1) + agents_responses[n] + "\n",
+                           "This is my initial belief.")
                 print("----------------------------------------------------")
                 round_counter += 1
             print("ROUND %s:\n"%str(round_counter+1))