cleaning up gradio messages and showing player profile

api/event_handlers/gradio_handler.py

@@ -5,6 +5,19 @@ from langchain_core.outputs.llm_result import LLMResult
 from typing import List
 from langchain_core.messages import BaseMessage
 
+image_base = """
+<img 
+    src="https://huggingface.co/spaces/ryanbalch/IFX-huge-league/resolve/main/assets/profiles/players_pics/{filename}"
+    style="max-width: 100%; max-height: 100%; object-fit: contain; display: block; margin: 0 auto;"
+/>
+"""
+team_image_map = {
+    'everglade-fc': 'Everglade_FC',
+    'fraser-valley-united': 'Fraser_Valley_United',
+    'tierra-alta-fc': 'Tierra_Alta_FC',
+    'yucatan-force': 'Yucatan_Force',
+}
+
 
 class GradioEventHandler(AsyncCallbackHandler):
     """
@@ -33,32 +46,43 @@ class GradioEventHandler(AsyncCallbackHandler):
         )
 
     async def on_chat_model_start(self, *args, **kwargs):
-        self.info_box('[CHAT START]')
-        self.ots_box("""
-        <img 
-            src="https://huggingface.co/spaces/ryanbalch/IFX-huge-league/resolve/main/assets/landing.png"
-            style="max-width: 100%; max-height: 100%; object-fit: contain; display: block; margin: 0 auto;"
-        />
-        """)
+        pass
+        # self.info_box('[CHAT START]')
+        # self.ots_box("""
+        # <img 
+        #     src="https://huggingface.co/spaces/ryanbalch/IFX-huge-league/resolve/main/assets/landing.png"
+        #     style="max-width: 100%; max-height: 100%; object-fit: contain; display: block; margin: 0 auto;"
+        # />
+        # """)
 
     async def on_llm_new_token(self, token: str, **kwargs):
         if token:
             self.queue.put(token) 
 
     async def on_llm_end(self, result: LLMResult, *args, **kwargs):
-        if self.is_chat_stream_end(result):
-            self.queue.put(None)
+        pass
+        # if self.is_chat_stream_end(result):
+        #     self.queue.put(None)
 
     async def on_tool_end(self, output: any, **kwargs):
-        print(f"\n{Fore.CYAN}[TOOL RESULT] {output}{Style.RESET_ALL}")
+        print(f"\n{Fore.CYAN}[TOOL END] {output}{Style.RESET_ALL}")
+        for doc in output:
+            if True:#doc.metadata.get("show_profile_card"):
+                img = image_base.format(filename=self.get_image_filename(doc))
+                print(f"\n{Fore.YELLOW}[TOOL END] {img}{Style.RESET_ALL}")
+                self.ots_box(img)
+                break
+            # else:
+                # self.info_box(doc)
 
     async def on_tool_start(self, input: any, *args, **kwargs):
-        self.info_box(f"[TOOL START]")
+        self.info_box(input.get("name", "[TOOL START]"))
 
     async def on_workflow_end(self, state, *args, **kwargs):
         print(f"\n{Fore.CYAN}[WORKFLOW END]{Style.RESET_ALL}")
-        for msg in state["messages"]:
-            print(f'{Fore.YELLOW}{msg.content}{Style.RESET_ALL}')
+        self.queue.put(None)
+        # for msg in state["messages"]:
+        #     print(f'{Fore.YELLOW}{msg.content}{Style.RESET_ALL}')
 
     @staticmethod
     def is_chat_stream_end(result: LLMResult) -> bool:
@@ -67,3 +91,7 @@ class GradioEventHandler(AsyncCallbackHandler):
             return bool(content and content.strip())
         except (IndexError, AttributeError):
             return False
+
+    @staticmethod
+    def get_image_filename(doc):
+        return f'{team_image_map.get(doc.metadata.get("team"))}_{doc.metadata.get("number")}.png'

api/event_handlers/print_handler.py

@@ -4,6 +4,19 @@ from langchain_core.outputs.llm_result import LLMResult
 from typing import List
 from langchain_core.messages import BaseMessage
 
+image_base = """
+<img 
+    src="https://huggingface.co/spaces/ryanbalch/IFX-huge-league/resolve/main/assets/profiles/{filename}"
+    style="max-width: 100%; max-height: 100%; object-fit: contain; display: block; margin: 0 auto;"
+/>
+"""
+team_image_map = {
+    'everglade-fc': 'Everglade_FC',
+    'fraser-valley-united': 'Fraser_Valley_United',
+    'tierra-alta-fc': 'Tierra_Alta_FC',
+    'yucatan-force': 'Yucatan_Force',
+}
+
 class PrintEventHandler(AsyncCallbackHandler):
     """
     Example async event handler: prints streaming tokens and tool results.
@@ -26,7 +39,12 @@ class PrintEventHandler(AsyncCallbackHandler):
             print('\n[END]')
 
     async def on_tool_end(self, output: any, **kwargs):
-        print(f"\n{Fore.CYAN}[TOOL RESULT] {output}{Style.RESET_ALL}")
+        for doc in output:
+            if doc.metadata.get("show_profile_card"):
+                img = image_base.format(filename=self.get_image_filename(doc))
+                print(f"\n{Fore.CYAN}[TOOL RESULT] {img}{Style.RESET_ALL}")
+            else:
+                print(f"\n{Fore.CYAN}[TOOL RESULT] {doc}{Style.RESET_ALL}")
 
     async def on_tool_start(self, input: any, *args, **kwargs):
         print(f"\n{Fore.CYAN}[TOOL START]{Style.RESET_ALL}")
@@ -42,6 +60,10 @@ class PrintEventHandler(AsyncCallbackHandler):
         except (IndexError, AttributeError):
             return False
 
+    @staticmethod
+    def get_image_filename(doc):
+        return f'{team_image_map.get(doc.metadata.get("team"))}_{doc.metadata.get("number")}.png'
+
     # def __getattribute__(self, name):
     #     attr = super().__getattribute__(name)
     #     if callable(attr) and name.startswith("on_"):

api/scripts/workflow_playground.py

@@ -36,8 +36,9 @@ workflow_bundle, state = build_workflow_with_state(
     last_name="Bigly",
     persona="Casual Fan",
     messages=[
-        HumanMessage(content="tell me about some players in everglade fc"),
+        # HumanMessage(content="tell me about some players in everglade fc"),
         # HumanMessage(content="tell me about the league")
+        HumanMessage(content="tell me about Ryan Martinez of everglade fc")
     ],
 )
 

api/server_gradio.py

@@ -114,13 +114,13 @@ def submit_helper(state, handler, user_query):
                 gr.Info(token["message"])
                 continue
             if token["type"] == "ots":
+                print('OTS: ' + token["message"])
                 state.ots_content = ots_default.format(content=token["message"])
                 state = AppState(**state.model_dump())
-                yield state, result
                 continue
         result += token
         yield state, result
-
+    
     state.history.append(AIMessage(content=result))
 
 ### Interface ###
@@ -210,12 +210,14 @@ with gr.Blocks() as demo:
 
     @submit_btn.click(inputs=[state, user_query], outputs=[state, llm_response])
     def submit(state, user_query):
-        user_query = user_query or "tell me about some players in everglade fc"
+        # user_query = user_query or "tell me about some players in everglade fc"
+        user_query = user_query or "tell me about Ryan Martinez of everglade fc"
         yield from submit_helper(state, handler, user_query)
         
     @user_query.submit(inputs=[state, user_query], outputs=[state, llm_response])
     def user_query_change(state, user_query):
-        user_query = user_query or "tell me about some players in everglade fc"
+        # user_query = user_query or "tell me about some players in everglade fc"
+        user_query = user_query or "tell me about Ryan Martinez of everglade fc"
         yield from submit_helper(state, handler, user_query)
 
     @persona.change(inputs=[persona, state], outputs=[persona_disp])

api/tools/player_search.py

@@ -31,6 +31,10 @@ class PlayerSearchSchema(BaseModel):
         "  • Everglade FC (Miami, USA): Flashy, wild, South Florida flair.\n"
         "  • Fraser Valley United (Abbotsford, Canada): Vineyard roots, top youth academy."
     ))
+    show_profile_card: bool = Field(description=(
+        "If true, only the best-matching player will be returned. The UI will display a player profile card for this result, in addition to the LLM's response."
+        "The LLM should use this flag when the user expects a single, specific player and a card UI."
+    ))
 
 
 class PlayerSearchTool(BaseTool):
@@ -43,22 +47,34 @@ class PlayerSearchTool(BaseTool):
     
     def _run(self,
              query: str,
+             show_profile_card: bool = False,
              run_manager: Optional[CallbackManagerForToolRun] = None,
              ) -> List[Document]:
         k = 5 if query[0] == "*" else 3
-        return vector_store.similarity_search(
+        if show_profile_card:
+            k = 1
+        results = vector_store.similarity_search(
             query,
             k=k,
             filter=lambda doc: doc.metadata.get("type") == "player",
         )
+        for result in results:
+            result.metadata["show_profile_card"] = show_profile_card
+        return results
     
     async def _arun(self,
                     query: str,
+                    show_profile_card: bool = False,
                     run_manager: Optional[AsyncCallbackManagerForToolRun] = None,
                     ) -> List[Document]:
         k = 5 if query[0] == "*" else 3
-        return await vector_store.asimilarity_search(
+        if show_profile_card:
+            k = 1
+        results = await vector_store.asimilarity_search(
             query,
             k=k,
             filter=lambda doc: doc.metadata.get("type") == "player",
         )
+        for result in results:
+            result.metadata["show_profile_card"] = show_profile_card
+        return results