Update app.py

app.py

@@ -1,92 +1,67 @@
-# app.py
+# app/app.py
 
-import time
 import gradio as gr
 from agents.programmer import ProgrammerAgent
 from agents.debugger import DebuggerAgent
 from agents.base_agent import ACPMessage
 
-# Initialize agents
+# Instantiate agents
 aymaan = ProgrammerAgent()
 zaid = DebuggerAgent()
 
-chat_history = []
+# Agent memory
+memory = []
 
-# Memory
-chat_memory = {
-    "aymaan": [],
-    "zaid": []
-}
+# Chat function
+def chat(user_input):
+    if not user_input.strip():
+        return memory
 
-# Chat Function
-def chat_with_friends(user_input):
-    global chat_history
+    # Format user's message
+    memory.append({"role": "user", "content": user_input, "name": "You", "avatar": "🧑", "color": "#ff0000"})
 
-    try:
-        chat_history.append({"role": "user", "content": user_input})
+    # Create messages for both agents
+    msg_for_aymaan = ACPMessage(sender="You", receiver="Aymaan", performative="inform", content=user_input)
+    msg_for_zaid = ACPMessage(sender="You", receiver="Zaid", performative="inform", content=user_input)
 
-        msg_aymaan = ACPMessage(sender="You", receiver="Aymaan", performative="request", content=user_input)
-        msg_zaid = ACPMessage(sender="You", receiver="Zaid", performative="request", content=user_input)
+    # Get responses
+    aymaan_reply = aymaan.receive_message(msg_for_aymaan)
+    zaid_reply = zaid.receive_message(msg_for_zaid)
 
-        reply_aymaan = aymaan.receive_message(msg_aymaan)
-        reply_zaid = zaid.receive_message(msg_zaid)
+    # Append replies to memory
+    memory.append({"role": "assistant", "content": aymaan_reply.content, "name": "Aymaan", "avatar": "💬", "color": "#00cc66"})
+    memory.append({"role": "assistant", "content": zaid_reply.content, "name": "Zaid", "avatar": "🐞", "color": "#ffa500"})
 
-        chat_memory["aymaan"].append(f"You: {user_input}")
-        chat_memory["aymaan"].append(f"Aymaan: {reply_aymaan.content}")
-        chat_memory["zaid"].append(f"You: {user_input}")
-        chat_memory["zaid"].append(f"Zaid: {reply_zaid.content}")
+    return memory
 
-        chat_history.append({"role": "assistant", "name": "Aymaan", "content": reply_aymaan.content})
-        chat_history.append({"role": "assistant", "name": "Zaid", "content": reply_zaid.content})
+# Reset memory
+def reset():
+    memory.clear()
+    return memory
 
-        return chat_history
-
-    except Exception as e:
-        chat_history.append({"role": "assistant", "name": "System", "content": f"⚠️ Error: {str(e)}"})
-        return chat_history
-
-# Reset button
-def reset_all():
-    chat_history.clear()
-    chat_memory["aymaan"].clear()
-    chat_memory["zaid"].clear()
-    return []
-
-# UI
-with gr.Blocks(css="""
-.gr-chatbot {
-    height: 250px !important;
-    overflow-y: auto;
-}
-.gr-chatbot .avatar {
-    border-radius: 50% !important;
-    height: 38px !important;
-    width: 38px !important;
-    object-fit: cover !important;
-    margin-right: 6px;
-}
-""") as demo:
-    gr.Markdown("## 🤖 BotTalks: Chat with Aymaan & Zaid — Your AI Coding Buddies")
-
-    chatbot = gr.Chatbot(label="Group Chat", show_copy_button=True, type="messages")
+# Custom chat styling
+with gr.Blocks(theme=gr.themes.Base(), css=".message { border-radius: 999px !important; min-height: 250px; }") as demo:
+    gr.Markdown("## 🤖 BotTalks: Chat with Aymaan & Zaid (TinyLLaMA + Mistral)")
+    chatbot = gr.Chatbot(label="Group Chat", type='messages', show_copy_button=True)
 
     with gr.Row():
-        msg_input = gr.Textbox(placeholder="Say hi or ask a coding question!", 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:
-    <a href="https://linkedin.com/in/aymnsk" target="_blank">LinkedIn</a> |
-    <a href="https://github.com/aymnsk" target="_blank">GitHub</a> |
-    <a href="https://instagram.com/damnn_aymn" target="_blank">Instagram</a> |
-    <a href="https://huggingface.co/aymnsk" target="_blank">HF</a>
-    </div>
-    """)
+        msg = gr.Textbox(placeholder="Say something...", scale=8)
+        send_btn = gr.Button("Send", scale=1)
+        reset_btn = gr.Button("🧹 Reset Memory", scale=1)
+
+    send_btn.click(chat, inputs=[msg], outputs=[chatbot])
+    msg.submit(chat, inputs=[msg], outputs=[chatbot])
+    reset_btn.click(reset, outputs=[chatbot])
+
+    # Footer with social links
+    gr.Markdown(
+        "<div style='text-align: right; font-size: 14px;'>"
+        "🔗 Connect: "
+        "<a href='https://www.linkedin.com/in/aymnsk' target='_blank'>LinkedIn</a> | "
+        "<a href='https://github.com/aymnsk' target='_blank'>GitHub</a> | "
+        "<a href='https://www.instagram.com/damnn_aymn/' target='_blank'>Instagram</a> | "
+        "<a href='https://huggingface.co/aymnsk' target='_blank'>HF</a>"
+        "</div>"
+    )
 
 demo.launch()