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.", "Tree", "Fact-Checker 🧠", "Error: No Data Found.", "Error: No Data Found."
+        return "No myths or facts available.", "Fact-Checker", "", "", ""
 
     # 🔄 Randomly select a row from the dataframe
     row = counternarratives.sample(1).iloc[0]
@@ -76,55 +76,33 @@ def get_random_myth_or_fact():
     else:
         fact_check_response = f"✅ **FACT**\n\nIndeed, {row['fact']}"
 
-    # Return the myth/fact, set personas, and fill the responses
-    return myth_or_fact, persona, "Fact-Checker 🧠", persona_response, fact_check_response
+    # Return the myth/fact, update the 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")]
 
 # Gradio Interface
 with gr.Blocks() as demo:
-    # Header with Image
     with gr.Row():
-        with gr.Column(scale=0.15):
-            gr.Image(value="data/WildVoices.png", label="Wild Voices", show_label=False)
-        with gr.Column():
-            gr.Markdown("""
-            # 🌲 **Wild Voices**  
-            *Listening to the More-than-Human World*  
-            Ask questions to rivers, trees, owls, and more. Generate myths and facts, and let nature's wisdom respond.
-            """)
-
-    # Persona Selection
+        gr.Markdown("# 🌲 **Wild Voices** — *Listening to the More-than-Human World*")
+        gr.Markdown("Ask questions to rivers, trees, owls, and more. Generate myths and facts, and let nature's wisdom respond.")
+
     with gr.Row():
         persona1 = gr.Dropdown(personas, label="Choose First Persona", value="Tree")
-        persona2 = gr.Dropdown(personas + ["Fact-Checker 🧠"], label="Choose Second Persona", value="Crow")
+        persona2 = gr.Dropdown(personas, label="Choose Second Persona", value="Crow")
 
-    # Question and buttons
     with gr.Row():
         user_input = gr.Textbox(label="🌱 Your Question", placeholder="e.g., What do you think of humans?", lines=2)
         random_button = gr.Button("🎲 Generate Random Question")
         myth_fact_button = gr.Button("🔍 Generate Random Myth/Fact")
 
-    # Submit button
     with gr.Row():
         ask_button = gr.Button("🌎 Submit Question")
 
-    # Response display (side by side)
     with gr.Row():
-        output1 = gr.Textbox(label="Response from Persona 1")
-        output2 = gr.Textbox(label="Response from Persona 2")
-
-    # Update labels when dropdown changes
-    def update_labels(p1, p2):
-        label1 = f"{p1} Responds"
-        label2 = f"{p2} Responds"
-        output1.label = label1
-        output2.label = label2
-        return label1, label2
-
-    persona1.change(fn=update_labels, inputs=[persona1, persona2], outputs=[output1, output2])
-    persona2.change(fn=update_labels, inputs=[persona1, persona2], outputs=[output1, output2])
+        output1 = gr.Textbox(label=f"{persona1.value} Responds")
+        output2 = gr.Textbox(label="Fact-Checker Responds")
 
     # Button events
     random_button.click(fn=get_random_question, inputs=[], outputs=[user_input])
@@ -137,16 +115,9 @@ 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 = f"I am here to verify myths and facts. Ask me anything related to {p1}."
-        else:
-            res2 = call_cohere_api(load_instruction(p2), q)
-        return res1, res2
-
     ask_button.click(
-        fn=ask_with_context, 
+        fn=lambda p1, p2, q: (call_cohere_api(load_instruction(p1), q), 
+                              call_cohere_api(load_instruction(p2), q)), 
         inputs=[persona1, persona2, user_input], 
         outputs=[output1, output2]
     )