Update main.py

main.py

@@ -1,33 +1,42 @@
 from fastapi import FastAPI, HTTPException, Security, Query
 from fastapi.security.api_key import APIKeyHeader
+from huggingface_hub import hf_hub_download
 import pandas as pd
 import os
 
-app = FastAPI(title="Paytm Database API", description="Developed by coder petro")
+app = FastAPI(title="Telegram Database API", description="Optimized by coder petro")
 
-# Updated API Key to demo as requested
 API_KEY = "demo"  
 API_KEY_NAME = "x-api-key"
 api_key_header = APIKeyHeader(name=API_KEY_NAME, auto_error=False)
 
 df = None
-CSV_FILE_PATH = "final_data.csv"
+REPO_ID = "coderpetro/telegram-data" 
+FILENAME = "Telegram_27.csv"
 
 @app.on_event("startup")
 def load_data():
     global df
-    if os.path.exists(CSV_FILE_PATH):
-        print("Loading database into memory...")
-        df = pd.read_csv(CSV_FILE_PATH, low_memory=False)
+    try:
+        print("Downloading database from private dataset...")
+        token = os.environ.get("HF_TOKEN")
+        csv_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME, repo_type="dataset", token=token)
+        
+        print("Loading only required columns into memory...")
+        # WARNING: Make sure 'name' and 'phone' match the exact column names in your CSV file (case-sensitive)
+        # If your CSV has 'Name' or 'Phone', change them accordingly below.
+        required_cols = ['name', 'phone'] 
+        
+        df = pd.read_csv(csv_path, usecols=required_cols, low_memory=False)
         df = df.astype(str).fillna("")
-        print("Database loaded successfully!")
-    else:
-        print("Error: final_data.csv not found!")
+        print("Database optimized and loaded successfully!")
+    except Exception as e:
+        print(f"Error loading database: {str(e)}")
 
 @app.get("/")
 def read_root():
     return {
-        "message": "API is running 24/7 thank you!",
+        "message": "Optimized API is running 24/7!",
         "developer": "coder petro"
     }