Update app.py

app.py

@@ -4,75 +4,47 @@ import gradio as gr
 from agents.programmer import ProgrammerAgent
 from agents.debugger import DebuggerAgent
 from agents.base_agent import ACPMessage
-# log_message_to_firestore(msg.to_dict())
+from firebase.firebase_config import log_message_to_firestore
 
-# Initialize agents
+# Init agents
 codebot = ProgrammerAgent()
 bugbot = DebuggerAgent()
 
-# Chat history
-conversation = []
+def chat_with_bots(user_input):
+    # Send same input to both agents
+    msg_to_codebot = ACPMessage(sender="User", receiver="CodeBot", performative="request", content=user_input)
+    msg_to_bugbot = ACPMessage(sender="User", receiver="BugBot", performative="request", content=user_input)
 
-def start_conversation():
-    global conversation
-    conversation = []  # reset
+    # Log inputs
+    log_message_to_firestore(msg_to_codebot.to_dict())
+    log_message_to_firestore(msg_to_bugbot.to_dict())
 
-    # Initial message from BugBot
-    msg = bugbot.create_message(
-        receiver=codebot.name,
-        performative="request",
-        content="Can you write a Python function to reverse a list?"
-    )
-    conversation.append(str(msg))
+    # Get both responses
+    reply_codebot = codebot.receive_message(msg_to_codebot)
+    reply_bugbot = bugbot.receive_message(msg_to_bugbot)
 
-    # CodeBot responds
-    reply = codebot.receive_message(msg)
-    conversation.append(str(reply))
+    # Log responses
+    log_message_to_firestore(reply_codebot.to_dict())
+    log_message_to_firestore(reply_bugbot.to_dict())
 
-    return "\n\n".join(conversation)
+    # Format reply for chat
+    full_reply = f"""
+🧠 **CodeBot** (Programmer):  
+{reply_codebot.content}
 
-def continue_conversation():
-    if len(conversation) < 2:
-        return "Please start the conversation first."
+🐞 **BugBot** (Debugger):  
+{reply_bugbot.content}
+"""
+    return full_reply.strip()
 
-    # Last reply came from CodeBot → BugBot now responds
-    last_msg = conversation[-1]
-    parsed = parse_acp_string(last_msg)
-    reply = bugbot.receive_message(parsed)
-    conversation.append(str(reply))
+# Gradio UI
+with gr.Blocks(title="BotTalks: Chat with 2 AI Agents") as demo:
+    gr.Markdown("# 🤖 BotTalks: Chat with 2 AI Friends!\nAsk anything and see how both agents respond.")
+    input_box = gr.Textbox(label="You", placeholder="Ask something like: How do I reverse a list?", lines=2)
+    output_box = gr.Textbox(label="Responses", lines=10)
 
-    # CodeBot replies again
-    reply2 = codebot.receive_message(reply)
-    conversation.append(str(reply2))
+    send_btn = gr.Button("Send")
 
-    return "\n\n".join(conversation)
+    send_btn.click(chat_with_bots, inputs=input_box, outputs=output_box)
 
-def parse_acp_string(message_str: str) -> ACPMessage:
-    """
-    Convert "[PERFORMATIVE] sender → receiver: content"
-    back to an ACPMessage object
-    """
-    try:
-        parts = message_str.split("] ", 1)
-        performative = parts[0][1:].lower()
-        meta, content = parts[1].split(": ", 1)
-        sender, receiver = [s.strip() for s in meta.split("→")]
-        return ACPMessage(sender, receiver, performative, content.strip())
-    except Exception:
-        return ACPMessage("Unknown", "Unknown", "inform", "Failed to parse message.")
-
-# Gradio interface
-with gr.Blocks(title="BotTalks") as demo:
-    gr.Markdown("# 🤖 BotTalks: Programmer vs Debugger\nAgent Communication Protocol Chat")
-
-    chatbox = gr.Textbox(label="Conversation", lines=20)
-
-    with gr.Row():
-        start_btn = gr.Button("Start Conversation 🚀")
-        next_btn = gr.Button("Next Turn 🔁")
-
-    start_btn.click(start_conversation, outputs=chatbox)
-    next_btn.click(continue_conversation, outputs=chatbox)
-
-# Run app
 demo.launch()