Update main.py

main.py

@@ -812,50 +812,66 @@ def _scrape_dashboard_info_brscraper():
         logging.error(f"Error scraping dashboard info with BRScraper: {e}")
     return dashboard_data
 
+
 PERP_KEY = os.getenv("PERPLEXITY_API_KEY")
 PERP_URL = "https://api.perplexity.ai/chat/completions"
 
 NBA_ANALYST_SYSTEM_PROMPT = (
     "You are a sharp, insightful NBA analyst AI with the tone and knowledge of a seasoned sports commentator. "
-    "Your expertise spans the entire history of the league—from hardwood legends to rising stars, tactical evolutions, "
+    "Your expertise spans the entire history of basketball—from hardwood legends to rising stars, tactical evolutions, "
     "advanced stats, trades, rivalries, and playoff lore. Speak with authority, depth, and the occasional flair of broadcast commentary. "
-    "Your job is to help users explore the NBA with analytical rigor and passion. You draw on statistics, game film analysis, "
+    "Your job is to help users explore basketball with analytical rigor and passion. You draw on statistics, game film analysis, "
     "player tendencies, team dynamics, historical context, and front-office strategy. You may reference key metrics like PER, TS%, "
     "on-off splits, or synergy data when relevant. You provide takes that are well-reasoned, never vague, and always rooted in basketball-specific insight. "
+    "CRITICAL: Stay strictly within basketball. When discussing 'rookie performances' or any basketball topic, ONLY reference NBA/basketball players and stats - never NFL or other sports. "
     "Do not respond to questions outside the world of basketball. If asked, steer the conversation back to the NBA with finesse, "
     "perhaps by connecting a topic metaphorically to hoops. Your personality is that of a knowledgeable but approachable analyst—"
     "a cross between a basketball scout, play-by-play commentator, and sportswriter. You love the game, and it shows."
 )
 
-def ask_perp(prompt, system=NBA_ANALYST_SYSTEM_PROMPT , max_tokens=1000, temp=0.2):
+def ask_perp(prompt, system=NBA_ANALYST_SYSTEM_PROMPT, max_tokens=1000, temp=0.2):
     if not PERP_KEY:
         logging.error("PERPLEXITY_API_KEY env var not set.")
         return "Perplexity API key is not configured."
-    hdr = {'Authorization':f'Bearer {PERP_KEY}','Content-Type':'application/json'}
+
+    headers = {
+        'Authorization': f'Bearer {PERP_KEY}',
+        'Content-Type': 'application/json'
+    }
+
     payload = {
-        "model":"sonar-pro",
-        "messages":[{"role":"system","content":system},{"role":"user","content":prompt}],
-        "max_tokens":max_tokens, "temperature":temp, "search_context_size": "high" 
+        "model": "sonar-pro",
+        "messages": [
+            {"role": "system", "content": system},
+            {"role": "user", "content": f"BASKETBALL ONLY: {prompt}"}
+        ],
+        "max_tokens": max_tokens,
+        "temperature": temp,
+        "web_search_options": {
+            "search_context_size": "high",
+            "search_domain_filter": ["nba.com", "espn.com", "basketball-reference.com"]
+        },
+        "emit_sources": True
     }
+
     try:
-        r = requests.post(PERP_URL, json=payload, headers=hdr, timeout=45)
-        r.raise_for_status()
-        return r.json().get("choices", [])[0].get("message", {}).get("content", "")
+        response = requests.post(PERP_URL, json=payload, headers=headers, timeout=45)
+        response.raise_for_status()
+        return response.json().get("choices", [])[0].get("message", {}).get("content", "")
     except requests.exceptions.RequestException as e:
         error_message = f"Error communicating with Perplexity API: {e}"
         if hasattr(e, 'response') and e.response is not None:
             try:
                 error_detail = e.response.json().get("error", {}).get("message", e.response.text)
-                error_message = f"Perplexity API error: {r.status_code} - {r.reason}"
+                error_message = f"Perplexity API error: {e.response.status_code} - {e.response.reason}"
             except ValueError:
-                error_message = f"Perplexity API error: {r.status_code} - {r.reason}"
+                error_message = f"Perplexity API error: {e.response.status_code} - {e.response.reason}"
         logging.error(f"Perplexity API request failed: {error_message}")
         return f"Error from AI: {error_message}"
     except Exception as e:
         logging.error(f"An unexpected error occurred with Perplexity API: {e}")
         return f"An unexpected error occurred with AI: {str(e)}"
 
-
 @app.route('/api/nba/players', methods=['GET'])
 @cross_origin()
 def get_players():