modify graph to set limits; visualization bugs

pages/play.py

@@ -278,6 +278,7 @@ with gr.Blocks(fill_width=True) as demo:
 
     def update_api_key_inputs(red_team, blue_team):
         """Update visibility and state of API key inputs based on team selection."""
+
         required_teams = get_required_teams(red_team, blue_team)
 
         # Return visibility states for each team's API input
@@ -315,8 +316,6 @@ with gr.Blocks(fill_width=True) as demo:
         accumulated_messages = list(messages) if messages else []
         async for new_msg, guessed, updated_board, updated_chat_history, winners in graph.stream_graph(user_msg, messages, players, board, dropdown, guessed, chat_history, is_human_playing, turn):
 
-            logger.error(f"Winners: {winners}")
-
             if len(new_msg) > len(accumulated_messages):
                 # Add only the new messages that weren't there before
                 accumulated_messages = list(new_msg)
@@ -328,16 +327,20 @@ with gr.Blocks(fill_width=True) as demo:
             feed_html = format_messages_as_feed(accumulated_messages, players, winners)
             yield feed_html, guessed, messages, updated_board, updated_chat_history, winners
 
-    def update_plot(guessed_words, board, previous_bord_img):
+    def update_plot(guessed_words, board, game_state):
+        # Don't update if no game is active
+        if game_state is None:
+            return gr.update()  # No update
+        
         if not guessed_words or not board:
-            return previous_bord_img
+            return gr.update()  # No update
+            
         logger.info(f"Calling update_plot: {guessed_words}")
         board_img = plot_game_board_with_guesses(board, guessed_words)
         return board_img
 
     def deactivate_send(game, winner_and_score):
         if winner_and_score is None or game is None:
-            logger.error(f"Winner state: {winner_and_score}")
             return gr.Button(visible=True)
         else:
             return gr.Button(visible=False)
@@ -482,7 +485,7 @@ with gr.Blocks(fill_width=True) as demo:
 
     guessed_words_state.change(
         fn=update_plot,
-        inputs=[guessed_words_state, board_state, board_plot],
+        inputs=[guessed_words_state, board_state, game_state],  # Remove board_plot from inputs
         outputs=[board_plot]
     )
 

support/build_graph.py

@@ -11,6 +11,7 @@ from langgraph.prebuilt import ToolNode
 from support.log_manager import logger
 from support.my_tools import captain_agent_tools, boss_agent_tools
 from support.tools.boss_agent_tools import create_fake_tool_call
+from support.tools.captain_agent_tools import create_agent_fake_tool_call
 from support.prompts import captain_agent_system_prompt, player_agent_system_prompt, boss_agent_system_prompt
 from typing import Annotated, List, Literal, Optional
 from typing_extensions import TypedDict
@@ -38,6 +39,11 @@ class State(TypedDict):
     end_round: bool
     winner_and_score: tuple
 
+    red_boss_is_called_counter: int
+    blue_boss_is_called_counter: int
+    red_captain_is_called_counter: int
+    blue_captain_is_called_counter: int
+
 
 class MyGraph:
     def __init__(self):
@@ -64,6 +70,8 @@ class MyGraph:
         builder.add_node("red_captain", self.red_captain)
         builder.add_node("red_agent_1", self.red_agent_1)
         builder.add_node("red_agent_2", self.red_agent_2)
+        builder.add_node("red_boss_is_called", self.red_boss_is_called)
+        builder.add_node("red_captain_is_called", self.red_captain_is_called)
 
         # Tool nodes
         choose_word_tool = ToolNode(tools=[boss_agent_tools[0]])
@@ -84,10 +92,11 @@ class MyGraph:
             self.boss_choice,
             {
                 "choose_word_tool": "choose_word_tool",
-                "red_boss": "red_boss",
+                "red_boss_is_called": "red_boss_is_called",
             },
         )
-        # builder.add_edge("red_boss", "choose_word_tool")
+
+        builder.add_edge("red_boss_is_called", "red_boss")
         builder.add_edge("choose_word_tool", "red_captain")
 
         builder.add_conditional_edges(
@@ -97,10 +106,11 @@ class MyGraph:
                 "final_choice": "final_choice",
                 "transfer_to_agent_1": "transfer_to_agent_1",
                 "transfer_to_agent_2": "transfer_to_agent_2",
-                "red_captain": "red_captain",
+                "red_captain_is_called": "red_captain_is_called",
             },
         )
 
+        builder.add_edge("red_captain_is_called", "red_captain")
         builder.add_edge("transfer_to_agent_1", "red_agent_1")
         builder.add_edge("transfer_to_agent_2", "red_agent_2")
         builder.add_edge("red_agent_1", "red_captain")
@@ -119,6 +129,8 @@ class MyGraph:
         builder.add_node("blue_captain", self.blue_captain)
         builder.add_node("blue_agent_1", self.blue_agent_1)
         builder.add_node("blue_agent_2", self.blue_agent_2)
+        builder.add_node("blue_boss_is_called", self.blue_boss_is_called)
+        builder.add_node("blue_captain_is_called", self.blue_captain_is_called)
 
         # Tool nodes
         choose_word_tool = ToolNode(tools=[boss_agent_tools[0]])
@@ -139,10 +151,11 @@ class MyGraph:
             self.boss_choice,
             {
                 "choose_word_tool": "choose_word_tool",
-                "blue_boss": "blue_boss",
+                "blue_boss_is_called": "blue_boss_is_called",
             },
         )
-        # builder.add_edge("blue_boss", "choose_word_tool")
+
+        builder.add_edge("blue_boss_is_called", "blue_boss")
         builder.add_edge("choose_word_tool", "blue_captain")
 
         builder.add_conditional_edges(
@@ -152,10 +165,11 @@ class MyGraph:
                 "final_choice": "final_choice",
                 "transfer_to_agent_1": "transfer_to_agent_1",
                 "transfer_to_agent_2": "transfer_to_agent_2",
-                "blue_captain": "blue_captain",
+                "blue_captain_is_called": "blue_captain_is_called",
             },
         )
 
+        builder.add_edge("blue_captain_is_called", "blue_captain")
         builder.add_edge("transfer_to_agent_1", "blue_agent_1")
         builder.add_edge("transfer_to_agent_2", "blue_agent_2")
         builder.add_edge("blue_agent_1", "blue_captain")
@@ -320,7 +334,13 @@ class MyGraph:
         llm_with_tools = captain.model.bind_tools(
             captain_agent_tools,
         )
-        answer = await llm_with_tools.ainvoke(new_message)
+
+        if state['red_captain_is_called_counter'] > 4:
+            logger.info("Creating fake answer to stop loop")
+            answer = create_agent_fake_tool_call()
+        else:
+            logger.info(f"red_captain_is_called_counter: {state['red_captain_is_called_counter']}")
+            answer = await llm_with_tools.ainvoke(new_message)
 
         logger.info(f"[RED CAPTAIN ANSWER]: {answer}")
         logger.info("°°°"*50)
@@ -576,7 +596,13 @@ class MyGraph:
         llm_with_tools = captain.model.bind_tools(
             captain_agent_tools,
         )
-        answer = await llm_with_tools.ainvoke(new_message)
+
+        if state['blue_captain_is_called_counter'] > 4:
+            logger.info("Creating fake answer to stop loop")
+            answer = create_agent_fake_tool_call()
+        else:
+            logger.info(f"blue_captain_is_called_counter: {state['blue_captain_is_called_counter']}")
+            answer = await llm_with_tools.ainvoke(new_message)
 
         logger.info(f"[BLUE CAPTAIN ANSWER] : {answer}")
         logger.info("°°°"*50)
@@ -718,6 +744,38 @@ class MyGraph:
             "turn": turn+1
         }
 
+    def red_boss_is_called(self, state: State):
+        """Update turn counter"""
+        _counter = state.get("red_boss_is_called_counter", 1)
+
+        return {
+            "red_boss_is_called_counter": _counter+1
+        }
+
+    def blue_boss_is_called(self, state: State):
+        """Update turn counter"""
+        _counter = state.get("blue_boss_is_called_counter", 1)
+
+        return {
+            "blue_boss_is_called_counter": _counter+1
+        }
+
+    def red_captain_is_called(self, state: State):
+        """Update turn counter"""
+        _counter = state.get("red_captain_is_called_counter", 1)
+
+        return {
+            "red_captain_is_called_counter": _counter+1
+        }
+
+    def blue_captain_is_called(self, state: State):
+        """Update turn counter"""
+        _counter = state.get("blue_captain_is_called_counter", 1)
+
+        return {
+            "blue_captain_is_called_counter": _counter+1
+        }
+
     def judge(self, state: State):
         """Evaluate guesses, update board, check win/lose conditions."""
 
@@ -1074,7 +1132,7 @@ class MyGraph:
             logger.warning("[WARNING] NESSUN TOOL E' STATO RICHIAMATO, TORNIAMO DAL BOSS")
             current_team = state['current_team']
             logger.info(f"[CURRENT TEAM: {current_team}]")
-            return f"{current_team}_boss"
+            return f"{current_team}_boss_is_called"
 
     def should_continue(self, state: State) -> str:
         """Determine whether to continue to tools or return a final answer."""
@@ -1099,7 +1157,7 @@ class MyGraph:
             logger.warning("[WARNING] NESSUN TOOL E' STATO RICHIAMATO, TORNIAMO DAL CAPITANO")
             current_team = state['current_team']
             logger.info(f"[CURRENT TEAM: {current_team}]")
-            return f"{current_team}_captain"
+            return f"{current_team}_captain_is_called"
 
     def _convert_to_string(self, value):
         """Helper to convert any value to string"""
@@ -1247,7 +1305,11 @@ class MyGraph:
             "teams_reviewed": [],
             "history_guessed_words": guessed_words,
             "end_round": False,
-            "winner_and_score": None
+            "winner_and_score": None,
+            "red_boss_is_called_counter": 0,
+            "blue_boss_is_called_counter": 0,
+            "red_captain_is_called_counter": 0,
+            "blue_captain_is_called_counter": 0,
         }
 
         final_msg = ""

support/my_tools.py

@@ -39,24 +39,6 @@ async def load_tools():
     return boss_agent_tools, captain_agent_tools
 
 
-# boss_client = MultiServerMCPClient(
-#     {
-#         "BossServer": {
-#             "url": "http://localhost:8000/mcp",
-#             "transport": "streamable_http",
-#         }
-#     }
-# )
-
-# captain_client = MultiServerMCPClient(
-#     {
-#         "CaptainServer": {
-#             "url": "http://localhost:8001/mcp",
-#             "transport": "streamable_http",
-#         }
-#     }
-# )
-
 boss_client = MultiServerMCPClient(
     {
         "BossServer": {

support/style/css.py

@@ -1854,7 +1854,7 @@ strong {
     margin-bottom: 16px;
     box-shadow: 0 2px 8px rgba(0,0,0,0.1);
     transition: transform 0.2s, box-shadow 0.2s;
-    border-left: 4px solid #e0e0e0;
+    border-left: 8px solid #e0e0e0;
 }
 
 .message_card:hover {
@@ -1908,11 +1908,19 @@ strong {
     padding-left: 52px;
 }
 
-.message_content li,
-.message_content code {
+.message_content,
+.message_content * {
     color: black;
 }
 
+.message_content think{
+    font-style: italic;
+}
+
+.message_content .prose * {
+    color: #000000 !important;
+}
+
 .message_thinking {
     border-left-color: #9b59b6;
     background: linear-gradient(to right, #f8f4fc, white);
@@ -2247,7 +2255,7 @@ final_css = navbar_css + stats_css + css + """
     margin-bottom: 16px;
     box-shadow: 0 2px 8px rgba(0,0,0,0.1);
     transition: transform 0.2s, box-shadow 0.2s;
-    border-left: 4px solid #e0e0e0;
+    border-left: 8px solid #e0e0e0;
 }
 
 .message_card:hover {

support/tools/captain_agent_tools.py

@@ -0,0 +1,23 @@
+# Fake tool call generator when human user is playing
+
+def create_agent_fake_tool_call():
+    from langchain_core.messages import AIMessage
+    from uuid import uuid4
+    """Creates a fake AIMessage that mimics an LLM tool call"""
+    tool_call_id = str(uuid4())
+
+    fake_message = AIMessage(
+        content="",
+        tool_calls=[
+            {
+                "name": "TeamFinalChoice",
+                "args": {
+                    "guesses": ['STOP_TURN'],
+                },
+                "id": tool_call_id,
+                "type": "tool_call"
+            }
+        ]
+    )
+
+    return fake_message

support/utils.py

@@ -1,10 +1,17 @@
 import io
 import markdown
+
+import matplotlib
+matplotlib.use('Agg')
 import matplotlib.pyplot as plt
 import matplotlib.patches as mpatches
+import os
 import regex
+import tempfile
+import uuid
 
 from PIL import Image
+from support.log_manager import logger
 
 
 def display_model_name(model_name):
@@ -369,124 +376,14 @@ def format_game_history_html(game_history):
     return "".join(games_html)
 
 
-def plot_game_board(board):
-    """
-    Create a single combined visualization showing words with their color-coded backgrounds.
-    Each card displays the word in white text on a colored background (red, blue, beige, black).
-    """
-    # Color mapping
-    color_map = {
-        'red': '#dc3545',
-        'blue': '#0d6efd',
-        'neutral': '#d4c5b9',  # beige
-        'killer': '#212529'     # black
-    }
-
-    # Get word-color pairs and arrange in 5x5 grid
-    word_color_pairs = board.get('word_color_pairs', [])
-    if len(word_color_pairs) < 25:
-        word_color_pairs = word_color_pairs + [("???", 'neutral')] * (25 - len(word_color_pairs))
-
-    # Create figure with specific size
-    fig, ax = plt.subplots(figsize=(12, 12))
-    ax.set_xlim(-0.5, 4.5)
-    ax.set_ylim(-0.5, 4.5)
-    ax.set_xticks([])
-    ax.set_yticks([])
-    ax.axis('off')
-    ax.set_aspect('equal')
-
-    # Draw grid of cards
-    idx = 0
-    for i in range(5):
-        for j in range(5):
-            if idx >= len(word_color_pairs):
-                break
-
-            word, color = word_color_pairs[idx]
-            bg_color = color_map.get(color, '#e9ecef')
-
-            # Draw card background with rounded corners effect
-            rect = mpatches.FancyBboxPatch(
-                (j - 0.48, i - 0.48), 0.96, 0.96,
-                boxstyle="round,pad=0.02",
-                facecolor=bg_color,
-                edgecolor='#495057',
-                linewidth=2.5,
-                zorder=1
-            )
-            ax.add_patch(rect)
-
-            # Add word text in white
-            # Adjust font size based on word length
-            word_length = len(word)
-            if word_length > 12:
-                fontsize = 13
-            elif word_length > 8:
-                fontsize = 15
-            else:
-                fontsize = 17
-
-            ax.text(
-                j, i, word,
-                ha="center", va="center",
-                fontsize=fontsize,
-                fontweight='bold',
-                color='white',
-                zorder=2,
-                wrap=True
-            )
-
-            idx += 1
-
-    plt.gca().invert_yaxis()
-
-    # Add title with starting team info
-    starting_team = board.get('starting_team', 'red')
-    title_color = color_map[starting_team]
-
-    fig.suptitle(
-        f'{starting_team.upper()} TEAM STARTS',
-        fontsize=18,
-        fontweight='bold',
-        color=title_color,
-        y=0.98
-    )
-
-    # Add legend
-    legend_elements = [
-        mpatches.Patch(facecolor=color_map['red'], edgecolor='#495057', label='Red Team'),
-        mpatches.Patch(facecolor=color_map['blue'], edgecolor='#495057', label='Blue Team'),
-        mpatches.Patch(facecolor=color_map['neutral'], edgecolor='#495057', label='Neutral'),
-        mpatches.Patch(facecolor=color_map['killer'], edgecolor='#495057', label='Killer ☠️')
-    ]
-    ax.legend(
-        handles=legend_elements,
-        loc='upper center',
-        bbox_to_anchor=(0.5, -0.02),
-        ncol=4,
-        frameon=True,
-        fancybox=True,
-        shadow=True,
-        fontsize=11
-    )
-
-    plt.tight_layout()
-
-    # Save to buffer
-    buf = io.BytesIO()
-    plt.savefig(buf, format="png", bbox_inches="tight", dpi=120, facecolor='white')
-    buf.seek(0)
-    plt.close(fig)
-
-    return Image.open(buf)
-
-
 def plot_game_board_with_guesses(board, guessed_words=None):
     """
     Plot the game board with crosses or marks over guessed words.
     """
 
+    logger.info(f"Board: {board}")
+    logger.info(f"Guessed words: {guessed_words}")
+
     color_map = {
         'red': '#dc3545',
         'blue': '#0d6efd',
@@ -499,7 +396,9 @@ def plot_game_board_with_guesses(board, guessed_words=None):
     if len(word_color_pairs) < 25:
         word_color_pairs += [("???", 'neutral')] * (25 - len(word_color_pairs))
 
-    fig, ax = plt.subplots(figsize=(12, 12))
+    # Create a new figure explicitly (don't rely on global state)
+    fig = plt.figure(figsize=(12, 12))
+    ax = fig.add_subplot(111)
     ax.set_xlim(-0.5, 4.5)
     ax.set_ylim(-0.5, 4.5)
     ax.axis('off')
@@ -514,7 +413,6 @@ def plot_game_board_with_guesses(board, guessed_words=None):
             word, color = word_color_pairs[idx]
             bg_color = color_map.get(color, '#e9ecef')
 
-            # Draw card background
             rect = mpatches.FancyBboxPatch(
                 (j - 0.48, i - 0.48), 0.96, 0.96,
                 boxstyle="round,pad=0.02",
@@ -525,7 +423,6 @@ def plot_game_board_with_guesses(board, guessed_words=None):
             )
             ax.add_patch(rect)
 
-            # Add word
             ax.text(
                 j, i, word,
                 ha="center", va="center",
@@ -536,7 +433,6 @@ def plot_game_board_with_guesses(board, guessed_words=None):
                 wrap=True
             )
 
-            # If word was guessed, overlay a cross
             if word in guessed_words:
                 ax.plot(
                     [j - 0.4, j + 0.4], [i - 0.4, i + 0.4],
@@ -549,7 +445,7 @@ def plot_game_board_with_guesses(board, guessed_words=None):
 
             idx += 1
 
-    plt.gca().invert_yaxis()
+    ax.invert_yaxis()
     starting_team = board.get('starting_team', 'red')
     title_color = color_map[starting_team]
 
@@ -561,9 +457,26 @@ def plot_game_board_with_guesses(board, guessed_words=None):
         y=0.98
     )
 
-    plt.tight_layout()
-    buf = io.BytesIO()
-    plt.savefig(buf, format="png", bbox_inches="tight", dpi=120, facecolor='white')
-    buf.seek(0)
-    plt.close(fig)
-    return Image.open(buf)
+    fig.tight_layout()
+
+    # Create a unique temporary file for this specific request
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png', prefix=f'board_{uuid.uuid4().hex}_')
+    temp_filename = temp_file.name
+    temp_file.close()
+
+    try:
+        # Save to the unique temporary file
+        fig.savefig(temp_filename, format="png", bbox_inches="tight", dpi=120, facecolor='white')
+        plt.close(fig)  # Close the figure immediately
+
+        # Load the image from the file
+        img = Image.open(temp_filename)
+        img.load()  # Load into memory
+
+        return img
+    finally:
+        # Clean up the temporary file
+        try:
+            os.unlink(temp_filename)
+        except:
+            pass