Upload folder using huggingface_hub

apps/app.py

@@ -53,8 +53,8 @@ class FantasyDraftApp:
         self.current_draft = None
         self.draft_output = ""
         
-        # Use basic multiagent draft
-        draft_generator = run_interactive_mock_draft()
+        # Use basic multiagent draft with custom prompts
+        draft_generator = run_interactive_mock_draft(custom_prompts=self.custom_prompts)
         
         for output in draft_generator:
             if isinstance(output, tuple):

apps/multiagent_draft.py

@@ -21,12 +21,19 @@ USE_ENHANCED = False
 class DraftAgent:
     """Base class for draft agents with specific strategies."""
     
-    def __init__(self, team_name: str, strategy: str, color: str, icon: str):
+    def __init__(self, team_name: str, strategy: str, color: str, icon: str, custom_instructions: Optional[str] = None):
         self.team_name = team_name
         self.strategy = strategy
         self.color = color
         self.icon = icon
-        self.agent = FantasyDraftAgent()
+        self.custom_instructions = custom_instructions
+        
+        # Create agent with custom instructions if provided
+        if custom_instructions:
+            self.agent = FantasyDraftAgent(custom_instructions=custom_instructions)
+        else:
+            self.agent = FantasyDraftAgent()
+            
         self.picks = []
         self.conversation_memory = []
         
@@ -90,8 +97,8 @@ Don't be polite - show confidence and give as good as you get."""
 class ZeroRBAgent(DraftAgent):
     """Agent that follows Zero RB strategy."""
     
-    def __init__(self, team_name: str):
-        super().__init__(team_name, "Zero RB Strategy", "#E3F2FD", "📘")
+    def __init__(self, team_name: str, custom_instructions: Optional[str] = None):
+        super().__init__(team_name, "Zero RB Strategy", "#E3F2FD", "📘", custom_instructions)
         self.person_emoji = "🤓"  # Analytical nerd
     
     def make_pick(self, available_players: List[str], draft_board: Dict) -> Tuple[str, str]:
@@ -168,8 +175,8 @@ Use terms like "value", "steal", "while others reached" - not raw numbers."""
 class BPAAgent(DraftAgent):
     """Agent that follows Best Player Available strategy."""
     
-    def __init__(self, team_name: str):
-        super().__init__(team_name, "Best Player Available", "#E8F5E9", "📗")
+    def __init__(self, team_name: str, custom_instructions: Optional[str] = None):
+        super().__init__(team_name, "Best Player Available", "#E8F5E9", "📗", custom_instructions)
         self.person_emoji = "🧑‍💼"  # Business-like, calculated
     
     def make_pick(self, available_players: List[str], draft_board: Dict) -> Tuple[str, str]:
@@ -215,8 +222,8 @@ Don't use raw ADP numbers - use terms like "best available", "top-ranked", "obvi
 class RobustRBAgent(DraftAgent):
     """Agent that follows Robust RB strategy."""
     
-    def __init__(self, team_name: str):
-        super().__init__(team_name, "Robust RB Strategy", "#FFF3E0", "📙")
+    def __init__(self, team_name: str, custom_instructions: Optional[str] = None):
+        super().__init__(team_name, "Robust RB Strategy", "#FFF3E0", "📙", custom_instructions)
         self.person_emoji = "🧔"  # Old-school, traditional
     
     def make_pick(self, available_players: List[str], draft_board: Dict) -> Tuple[str, str]:
@@ -291,8 +298,8 @@ Focus on your "foundation" and "championship formula" - avoid raw rankings."""
 class UpsideAgent(DraftAgent):
     """Agent that hunts for upside/breakout players."""
     
-    def __init__(self, team_name: str):
-        super().__init__(team_name, "Upside Hunter", "#FFFDE7", "📓")
+    def __init__(self, team_name: str, custom_instructions: Optional[str] = None):
+        super().__init__(team_name, "Upside Hunter", "#FFFDE7", "📓", custom_instructions)
         self.person_emoji = "🤠"  # Risk-taking cowboy
     
     def make_pick(self, available_players: List[str], draft_board: Dict) -> Tuple[str, str]:
@@ -358,8 +365,8 @@ Use exciting terms like "breakout", "league-winner", "explosive" - not rankings.
 class UserAdvisorAgent(DraftAgent):
     """Agent that advises the user during their picks."""
     
-    def __init__(self):
-        super().__init__("Your Advisor", "Strategic Advisor", "#FFEBEE", "📕")
+    def __init__(self, custom_instructions: Optional[str] = None):
+        super().__init__("Your Advisor", "Strategic Advisor", "#FFEBEE", "📕", custom_instructions)
         self.person_emoji = "🧙"  # Wise advisor
         self.user_picks = []
     
@@ -453,18 +460,20 @@ class CommissionerAgent:
 class MultiAgentMockDraft:
     """Orchestrates the multi-agent mock draft."""
     
-    def __init__(self, user_pick_position: int = 4):
-        # Initialize agents
+    def __init__(self, user_pick_position: int = 4, custom_prompts: Optional[Dict[int, str]] = None):
+        # Initialize agents with custom prompts if provided
+        self.custom_prompts = custom_prompts or {}
+        
         self.agents = {
-            1: ZeroRBAgent("Team 1"),
-            2: BPAAgent("Team 2"), 
-            3: RobustRBAgent("Team 3"),
-            5: UpsideAgent("Team 5")
+            1: ZeroRBAgent("Team 1", self.custom_prompts.get(1)),
+            2: BPAAgent("Team 2", self.custom_prompts.get(2)), 
+            3: RobustRBAgent("Team 3", self.custom_prompts.get(3)),
+            5: UpsideAgent("Team 5", self.custom_prompts.get(5))
         }
         
         # Add Team 6 as BPA if it's not the user position
         if 6 != user_pick_position:
-            self.agents[6] = BPAAgent("Team 6")
+            self.agents[6] = BPAAgent("Team 6", self.custom_prompts.get(6))
             self.agents[6].person_emoji = "👨‍🏫"  # Professor, methodical
         
         self.user_position = user_pick_position

apps/multiagent_scenarios.py

@@ -151,13 +151,13 @@ def create_mock_draft_visualization(draft: MultiAgentMockDraft,
     return output
 
 
-def run_interactive_mock_draft():
+def run_interactive_mock_draft(custom_prompts=None):
     """Run an interactive mock draft demo that yields formatted output."""
     
     try:
         # Initialize the draft
         print("Initializing MultiAgentMockDraft...")
-        draft = MultiAgentMockDraft(user_pick_position=4)
+        draft = MultiAgentMockDraft(user_pick_position=4, custom_prompts=custom_prompts)
         print("Draft initialized successfully")
         
         # Skip introductions and go straight to commissioner welcome

core/agent.py

@@ -24,10 +24,11 @@ if not os.getenv("OPENAI_API_KEY"):
 
 
 class FantasyDraftAgent:
-    def __init__(self, framework: str = "tinyagent", model_id: str = "gpt-4o-mini"):
+    def __init__(self, framework: str = "tinyagent", model_id: str = "gpt-4o-mini", custom_instructions: Optional[str] = None):
         """Initialize the Fantasy Draft Agent."""
         self.framework = framework
         self.model_id = model_id
+        self.custom_instructions = custom_instructions
         
         # Draft state management
         self.draft_state = {
@@ -89,6 +90,10 @@ class FantasyDraftAgent:
     
     def _get_instructions(self) -> str:
         """Get the agent's system instructions."""
+        # Use custom instructions if provided, otherwise use default
+        if self.custom_instructions:
+            return self.custom_instructions
+            
         return """You are an expert fantasy football draft assistant with deep knowledge of:
         - Player values, tiers, and ADP (Average Draft Position)
         - Draft strategy (Zero RB, Hero RB, Robust RB, etc.)