Update main.py

main.py

@@ -15,9 +15,8 @@ from firebase_admin import credentials, db, storage, auth
 import firebase_admin
 import logging
 import traceback
-from bs4 import BeautifulSoup, Comment # BeautifulSoup is still imported but not used for team stats now.
+from bs4 import BeautifulSoup, Comment
 
-# Import BRScraper
 try:
     from BRScraper import nba
     BRSCRAPER_AVAILABLE = True
@@ -25,18 +24,14 @@ except ImportError:
     BRSCRAPER_AVAILABLE = False
     logging.error("BRScraper not found. Please install with: `pip install BRScraper`")
 
-# Initialize Flask app and CORS
 app = Flask(__name__)
 CORS(app)
 
-# Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-# Firebase initialization
 Firebase_DB = os.getenv("Firebase_DB")
 Firebase_Storage = os.getenv("Firebase_Storage")
 
-# Global flag for Firebase initialization status
 FIREBASE_INITIALIZED = False
 
 try:
@@ -56,12 +51,9 @@ except Exception as e:
     logging.error(f"Error initializing Firebase: {e}")
     traceback.print_exc()
 
-# Only initialize bucket if Firebase is initialized
 bucket = storage.bucket() if FIREBASE_INITIALIZED else None
 
-# Helper functions
 def verify_token(token):
-    """Verifies Firebase ID token and returns UID."""
     try:
         decoded_token = auth.verify_id_token(token)
         return decoded_token['uid']
@@ -70,7 +62,6 @@ def verify_token(token):
         return None
 
 def verify_admin(auth_header):
-    """Verifies admin privileges based on Firebase Realtime DB 'is_admin' flag."""
     if not auth_header or not auth_header.startswith('Bearer '):
         raise ValueError('Invalid token format')
     token = auth_header.split(' ')[1]
@@ -83,12 +74,7 @@ def verify_admin(auth_header):
         raise PermissionError('Admin access required')
     return uid
 
-# Decorator for credit deduction
 def credit_required(cost=1):
-    """
-    Decorator to check user authentication, suspension status, and deduct credits.
-    If cost is 0, it only checks authentication and suspension.
-    """
     def decorator(f):
         def wrapper(*args, **kwargs):
             auth_header = request.headers.get('Authorization', '')
@@ -112,7 +98,6 @@ def credit_required(cost=1):
                 return jsonify({'error': f'Insufficient credits. You need {cost} credits, but have {current_credits}.'}), 403
 
             try:
-                # Deduct credits only if cost > 0
                 if cost > 0:
                     user_ref.update({'credits': current_credits - cost})
                     logging.info(f"Deducted {cost} credits from user {uid}. New balance: {current_credits - cost}")
@@ -120,16 +105,13 @@ def credit_required(cost=1):
             except Exception as e:
                 logging.error(f"Failed to process credits for user {uid}: {e}")
                 return jsonify({'error': 'Failed to process credits. Please try again.'}), 500
-        wrapper.__name__ = f.__name__ # Preserve original function name for Flask routing
+        wrapper.__name__ = f.__name__
         return wrapper
     return decorator
 
 
-# ---------- Authentication Endpoints ----------
-
 @app.route('/api/auth/signup', methods=['POST'])
 def signup():
-    """Handles user signup and creates a new user entry in Firebase Auth and DB."""
     try:
         data = request.get_json()
         email = data.get('email')
@@ -141,7 +123,7 @@ def signup():
         user_ref = db.reference(f'users/{user.uid}')
         user_data = {
             'email': email,
-            'credits': 10, # Initial credits for new users
+            'credits': 10,
             'is_admin': False,
             'created_at': datetime.utcnow().isoformat()
         }
@@ -159,7 +141,6 @@ def signup():
 
 @app.route('/api/user/profile', methods=['GET'])
 def get_user_profile():
-    """Retrieves the authenticated user's profile information."""
     try:
         auth_header = request.headers.get('Authorization', '')
         if not auth_header.startswith('Bearer '):
@@ -186,7 +167,6 @@ def get_user_profile():
 
 @app.route('/api/auth/google-signin', methods=['POST'])
 def google_signin():
-    """Handles Google Sign-In, creating user DB entry if it doesn't exist."""
     try:
         auth_header = request.headers.get('Authorization', '')
         if not auth_header.startswith('Bearer '):
@@ -203,7 +183,7 @@ def google_signin():
         if not user_data:
             user_data = {
                 'email': email,
-                'credits': 10,  # Initial credits for new users
+                'credits': 10,
                 'is_admin': False,
                 'created_at': datetime.utcnow().isoformat(),
             }
@@ -222,13 +202,12 @@ def google_signin():
         return jsonify({'error': str(e)}), 400 
 
 @app.route('/api/user/request-credits', methods=['POST'])
-@credit_required(cost=0) # This endpoint doesn't cost credits, but requires auth
+@credit_required(cost=0)
 def request_credits():
-    """Allows a user to request more credits from an admin."""
     try:
         auth_header = request.headers.get('Authorization', '')
         token = auth_header.split(' ')[1]
-        uid = verify_token(token) # uid is already verified by decorator, but get it here
+        uid = verify_token(token)
         
         data = request.get_json()
         requested_credits = data.get('requested_credits')
@@ -248,44 +227,40 @@ def request_credits():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/api/user/submit_feedback', methods=['POST'])
-@credit_required(cost=0) # Requires authentication but no credit cost
+@credit_required(cost=0)
 @cross_origin()
 def submit_feedback():
-    """Allows a user to submit feedback to administrators."""
     try:
         auth_header = request.headers.get('Authorization', '')
         token = auth_header.split(' ')[1]
-        uid = verify_token(token) # UID already verified by decorator
+        uid = verify_token(token)
 
         data = request.get_json()
-        feedback_type = data.get('type') # e.g., 'bug', 'feature_request', 'general'
+        feedback_type = data.get('type')
         message = data.get('message')
 
         if not feedback_type or not message:
             return jsonify({'error': 'Feedback type and message are required'}), 400
 
         user_data = db.reference(f'users/{uid}').get()
-        user_email = user_data.get('email', 'unknown_email') # Get email for admin view
+        user_email = user_data.get('email', 'unknown_email')
 
         feedback_ref = db.reference('feedback').push()
         feedback_ref.set({
             'user_id': uid,
-            'user_email': user_email, # Store user email for easier admin lookup
+            'user_email': user_email,
             'type': feedback_type,
             'message': message,
             'created_at': datetime.utcnow().isoformat(),
-            'status': 'open' # Initial status for new feedback
+            'status': 'open'
         })
         return jsonify({'success': True, 'feedback_id': feedback_ref.key}), 201
     except Exception as e:
         logging.error(f"Submit feedback error: {e}")
         return jsonify({'error': str(e)}), 500
 
-# ---------- Admin Endpoints ----------
-
 @app.route('/api/admin/profile', methods=['GET'])
 def get_admin_profile():
-    """Retrieves admin profile and aggregated user/credit statistics."""
     try:
         admin_uid = verify_admin(request.headers.get('Authorization', ''))
         admin_data = db.reference(f'users/{admin_uid}').get()
@@ -301,7 +276,6 @@ def get_admin_profile():
         total_current_credits = sum(user.get('credits', 0) for user in all_users_data.values())
         total_normal_current_credits = sum(user.get('credits', 0) for user in normal_users_data)
         
-        # Updated initial credits calculation to 10
         total_initial_credits = total_normal_users * 10 
         credit_usage = total_initial_credits - total_normal_current_credits
 
@@ -325,7 +299,6 @@ def get_admin_profile():
 
 @app.route('/api/admin/credit_requests', methods=['GET'])
 def list_credit_requests():
-    """Lists all credit requests submitted by users."""
     try:
         verify_admin(request.headers.get('Authorization', ''))
         requests_ref = db.reference('credit_requests')
@@ -338,7 +311,6 @@ def list_credit_requests():
 
 @app.route('/api/admin/credit_requests/<string:request_id>', methods=['PUT'])
 def process_credit_request(request_id):
-    """Processes a specific credit request (approves or declines)."""
     try:
         admin_uid = verify_admin(request.headers.get('Authorization', ''))
         req_ref = db.reference(f'credit_requests/{request_id}')
@@ -377,7 +349,6 @@ def process_credit_request(request_id):
 
 @app.route('/api/admin/users', methods=['GET'])
 def admin_list_users():
-    """Lists all users in the system."""
     try:
         verify_admin(request.headers.get('Authorization', ''))
         users_ref = db.reference('users')
@@ -400,7 +371,6 @@ def admin_list_users():
 
 @app.route('/api/admin/users/search', methods=['GET'])
 def admin_search_users():
-    """Searches for users by email."""
     try:
         verify_admin(request.headers.get('Authorization', ''))
         email_query = request.args.get('email', '').lower().strip()
@@ -429,7 +399,6 @@ def admin_search_users():
 
 @app.route('/api/admin/users/<string:uid>/suspend', methods=['PUT'])
 def admin_suspend_user(uid):
-    """Suspends or unsuspends a user account."""
     try:
         verify_admin(request.headers.get('Authorization', ''))
         data = request.get_json()
@@ -454,7 +423,6 @@ def admin_suspend_user(uid):
 
 @app.route('/api/admin/notifications', methods=['POST'])
 def send_notifications():
-    """Sends notifications to users (all, single, or list)."""
     try:
         admin_uid = verify_admin(request.headers.get('Authorization', ''))
         data = request.get_json()
@@ -502,7 +470,6 @@ def send_notifications():
 
 @app.route('/api/admin/feedback', methods=['GET'])
 def admin_view_feedback():
-    """Retrieves user feedback, with optional filtering by type and status."""
     try:
         admin_uid = verify_admin(request.headers.get('Authorization', ''))
         feedback_type = request.args.get('type')
@@ -534,7 +501,6 @@ def admin_view_feedback():
         
 @app.route('/api/admin/users/<string:uid>/credits', methods=['PUT'])
 def admin_update_credits(uid):
-    """Directly adds or subtracts credits from a user's account."""
     try:
         verify_admin(request.headers.get('Authorization', ''))
         data = request.get_json()
@@ -559,57 +525,40 @@ def admin_update_credits(uid):
 # NBA Analytics Hub Data Fetching Utilities
 # ——————————————————————————————————————————————
 
-# Helper to clean DataFrame column names for Firebase keys
 def clean_firebase_keys(key_name):
-    """Cleans a string to be a valid Firebase Realtime Database key."""
     if not isinstance(key_name, str):
         key_name = str(key_name)
-    # Firebase invalid characters: . $ # [ ] /
-    # Replace them with underscores or remove them
     cleaned_key = key_name.replace('.', '_').replace('$', '').replace('#', '').replace('[', '').replace(']', '').replace('/', '_')
-    # Also handle '%' if it's problematic, though not explicitly listed by Firebase for keys
-    cleaned_key = cleaned_key.replace('%', 'Pct') # Example: W/L% -> W_LPct
-    # Ensure no empty keys
+    cleaned_key = cleaned_key.replace('%', 'Pct')
     if not cleaned_key:
-        return "empty_key_" + str(uuid.uuid4())[:8] # Fallback for completely empty keys
+        return "empty_key_" + str(uuid.uuid4())[:8]
     return cleaned_key
 
 def clean_df_for_firebase(df):
-    """Applies Firebase key cleaning to all column names of a DataFrame."""
     if df.empty:
         return df
     df.columns = [clean_firebase_keys(col) for col in df.columns]
     return df
 
 def clean_team_name(team_name):
-    """
-    Clean and standardize team names from Basketball Reference.
-    """
     if pd.isna(team_name):
         return team_name
-    team_name = str(team_name).strip().replace('*', '')
-    team_mapping = {
-        'NOP': 'NO', 'PHX': 'PHO', 'BRK': 'BKN', 'CHA': 'CHO', 'UTA': 'UTH'
-    }
-    return team_mapping.get(team_name, team_name)
+    team_name = str(team_name).strip()
+    team_name = re.sub(r'\s*\(\d+\)$', '', team_name) # Remove playoff seeding suffix
+    team_name = team_name.replace('*', '') # Remove asterisks
+    return team_name
 
 def is_data_stale(timestamp_str, max_age_hours=24):
-    """Checks if a timestamp string indicates data is older than max_age_hours."""
     if not timestamp_str:
-        return True # No timestamp means data is stale or never fetched
+        return True
     try:
         last_updated = datetime.fromisoformat(timestamp_str)
         return (datetime.utcnow() - last_updated) > timedelta(hours=max_age_hours)
     except ValueError:
         logging.error(f"Invalid timestamp format: {timestamp_str}")
-        return True # Treat invalid format as stale
+        return True
 
 def get_team_standings_brscraper(year):
-    """
-    Scrapes the league’s team standings table from Basketball-Reference
-    using BRScraper, with Firebase caching.
-    Returns cleaned DataFrame.
-    """
     if not FIREBASE_INITIALIZED:
         logging.warning("Firebase not initialized. Cannot use caching for team standings. Scraping directly.")
         return _scrape_team_standings_brscraper(year)
@@ -617,7 +566,6 @@ def get_team_standings_brscraper(year):
     db_ref = db.reference(f'scraped_data/team_standings/{year}')
     cached_data = db_ref.get()
 
-    # Cache for 12 hours for standings data
     if cached_data and not is_data_stale(cached_data.get('last_updated'), max_age_hours=12):
         logging.info(f"Loading team standings for {year} from Firebase cache.")
         return pd.DataFrame.from_records(cached_data['data'])
@@ -633,33 +581,29 @@ def get_team_standings_brscraper(year):
         return df
 
 def _scrape_team_standings_brscraper(year):
-    """Internal function to perform the actual scraping of team standings."""
     if not BRSCRAPER_AVAILABLE:
         logging.error("BRScraper not available for team standings.")
         return pd.DataFrame()
     try:
-        df = nba.get_standings(year, info='total', rename=False)
+        df = nba.get_standings(year, info='total') # Removed rename=False
         if df.empty:
             logging.warning(f"Could not find team standings for {year} using BRScraper.")
             return pd.DataFrame()
 
-        # Standardize column names
         column_mapping = {
             'Tm': 'Team', 'W': 'WINS', 'L': 'LOSSES', 'W/L%': 'WIN_LOSS_PCT',
-            'Rk': 'RANK' # Rank might be useful
+            'Rk': 'RANK'
         }
         df = df.rename(columns={old_col: new_col for old_col, new_col in column_mapping.items() if old_col in df.columns})
 
-        # Clean team names
         if 'Team' in df.columns:
+            df['Team'] = df['Team'].astype(str) # Ensure string type
             df['Team'] = df['Team'].apply(clean_team_name)
 
-        # Convert numeric columns, coercing errors to NaN
         numeric_cols = [col for col in df.columns if col not in ['Team']]
         for col in numeric_cols:
             df[col] = pd.to_numeric(df[col], errors="coerce")
 
-        # Replace NaN values with None for JSON compliance
         df = df.replace({np.nan: None})
         return df
     except Exception as e:
@@ -670,7 +614,7 @@ def get_available_seasons_util(num_seasons=6):
     current_year = datetime.now().year
     current_month = datetime.now().month
     latest_season_end_year = current_year
-    if current_month >= 7: # NBA season typically ends in June, so July onwards means next season's data
+    if current_month >= 7:
         latest_season_end_year += 1
     seasons_list = []
     for i in range(num_seasons):
@@ -705,7 +649,6 @@ def get_player_index_brscraper():
         return df
 
 def _scrape_player_index_brscraper():
-    """Internal function to perform the actual scraping of player index."""
     try:
         latest_season_end_year = int(get_available_seasons_util(1)[0].split('–')[1])
         df = nba.get_stats(latest_season_end_year, info='per_game', rename=False)
@@ -725,9 +668,6 @@ def _scrape_player_index_brscraper():
         return pd.DataFrame({'name': common_players})
 
 def get_player_career_stats_brscraper(player_name, seasons_to_check=10, playoffs=False):
-    """
-    Fetches career per-game or playoff stats for a given player using BRScraper.
-    """
     if not BRSCRAPER_AVAILABLE:
         return pd.DataFrame()
     all_rows = []
@@ -753,7 +693,7 @@ def get_player_career_stats_brscraper(player_name, seasons_to_check=10, playoffs
         'G':'GP','GS':'GS','MP':'MIN', 'FG%':'FG_PCT','3P%':'FG3_PCT','FT%':'FT_PCT',
         'TRB':'REB','AST':'AST','STL':'STL','BLK':'BLK','TOV':'TO',
         'PF':'PF','PTS':'PTS','ORB':'OREB','DRB':'DREB',
-        'FG':'FGM','FGA':'FGA','3P':'FG3M','3PA':'FG3A', # Corrected 3PA mapping
+        'FG':'FGM','FGA':'FGA','3P':'FG3M','3PA':'FG3A',
         '2P':'FGM2','2PA':'FGA2','2P%':'FG2_PCT','eFG%':'EFG_PCT',
         'FT':'FTM','FTA':'FTA'
     }
@@ -765,14 +705,10 @@ def get_player_career_stats_brscraper(player_name, seasons_to_check=10, playoffs
             df[col] = pd.to_numeric(df[col], errors='coerce')
 
     df['Player'] = player_name
-    df = df.replace({np.nan: None}) # Ensure NaN are None for JSON compliance
+    df = df.replace({np.nan: None})
     return df
 
 def get_dashboard_info_brscraper():
-    """
-    Fetches various dashboard-level information using BRScraper with caching.
-    Includes MVP votings and playoff probabilities.
-    """
     if not BRSCRAPER_AVAILABLE:
         logging.error("BRScraper not available for dashboard info.")
         return {}
@@ -784,7 +720,6 @@ def get_dashboard_info_brscraper():
     db_ref = db.reference('scraped_data/dashboard_info')
     cached_data = db_ref.get()
 
-    # Cache for 24 hours for dashboard info
     if cached_data and not is_data_stale(cached_data.get('last_updated'), max_age_hours=24):
         logging.info("Loading dashboard info from Firebase cache.")
         return cached_data['data']
@@ -800,31 +735,27 @@ def get_dashboard_info_brscraper():
         return data
 
 def _scrape_dashboard_info_brscraper():
-    """Internal function to perform the actual scraping for dashboard info."""
     dashboard_data = {}
     try:
-        # MVP Votings for 2025
         mvp_2025_df = nba.get_award_votings('mvp', 2025)
         if not mvp_2025_df.empty:
-            mvp_2025_df = clean_df_for_firebase(mvp_2025_df) # Clean column names
+            mvp_2025_df = clean_df_for_firebase(mvp_2025_df)
             dashboard_data['mvp_2025_votings'] = mvp_2025_df.replace({np.nan: None}).to_dict(orient='records')
         else:
             dashboard_data['mvp_2025_votings'] = []
             logging.warning("Could not retrieve 2025 MVP votings.")
 
-        # Current Season Playoff Probabilities (East)
         east_probs_df = nba.get_playoffs_probs('east')
         if not east_probs_df.empty:
-            east_probs_df = clean_df_for_firebase(east_probs_df) # Clean column names
+            east_probs_df = clean_df_for_firebase(east_probs_df)
             dashboard_data['playoff_probs_east'] = east_probs_df.replace({np.nan: None}).to_dict(orient='records')
         else:
             dashboard_data['playoff_probs_east'] = []
             logging.warning("Could not retrieve Eastern Conference playoff probabilities.")
 
-        # Current Season Playoff Probabilities (West)
         west_probs_df = nba.get_playoffs_probs('west')
         if not west_probs_df.empty:
-            west_probs_df = clean_df_for_firebase(west_probs_df) # Clean column names
+            west_probs_df = clean_df_for_firebase(west_probs_df)
             dashboard_data['playoff_probs_west'] = west_probs_df.replace({np.nan: None}).to_dict(orient='records')
         else:
             dashboard_data['playoff_probs_west'] = []
@@ -832,21 +763,18 @@ def _scrape_dashboard_info_brscraper():
 
     except Exception as e:
         logging.error(f"Error scraping dashboard info with BRScraper: {e}")
-        # Return partial data if some parts fail
     return dashboard_data
 
-# Perplexity API
 PERP_KEY = os.getenv("PERPLEXITY_API_KEY")
 PERP_URL = "https://api.perplexity.ai/chat/completions"
 
 def ask_perp(prompt, system="You are a helpful NBA analyst AI.", max_tokens=500, temp=0.2):
-    """Sends a prompt to the Perplexity AI API and returns the response."""
     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'}
     payload = {
-        "model":"sonar-pro", # do not change
+        "model":"sonar-pro",
         "messages":[{"role":"system","content":system},{"role":"user","content":prompt}],
         "max_tokens":max_tokens, "temperature":temp
     }
@@ -869,12 +797,9 @@ def ask_perp(prompt, system="You are a helpful NBA analyst AI.", max_tokens=500,
         return f"An unexpected error occurred with AI: {str(e)}"
 
 
-# ---------- NBA Analytics Hub Endpoints ----------
-
 @app.route('/api/nba/players', methods=['GET'])
 @cross_origin()
 def get_players():
-    """Returns a list of available NBA player names."""
     try:
         players_df = get_player_index_brscraper()
         if players_df.empty:
@@ -887,7 +812,6 @@ def get_players():
 @app.route('/api/nba/seasons', methods=['GET'])
 @cross_origin()
 def get_seasons():
-    """Returns a list of available NBA seasons."""
     try:
         seasons_list = get_available_seasons_util()
         return jsonify({'seasons': seasons_list})
@@ -898,7 +822,6 @@ def get_seasons():
 @app.route('/api/nba/player_stats', methods=['POST'])
 @cross_origin()
 def get_player_stats():
-    """Retrieves per-game statistics for selected NBA players and seasons (regular season)."""
     try:
         data = request.get_json()
         selected_players = data.get('players')
@@ -911,7 +834,6 @@ def get_player_stats():
         players_with_no_data = []
 
         for player_name in selected_players:
-            # Use the general helper for regular season stats (playoffs=False by default)
             df_player_career = get_player_career_stats_brscraper(player_name, playoffs=False)
             if not df_player_career.empty:
                 filtered_df = df_player_career[df_player_career['Season'].isin(selected_seasons)].copy()
@@ -930,7 +852,6 @@ def get_player_stats():
 
         comparison_df_raw = pd.concat(all_player_season_data, ignore_index=True)
 
-        # Calculate basic stats
         if len(selected_seasons) > 1:
             basic_display_df = comparison_df_raw.groupby('Player').mean(numeric_only=True).reset_index()
         else:
@@ -939,7 +860,6 @@ def get_player_stats():
         basic_cols = ['Player', 'Season', 'GP', 'MIN', 'PTS', 'REB', 'AST', 'STL', 'BLK', 'FG_PCT', 'FT_PCT', 'FG3_PCT']
         basic_display_df = basic_display_df[[c for c in basic_cols if c in basic_display_df.columns]].round(2)
 
-        # Calculate advanced stats
         advanced_df = comparison_df_raw.copy()
         advanced_df['FGA'] = pd.to_numeric(advanced_df.get('FGA', 0), errors='coerce').fillna(0)
         advanced_df['FTA'] = pd.to_numeric(advanced_df.get('FTA', 0), errors='coerce').fillna(0)
@@ -968,7 +888,6 @@ def get_player_stats():
 @app.route('/api/nba/player_playoff_stats', methods=['POST'])
 @cross_origin()
 def get_player_playoff_stats():
-    """Retrieves per-game playoff statistics for selected NBA players and seasons."""
     try:
         data = request.get_json()
         selected_players = data.get('players')
@@ -981,7 +900,6 @@ def get_player_playoff_stats():
         players_with_no_data = []
 
         for player_name in selected_players:
-            # Use the general helper for playoff stats (playoffs=True)
             df_player_career = get_player_career_stats_brscraper(player_name, playoffs=True)
             if not df_player_career.empty:
                 filtered_df = df_player_career[df_player_career['Season'].isin(selected_seasons)].copy()
@@ -1000,7 +918,6 @@ def get_player_playoff_stats():
 
         comparison_df_raw = pd.concat(all_player_season_data, ignore_index=True)
 
-        # Calculate basic stats
         if len(selected_seasons) > 1:
             basic_display_df = comparison_df_raw.groupby('Player').mean(numeric_only=True).reset_index()
         else:
@@ -1009,7 +926,6 @@ def get_player_playoff_stats():
         basic_cols = ['Player', 'Season', 'GP', 'MIN', 'PTS', 'REB', 'AST', 'STL', 'BLK', 'FG_PCT', 'FT_PCT', 'FG3_PCT']
         basic_display_df = basic_display_df[[c for c in basic_cols if c in basic_display_df.columns]].round(2)
 
-        # Calculate advanced stats
         advanced_df = comparison_df_raw.copy()
         advanced_df['FGA'] = pd.to_numeric(advanced_df.get('FGA', 0), errors='coerce').fillna(0)
         advanced_df['FTA'] = pd.to_numeric(advanced_df.get('FTA', 0), errors='coerce').fillna(0)
@@ -1035,44 +951,54 @@ def get_player_playoff_stats():
         logging.error(f"Error in /api/nba/player_playoff_stats: {e}")
         return jsonify({'error': str(e)}), 500
 
-# In your Flask script (app.py or main.py on Hugging Face)
-
 @app.route('/api/nba/team_stats', methods=['POST'])
 @cross_origin()
 def get_team_stats():
-    # --- ADD THIS LINE AT THE VERY TOP OF THE FUNCTION ---
     logging.info("DEBUG: Request successfully entered get_team_stats function!")
-    # --- END ADDITION ---
     try:
         data = request.get_json()
-        selected_teams = data.get('teams')
+        selected_teams_abbrs = data.get('teams')
         selected_season_str = data.get('season')
 
-        if not selected_teams or not selected_season_str:
-            # This would return a 400, not a 404
+        if not selected_teams_abbrs or not selected_season_str:
             return jsonify({'error': 'Teams and season are required'}), 400
 
         year_for_team_stats = int(selected_season_str.split('–')[1])
         tm_df = get_team_standings_brscraper(year_for_team_stats)
 
         if tm_df.empty:
-            # This would return a 404 with a specific message
             return jsonify({'error': f'No team data available for {selected_season_str}'}), 404
 
+        # Map input abbreviations to full names for lookup
+        full_team_names_map = {
+            "ATL": "Atlanta Hawks", "BOS": "Boston Celtics", "BRK": "Brooklyn Nets",
+            "CHO": "Charlotte Hornets", "CHI": "Chicago Bulls", "CLE": "Cleveland Cavaliers",
+            "DAL": "Dallas Mavericks", "DEN": "Denver Nuggets", "DET": "Detroit Pistons",
+            "GSW": "Golden State Warriors", "HOU": "Houston Rockets", "IND": "Indiana Pacers",
+            "LAC": "Los Angeles Clippers", "LAL": "Los Angeles Lakers", "MEM": "Memphis Grizzlies",
+            "MIA": "Miami Heat", "MIL": "Milwaukee Bucks", "MIN": "Minnesota Timberwolves",
+            "NOP": "New Orleans Pelicans", "NYK": "New York Knicks", "OKC": "Oklahoma City Thunder",
+            "ORL": "Orlando Magic", "PHI": "Philadelphia 76ers", "PHX": "Phoenix Suns",
+            "POR": "Portland Trail Blazers", "SAC": "Sacramento Kings", "SAS": "San Antonio Spurs",
+            "TOR": "Toronto Raptors", "UTA": "Utah Jazz", "WAS": "Washington Wizards"
+        }
+        selected_teams_full_names = [full_team_names_map.get(abbr, abbr) for abbr in selected_teams_abbrs]
+
         stats = []
         teams_with_no_data = []
 
-        for t in selected_teams:
-            df = tm_df[tm_df.Team == t].copy()
-            if not df.empty:
-                df_dict = df.iloc[0].to_dict()
+        for team_full_name_lookup in selected_teams_full_names:
+            df_row = tm_df[tm_df.Team == team_full_name_lookup].copy()
+            if not df_row.empty:
+                df_dict = df_row.iloc[0].to_dict()
                 df_dict['Season'] = selected_season_str
                 stats.append(df_dict)
             else:
-                teams_with_no_data.append(t)
+                # Add original abbreviation to not_found list
+                original_abbr = next((abbr for abbr, name in full_team_names_map.items() if name == team_full_name_lookup), team_full_name_lookup)
+                teams_with_no_data.append(original_abbr)
 
         if not stats:
-            # This would return a 404 with a specific message
             return jsonify({
                 'error': 'No data available for selected teams.',
                 'teams_with_no_data': teams_with_no_data
@@ -1093,13 +1019,9 @@ def get_team_stats():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/api/nba/dashboard_info', methods=['GET'])
-@credit_required(cost=0) # No credit cost, but requires authentication
+@credit_required(cost=0)
 @cross_origin()
 def dashboard_info():
-    """
-    Provides various dashboard-level information including MVP votings
-    and current playoff probabilities for East and West conferences.
-    """
     try:
         dashboard_data = get_dashboard_info_brscraper()
         if not dashboard_data:
@@ -1110,12 +1032,9 @@ def dashboard_info():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/api/nba/perplexity_explain', methods=['POST'])
-@credit_required(cost=1) # Costs 1 credit
+@credit_required(cost=1)
 @cross_origin()
 def perplexity_explain():
-    """
-    Provides an AI explanation for a given prompt and stores it in the database.
-    """
     try:
         data = request.get_json()
         prompt = data.get('prompt')
@@ -1127,15 +1046,13 @@ def perplexity_explain():
         if "Error from AI" in explanation:
             return jsonify({'error': explanation}), 500
         
-        # Get UID from the authenticated request (credit_required decorator ensures it's available)
         auth_header = request.headers.get('Authorization', '')
         token = auth_header.split(' ')[1]
         uid = verify_token(token)
 
-        # Store the analysis in Firebase Realtime Database
         if FIREBASE_INITIALIZED:
             user_analyses_ref = db.reference(f'user_analyses/{uid}')
-            analysis_id = str(uuid.uuid4()) # Generate a unique ID for this analysis
+            analysis_id = str(uuid.uuid4())
             analysis_data = {
                 'prompt': prompt,
                 'explanation': explanation,
@@ -1152,10 +1069,9 @@ def perplexity_explain():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/api/user/analyses', methods=['GET'])
-@credit_required(cost=0) # No credit cost, but requires authentication
+@credit_required(cost=0)
 @cross_origin()
 def get_user_analyses():
-    """Retrieves all stored AI analyses for the authenticated user."""
     try:
         auth_header = request.headers.get('Authorization', '')
         token = auth_header.split(' ')[1]
@@ -1176,7 +1092,6 @@ def get_user_analyses():
                 'created_at': data.get('created_at')
             })
         
-        # Sort by creation date, newest first
         analyses_list.sort(key=lambda x: x.get('created_at', ''), reverse=True)
 
         return jsonify({'analyses': analyses_list})
@@ -1185,10 +1100,9 @@ def get_user_analyses():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/api/user/analyses/<string:analysis_id>', methods=['DELETE'])
-@credit_required(cost=0) # No credit cost, but requires authentication
+@credit_required(cost=0)
 @cross_origin()
 def delete_user_analysis(analysis_id):
-    """Deletes a specific AI analysis for the authenticated user."""
     try:
         auth_header = request.headers.get('Authorization', '')
         token = auth_header.split(' ')[1]
@@ -1212,10 +1126,9 @@ def delete_user_analysis(analysis_id):
 
 
 @app.route('/api/nba/perplexity_chat', methods=['POST'])
-@credit_required(cost=1) # Costs 1 credit
+@credit_required(cost=1)
 @cross_origin()
 def perplexity_chat():
-    """Handles AI chat interactions and stores chat history."""
     try:
         data = request.get_json()
         prompt = data.get('prompt')
@@ -1223,26 +1136,21 @@ def perplexity_chat():
         if not prompt:
             return jsonify({'error': 'Prompt is required'}), 400
         
-        # Get UID from the authenticated request for chat history storage
         auth_header = request.headers.get('Authorization', '')
         token = auth_header.split(' ')[1]
-        uid = verify_token(token) # Already verified by decorator, but need uid here
+        uid = verify_token(token)
 
-        # Get AI response
         response_content = ask_perp(prompt, system="You are an NBA expert analyst AI.")
         if "Error from AI" in response_content:
             return jsonify({'error': response_content}), 500
         
-        # Store chat history in Firebase
         if FIREBASE_INITIALIZED:
             user_chat_ref = db.reference(f'users/{uid}/chat_history')
-            # Push user's message
             user_chat_ref.push({
                 'role': 'user',
                 'content': prompt,
                 'timestamp': datetime.utcnow().isoformat()
             })
-            # Push AI's response
             user_chat_ref.push({
                 'role': 'assistant',
                 'content': response_content,
@@ -1258,10 +1166,9 @@ def perplexity_chat():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/api/nba/awards_predictor', methods=['POST'])
-@credit_required(cost=1) # Costs 1 credit
+@credit_required(cost=1)
 @cross_origin()
 def awards_predictor():
-    """Predicts NBA award candidates based on user criteria."""
     try:
         data = request.get_json()
         award_type = data.get('award_type')
@@ -1281,10 +1188,9 @@ def awards_predictor():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/api/nba/young_player_projection', methods=['POST'])
-@credit_required(cost=1) # Costs 1 credit
+@credit_required(cost=1)
 @cross_origin()
 def young_player_projection():
-    """Projects the future potential of a young NBA player."""
     try:
         data = request.get_json()
         player_name = data.get('player_name')
@@ -1315,10 +1221,9 @@ def young_player_projection():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/api/nba/similar_players', methods=['POST'])
-@credit_required(cost=1) # Costs 1 credit
+@credit_required(cost=1)
 @cross_origin()
 def similar_players():
-    """Finds similar players based on specified criteria."""
     try:
         data = request.get_json()
         target_player = data.get('target_player')
@@ -1338,10 +1243,9 @@ def similar_players():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/api/nba/manual_player_compare', methods=['POST'])
-@credit_required(cost=1) # Costs 1 credit
+@credit_required(cost=1)
 @cross_origin()
 def manual_player_compare():
-    """Compares two NBA players in detail."""
     try:
         data = request.get_json()
         player1 = data.get('player1')
@@ -1365,10 +1269,9 @@ def manual_player_compare():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/api/nba/roster_suggestions', methods=['POST'])
-@credit_required(cost=1) # Costs 1 credit
+@credit_required(cost=1)
 @cross_origin()
 def roster_suggestions():
-    """Generates NBA roster suggestions based on user-defined constraints."""
     try:
         data = request.get_json()
         salary_cap = data.get('salary_cap')
@@ -1406,10 +1309,9 @@ def roster_suggestions():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/api/nba/trade_analysis', methods=['POST'])
-@credit_required(cost=1) # Costs 1 credit
+@credit_required(cost=1)
 @cross_origin()
 def trade_analysis():
-    """Analyzes a potential NBA trade scenario."""
     try:
         data = request.get_json()
         team1_trades = data.get('team1_trades')
@@ -1435,6 +1337,5 @@ def trade_analysis():
         return jsonify({'error': str(e)}), 500
 
 
-# ---------- Main ----------
 if __name__ == '__main__':
     app.run(debug=True, host="0.0.0.0", port=7860)