Update app.py

app.py

@@ -1,76 +1,75 @@
 import gradio as gr
-import json
-import os
 from agents.programmer import ProgrammerAgent
 from agents.debugger import DebuggerAgent
 from agents.base_agent import ACPMessage
+import json
+import os
 
-# === Init Agents ===
+# Initialize agents
 aymaan = ProgrammerAgent()
 zaid = DebuggerAgent()
+
+# In-memory chat
 chat_memory = []
 
-# === Load example prompts from JSON ===
-def load_prompts():
-    try:
-        with open("data/prompts.json", "r", encoding="utf-8") as f:
-            return json.load(f).get("prompts", [])
-    except Exception as e:
-        print(f"Error loading prompts: {e}")
-        return []
+# Load greeting shortcut mappings (optional JSON file)
+GREETING_RESPONSES = {
+    "hi": "Hey there! 👋",
+    "hello": "Hello! How can I help?",
+    "how are you": "Doing great! Thanks for asking 😊",
+}
 
-example_prompts = load_prompts()
+def fast_greeting_handler(msg):
+    lowered = msg.lower()
+    for greeting in GREETING_RESPONSES:
+        if greeting in lowered:
+            return True, GREETING_RESPONSES[greeting]
+    return False, ""
 
-# === Main Chat Handler ===
 def chat(user_input):
     if not user_input.strip():
         return chat_memory
 
-    user_msg = ACPMessage(sender="User", receiver="Aymaan", performative="inform", content=user_input)
-
-    # Fast handling for greetings
-    if aymaan.is_greeting(user_input):
-        response_aymaan = aymaan.create_message("User", "inform", "Hey there! 😊 I'm ready to help you.")
-        response_zaid = zaid.create_message("User", "inform", "Hi hi! 👋 How can I assist you today?")
+    is_greeting, shortcut = fast_greeting_handler(user_input)
+    
+    if is_greeting:
+        response_aymaan = shortcut + " (Aymaan)"
+        response_zaid = shortcut + " (Zaid)"
     else:
-        response_aymaan = aymaan.receive_message(user_msg)
-        response_zaid = zaid.receive_message(user_msg)
+        user_msg = ACPMessage(sender="User", receiver="Aymaan", performative="inform", content=user_input)
+        response_aymaan = aymaan.receive_message(user_msg).content
+
+        user_msg = ACPMessage(sender="User", receiver="Zaid", performative="inform", content=user_input)
+        response_zaid = zaid.receive_message(user_msg).content
 
     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})
+    chat_memory.append({"role": "Aymaan", "content": response_aymaan})
+    chat_memory.append({"role": "Zaid", "content": response_zaid})
 
-    return chat_memory
+    formatted = []
+    for item in chat_memory:
+        formatted.append((item["role"], item["content"]))
+    return formatted
 
-# === Memory Reset ===
 def reset_memory():
     global chat_memory
     chat_memory = []
-    return chat_memory
+    return []
 
-# === App UI ===
 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=300)
-    msg = gr.Textbox(label="You", placeholder="Ask anything...")
-    send_btn = gr.Button("📤 Send")
-    clear_btn = gr.Button("🧹 Reset Memory")
+    gr.Markdown("## 🤖 BotTalks (Groq Edition) — Chat with Aymaan & Zaid")
 
-    with gr.Row():
-        for prompt in example_prompts[:4]:  # show first 4 prompts
-            gr.Button(value=prompt).click(fn=chat, inputs=gr.Textbox(value=prompt, visible=False), outputs=chatbot)
+    chatbot = gr.Chatbot(label="Group Chat", height=300)
+    msg = gr.Textbox(label="Type your message...")
+    send_btn = gr.Button("Send")
+    clear_btn = gr.Button("🧹 Clear Chat")
 
     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>
+        👨‍💻 Made by <a href="https://github.com/aymnsk" target="_blank">aymnsk</a> • Powered by <a href="https://groq.com/" target="_blank">Groq API</a>
     </footer>
     """)