Update app.py

app.py

@@ -58,7 +58,7 @@ counternarratives = load_counternarratives()
 # Generate Random Myth or Fact and trigger persona response
 def get_random_myth_or_fact():
     if counternarratives.empty:
-        return "No myths or facts available.", "Fact-Checker", "", "", ""
+        return "No myths or facts available.", "Tree", "Crow", "Error: No Data Found.", "Error: No Data Found."
 
     # 🔄 Randomly select a row from the dataframe
     row = counternarratives.sample(1).iloc[0]
@@ -76,8 +76,8 @@ def get_random_myth_or_fact():
     else:
         fact_check_response = f"✅ **FACT**\n\nIndeed, {row['fact']}"
 
-    # Return the myth/fact, update the personas, and fill the responses
-    return myth_or_fact, persona, "Fact-Checker", persona_response, fact_check_response
+    # Return the myth/fact, set personas, and fill the responses
+    return myth_or_fact, persona, "Fact-Checker 🧠", persona_response, fact_check_response
 
 # Dynamically load persona names from instructions folder
 personas = [os.path.splitext(f)[0].capitalize() for f in os.listdir("instructions") if f.endswith(".txt")]
@@ -87,7 +87,7 @@ with gr.Blocks() as demo:
     # Header with Image
     with gr.Row():
         with gr.Column(scale=0.15):
-            gr.Image(value="data/MoreThanHumanVoices.png", label="More Than Human Voices", show_label=False)
+            gr.Image(value="data/WildVoices.png", label="Wild Voices", show_label=False)
         with gr.Column():
             gr.Markdown("""
             # 🌲 **Wild Voices**  
@@ -98,7 +98,7 @@ with gr.Blocks() as demo:
     # Persona Selection
     with gr.Row():
         persona1 = gr.Dropdown(personas, label="Choose First Persona", value="Tree")
-        persona2 = gr.Dropdown(personas, label="Choose Second Persona", value="Crow")
+        persona2 = gr.Dropdown(personas + ["Fact-Checker 🧠"], label="Choose Second Persona", value="Crow")
 
     # Question and buttons
     with gr.Row():
@@ -110,9 +110,10 @@ with gr.Blocks() as demo:
     with gr.Row():
         ask_button = gr.Button("🌎 Submit Question")
 
-    # Response display
-    output1 = gr.Textbox(label="Tree Responds")
-    output2 = gr.Textbox(label="Crow Responds")
+    # Response display (side by side)
+    with gr.Row():
+        output1 = gr.Textbox(label=f"{persona1.value} Responds")
+        output2 = gr.Textbox(label=f"{persona2.value} Responds")
 
     # Update labels when dropdown changes
     def update_labels(p1, p2):
@@ -132,9 +133,16 @@ with gr.Blocks() as demo:
     )
 
     # Ask button for normal questions
+    def ask_with_context(p1, p2, q):
+        res1 = call_cohere_api(load_instruction(p1), q)
+        if p2 == "Fact-Checker 🧠":
+            res2 = "I am here to verify facts and dispel myths."
+        else:
+            res2 = call_cohere_api(load_instruction(p2), q)
+        return res1, res2
+
     ask_button.click(
-        fn=lambda p1, p2, q: (call_cohere_api(load_instruction(p1), q), 
-                              call_cohere_api(load_instruction(p2), q)), 
+        fn=ask_with_context, 
         inputs=[persona1, persona2, user_input], 
         outputs=[output1, output2]
     )