Update src/ux_agent.py

src/ux_agent.py

@@ -61,8 +61,10 @@ class AgentUX:
         self.last_audio_path = None
         self.last_pending_count = 0
         
-        # 🟢 NEW: In-memory mailbox for live device-to-device routing
-        self.live_rooms = {} 
+        # 🟢 NEW: Matchmaking & Live Room Memory
+        self.waiting_pool = []       # List of operators looking for a match
+        self.active_matches = {}     # Maps Operator ID -> Room Code
+        self.live_rooms = {}         # Stores the actual chat logs per room
         
         self.alert_sound = None
         
@@ -179,51 +181,122 @@ class AgentUX:
             print(f"Peer Translation Error: {e}")
         return text # Fallback to original text if API fails
 
-    # 🟢 NEW: Evaluates deviation from Standard English and routes to the correct room
-    def api_remote_eval_and_send(self, room_code, sender_id, text, dialect):
-        if not text: return json.dumps({"status": "error", "msg": "Empty text"})
+    # ==========================================
+    #      DIALECT RELAY (MATCHMAKING & CHAT)
+    # ==========================================
+    def api_join_queue(self, operator_id, dialect):
+        import uuid, json, time
+        # Clean up any old ghost matches
+        if operator_id in self.active_matches:
+            del self.active_matches[operator_id]
+            
+        # 1. Check if someone else is waiting
+        partner = None
+        for p in self.waiting_pool:
+            if p['operator_id'] != operator_id:
+                partner = p
+                break
+                
+        if partner:
+            # 2. Match found! Create a room.
+            self.waiting_pool.remove(partner)
+            room_id = f"FREQ-{uuid.uuid4().hex[:6].upper()}"
+            
+            self.active_matches[operator_id] = {"room_id": room_id, "partner_dialect": partner['dialect']}
+            self.active_matches[partner['operator_id']] = {"room_id": room_id, "partner_dialect": dialect}
+            self.live_rooms[room_id] = []
+            
+            return json.dumps({"status": "matched", "room_id": room_id, "partner_dialect": partner['dialect']})
+        else:
+            # 3. No one is waiting. Add self to pool.
+            self.waiting_pool = [p for p in self.waiting_pool if p['operator_id'] != operator_id] # Prevent duplicates
+            self.waiting_pool.append({"operator_id": operator_id, "dialect": dialect, "time": time.time()})
+            return json.dumps({"status": "waiting"})
+
+    def api_check_match(self, operator_id):
+        import json
+        if operator_id in self.active_matches:
+            match = self.active_matches[operator_id]
+            return json.dumps({"status": "matched", "room_id": match["room_id"], "partner_dialect": match["partner_dialect"]})
+        return json.dumps({"status": "waiting"})
         
-        # The "Linguistic Bouncer" Prompt
-        prompt = f"""
-        Analyze this utterance from a speaker of '{dialect}': "{text}"
-        Is this mutually intelligible Standard English? 
-        If YES, return EXACTLY this JSON: {{"needs_help": false, "translation": ""}}
-        If NO (it uses heavy dialect, slang, or non-standard grammar), return EXACTLY this JSON: {{"needs_help": true, "translation": "The direct Standard English meaning here"}}
-        """
+    def api_leave_queue(self, operator_id):
+        import json
+        self.waiting_pool = [p for p in self.waiting_pool if p['operator_id'] != operator_id]
+        if operator_id in self.active_matches:
+            del self.active_matches[operator_id]
+        return json.dumps({"status": "left"})
+
+    def api_remote_eval_and_send(self, room_code, sender_id, text, source_dialect, target_dialect):
+        import json, os, re, glob, uuid, threading
+        import pandas as pd
+        
+        if not text: return json.dumps({"status": "error", "msg": "Empty text"})
         
         translation = ""
-        try:
-            res = self.brain.gemini_manager.client.models.generate_content(model='gemini-2.0-flash', contents=prompt)
-            clean_json = res.text.replace("```json", "").replace("```", "").strip()
-            data = json.loads(clean_json)
-            if data.get("needs_help"):
-                translation = data.get("translation", "")
-        except Exception as e:
-            print(f"Smart Eval Error: {e}")
+        needs_help = False
+        
+        # 1. Forward Lookup (Understand Source)
+        eval_result = self.brain.search_local_dataset(text)
+        if not eval_result: eval_result = self.brain.search_personas(text)
+        standard_meaning = eval_result.get("clarification", text) if eval_result else text
+        
+        if standard_meaning.lower() != text.lower():
+            needs_help = True
             
-        # Route to the mailbox
-        if room_code not in self.live_rooms:
-            self.live_rooms[room_code] = []
+        # 2. Reverse Lookup (Translate to Target)
+        target_utterance = None
+        if target_dialect and target_dialect != "General English" and needs_help:
+            try:
+                if hasattr(self.brain, 'config') and os.path.exists(self.brain.config.DATASET_DIR):
+                    target_file = os.path.join(self.brain.config.DATASET_DIR, f"{target_dialect}.csv")
+                    if os.path.exists(target_file):
+                        df = pd.read_csv(target_file)
+                        if 'Clarification' in df.columns and 'Utterance' in df.columns:
+                            pattern = re.compile(f"^{re.escape(standard_meaning)}$", re.IGNORECASE)
+                            matches = df[df['Clarification'].astype(str).str.match(pattern, na=False)]
+                            if not matches.empty: target_utterance = str(matches.iloc[0]['Utterance'])
+            except Exception as e: pass
+                
+        # 3. Gemini Fallback
+        if not target_utterance:
+            prompt = f"""
+            Analyze this utterance from a speaker of '{source_dialect}': "{text}"
+            Is this mutually intelligible Standard English? 
+            If YES, return EXACTLY: {{"needs_help": false, "translation": ""}}
+            If NO, translate it to '{target_dialect}' and return EXACTLY: {{"needs_help": true, "translation": "The translated text here"}}
+            """
+            try:
+                res = self.brain.gemini_manager.client.models.generate_content(model='gemini-2.0-flash', contents=prompt)
+                clean_json = res.text.replace("```json", "").replace("```", "").strip()
+                data = json.loads(clean_json)
+                if data.get("needs_help"): translation = data.get("translation", "")
+            except: translation = standard_meaning if needs_help else ""
+        else:
+            translation = target_utterance
+
+        # 🟢 THE DATA COLLECTION HOOK (Silent Background Saving)
+        clarification_to_save = translation if needs_help and translation else standard_meaning
+        try:
+            threading.Thread(
+                target=self.check_and_submit_logic,
+                args=(text, source_dialect, "", clarification_to_save, "Conversational", "Relay Peer-to-Peer Chat", "Automated Relay Extraction", "Game: Relay Pair", "AI Relay Bouncer", sender_id, None, False),
+                daemon=True
+            ).start()
+        except Exception as save_err:
+            print(f"Data Hook Error: {save_err}")
             
-        msg = {
-            "sender": sender_id,
-            "original": text,
-            "translation": translation,
-            "dialect": dialect,
-            "id": str(uuid.uuid4())[:8]
-        }
+        # 4. Route to Room
+        if room_code not in self.live_rooms: self.live_rooms[room_code] = []
+        msg = {"sender": sender_id, "original": text, "translation": translation, "dialect": source_dialect, "target_dialect": target_dialect, "id": str(uuid.uuid4())[:8]}
         self.live_rooms[room_code].append(msg)
         return json.dumps({"status": "success"})
 
-    # 🟢 NEW: Allows device B to pull messages sent by device A
     def api_remote_poll(self, room_code, last_index):
-        if room_code not in self.live_rooms:
-            return json.dumps([])
-        
+        import json
+        if room_code not in self.live_rooms: return json.dumps([])
         idx = int(last_index)
-        new_msgs = self.live_rooms[room_code][idx:]
-        return json.dumps(new_msgs)
-
+        return json.dumps(self.live_rooms[room_code][idx:])
     # ==========================================
     #            SOCIOLINGUISTIC PIPELINE
     # ==========================================
@@ -1196,20 +1269,30 @@ class AgentUX:
                     api_name="translate_peer"
                 )
 
-                # 🟢 NEW: Dialect Relay (Device-to-Device) Endpoints
+                # 🟢 NEW: Dialect Relay Endpoints
+                api_btn_join = gr.Button()
+                api_q_op = gr.Textbox()
+                api_q_dialect = gr.Textbox()
+                api_q_out = gr.Textbox()
+                api_btn_join.click(fn=self.api_join_queue, inputs=[api_q_op, api_q_dialect], outputs=[api_q_out], api_name="join_queue")
+
+                api_btn_check = gr.Button()
+                api_c_out = gr.Textbox()
+                api_btn_check.click(fn=self.api_check_match, inputs=[api_q_op], outputs=[api_c_out], api_name="check_match")
+                
+                api_btn_leave = gr.Button()
+                api_btn_leave.click(fn=self.api_leave_queue, inputs=[api_q_op], outputs=[api_c_out], api_name="leave_queue")
+
                 api_btn_relay_send = gr.Button()
                 api_relay_room = gr.Textbox()
-                api_relay_sender = gr.Textbox()
                 api_relay_text = gr.Textbox()
-                api_relay_dialect = gr.Textbox()
+                api_relay_target = gr.Textbox() 
                 api_relay_out = gr.Textbox()
-                api_btn_relay_send.click(fn=self.api_remote_eval_and_send, inputs=[api_relay_room, api_relay_sender, api_relay_text, api_relay_dialect], outputs=[api_relay_out], api_name="relay_send")
+                api_btn_relay_send.click(fn=self.api_remote_eval_and_send, inputs=[api_relay_room, api_q_op, api_relay_text, api_q_dialect, api_relay_target], outputs=[api_relay_out], api_name="relay_send")
                 
                 api_btn_relay_poll = gr.Button()
-                api_poll_room = gr.Textbox()
                 api_poll_idx = gr.Number()
-                api_poll_out = gr.Textbox()
-                api_btn_relay_poll.click(fn=self.api_remote_poll, inputs=[api_poll_room, api_poll_idx], outputs=[api_poll_out], api_name="relay_poll")
+                api_btn_relay_poll.click(fn=self.api_remote_poll, inputs=[api_relay_room, api_poll_idx], outputs=[api_c_out], api_name="relay_poll")
                 
                 # 7. Secure Feedback Endpoint
                 api_fb_btn = gr.Button()