Update app.py

app.py

@@ -1,26 +1,39 @@
 from fastapi import FastAPI, HTTPException, Query
 import duckdb
+from contextlib import asynccontextmanager
 
-app = FastAPI(title="Fast Search API")
-
-# DuckDB connection aur httpfs extension load karo
+# Global connection object
 con = duckdb.connect()
-con.execute("INSTALL httpfs; LOAD httpfs;")
 
-# Public Parquet URL (Isme koi token ya header nahi chahiye)
-PARQUET_URL = "https://huggingface.co/datasets/tfqdeadlo/Inddatainonefile/resolve/main/users_data.parquet"
+# Lifespan use karke app start hote hi heavy initialization ek baar karenge
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    print("DuckDB extensions load ho rahi hain aur View ban raha hai...")
+    con.execute("INSTALL httpfs; LOAD httpfs;")
+    
+    # Parquet file ka correct URL
+    PARQUET_URL = "https://huggingface.co/datasets/tfqdeadlo/Inddatainonefile/resolve/main/users_data.parquet"
+    
+    # Startup par hi Ek baar View bana kar metadata cache kar lete hain
+    # Isse har API request par network par load nahi padega
+    con.execute(f"CREATE OR REPLACE VIEW users_view AS SELECT * FROM read_parquet('{PARQUET_URL}')")
+    print("View taiyar hai! API requests ke liye taiyar hai.")
+    yield
+    con.close()
+
+app = FastAPI(title="Super Fast Search API", lifespan=lifespan)
 
 @app.get("/")
 def home():
-    return {"message": "API online hai! Use /search?mobile=YOUR_NUMBER"}
+    return {"message": "Fast API online hai! Use /search?mobile=YOUR_NUMBER"}
 
 @app.get("/search")
 async def search_mobile(mobile: str = Query(..., description="Enter mobile number to search")):
     try:
-        # Clean SQL Query bina kisi extra headers ke
-        query = f"""
+        # Ab hum direct cache kiye hue 'users_view' se data nikalenge
+        query = """
             SELECT mobile, name, fname, address, alt, circle, id, email 
-            FROM read_parquet('{PARQUET_URL}') 
+            FROM users_view 
             WHERE mobile = ? 
             LIMIT 1
         """
@@ -33,7 +46,7 @@ async def search_mobile(mobile: str = Query(..., description="Enter mobile numbe
                 "message": f"Mobile number {mobile} nahi mila."
             }
         
-        # Data ko dictionary format me convert karo
+        # Data mapping
         columns = [desc[0] for desc in con.description]
         row_dict = dict(zip(columns, result[0]))
         
@@ -43,6 +56,6 @@ async def search_mobile(mobile: str = Query(..., description="Enter mobile numbe
         }
         
     except Exception as e:
-        # Safe logging: Taki koi internal server details public na hon
-        raise HTTPException(status_code=500, detail="Server query process nahi kar paya.")
+        # Internal error ko secure rakhne ke liye mask kiya hai
+        raise HTTPException(status_code=500, detail="Query timeout ya process nahi ho payi.")