Update app.py

app.py

@@ -13,6 +13,13 @@ zohaib = DebuggerAgent()
 
 chat_history = []
 
+# ✅ Memory in RAM
+chat_memory = {
+    "riyaz": [],
+    "zohaib": []
+}
+
+# ✅ Chat function with memory
 def chat_with_friends(user_input):
     global chat_history
 
@@ -21,16 +28,21 @@ def chat_with_friends(user_input):
     msg_riyaz = ACPMessage(sender="You", receiver="Riyaz", performative="request", content=user_input)
     msg_zohaib = ACPMessage(sender="You", receiver="Zohaib", performative="request", content=user_input)
 
-    # Log messages
     log_message_to_firestore(msg_riyaz.to_dict())
     log_message_to_firestore(msg_zohaib.to_dict())
 
-    # Typing delays
+    # Typing delay
     time.sleep(1.2)
     reply_riyaz = riyaz.receive_message(msg_riyaz)
     time.sleep(1.0)
     reply_zohaib = zohaib.receive_message(msg_zohaib)
 
+    # Store in memory
+    chat_memory["riyaz"].append(f"User: {user_input.strip()}")
+    chat_memory["riyaz"].append(f"Riyaz: {reply_riyaz.content}")
+    chat_memory["zohaib"].append(f"User: {user_input.strip()}")
+    chat_memory["zohaib"].append(f"Zohaib: {reply_zohaib.content}")
+
     log_message_to_firestore(reply_riyaz.to_dict())
     log_message_to_firestore(reply_zohaib.to_dict())
 
@@ -39,7 +51,14 @@ def chat_with_friends(user_input):
 
     return chat_history
 
-# Gradio Interface
+# ✅ Reset memory + chat
+def reset_all():
+    chat_history.clear()
+    chat_memory["riyaz"].clear()
+    chat_memory["zohaib"].clear()
+    return []
+
+# ✅ Gradio Interface
 with gr.Blocks(css="""
 .user-msg {
     background-color: #ffd6d6;
@@ -66,7 +85,6 @@ with gr.Blocks(css="""
     height: 250px !important;
     overflow-y: auto;
 }
-/* Circular avatar */
 .gr-chatbot .avatar {
     border-radius: 50% !important;
     height: 38px !important;
@@ -75,17 +93,20 @@ with gr.Blocks(css="""
     margin-right: 6px;
 }
 """) as demo:
-    gr.Markdown("## 🤖 BotTalks: Chat with Friends Riyaz & Zohaib\nAsk anything and see what your AI friends say!")
+    gr.Markdown("## 🤖 BotTalks: Chat with Friends Riyaz & Zohaib\nAsk anything, watch how they reply 💬")
 
     chatbot = gr.Chatbot(label="Group Chat", bubble_full_width=False, show_copy_button=True)
 
     with gr.Row():
-        msg_input = gr.Textbox(placeholder="Ask anything, e.g., 'how to sort a list in Python?'", lines=1, show_label=False)
+        msg_input = gr.Textbox(placeholder="Ask anything like 'how do I write a function?'", lines=1, show_label=False)
         send_btn = gr.Button("Send", variant="primary")
+        reset_btn = gr.Button("🧹 Reset Memory", variant="secondary")
 
     send_btn.click(chat_with_friends, inputs=msg_input, outputs=chatbot)
     msg_input.submit(chat_with_friends, inputs=msg_input, outputs=chatbot)
 
+    reset_btn.click(fn=reset_all, inputs=[], outputs=chatbot)
+
     gr.Markdown("""
     <div style='text-align: right; font-size: 0.9em; padding-top: 15px;'>
     🔗 Connect: