Update app.py

app.py

@@ -1,67 +1,59 @@
-# app/app.py
-
 import gradio as gr
-from agents.programmer import ProgrammerAgent
-from agents.debugger import DebuggerAgent
+from app.agents.programmer import ProgrammerAgent
+from app.agents.debugger import DebuggerAgent
 from agents.base_agent import ACPMessage
 
-# Instantiate agents
 aymaan = ProgrammerAgent()
 zaid = DebuggerAgent()
 
-# Agent memory
-memory = []
+chat_memory = []
 
-# Chat function
 def chat(user_input):
     if not user_input.strip():
-        return memory
-
-    # Format user's message
-    memory.append({"role": "user", "content": user_input, "name": "You", "avatar": "🧑", "color": "#ff0000"})
-
-    # 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)
-
-    # Get responses
-    aymaan_reply = aymaan.receive_message(msg_for_aymaan)
-    zaid_reply = zaid.receive_message(msg_for_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"})
-
-    return memory
-
-# Reset memory
-def reset():
-    memory.clear()
-    return memory
-
-# 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)
-
+        return chat_memory
+
+    user_msg = ACPMessage(sender="User", receiver="Aymaan", performative="inform", content=user_input)
+
+    # Responses from agents
+    response_aymaan = aymaan.receive_message(user_msg)
+    response_zaid = zaid.receive_message(user_msg)
+
+    # Save to memory
+    chat_memory.append({"role": "user", "content": user_input})
+    chat_memory.append({"role": "aymaan", "content": response_aymaan.content})
+    chat_memory.append({"role": "zaid", "content": response_zaid.content})
+
+    return chat_memory
+
+def reset_memory():
+    global chat_memory
+    chat_memory = []
+    return chat_memory
+
+with gr.Blocks(css="footer {margin-top: 2em; text-align: center;}") as demo:
+    gr.Markdown("## 🤖 BotTalks: Chat with 2 AI Friends!\nAsk anything and see what Aymaan and Zaid say.")
+    
+    chatbot = gr.Chatbot(label="Group Chat", type='messages', height=250)
+    msg = gr.Textbox(label="You")
+    send_btn = gr.Button("Send")
+    clear_btn = gr.Button("🧹 Reset Memory")
+    
     with gr.Row():
-        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()
+        send_btn.render()
+        clear_btn.render()
+
+    send_btn.click(chat, inputs=msg, outputs=chatbot)
+    clear_btn.click(reset_memory, outputs=chatbot)
+    
+    gr.HTML("""
+    <footer>
+        🔗 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>
+    </footer>
+    """)
+
+if __name__ == "__main__":
+    demo.launch()