Update app.py

app.py

@@ -1,60 +1,99 @@
 from fastapi import FastAPI, HTTPException, Query
+from fastapi.middleware.cors import CORSMiddleware
 import duckdb
+import threading
 from contextlib import asynccontextmanager
+import os
 
-# Global connection object
-con = duckdb.connect()
+# Thread-local storage — har thread ka apna connection hoga
+# Isse concurrent requests pe crash nahi hoga
+_local = threading.local()
+
+PARQUET_URL = "https://huggingface.co/datasets/CodeXDevloper/Inddatainonefile/resolve/main/users_data.parquet"
+
+def get_con():
+    """Har thread ke liye alag DuckDB connection dega"""
+    if not hasattr(_local, 'con') or _local.con is None:
+        con = duckdb.connect()
+        con.execute("INSTALL httpfs; LOAD httpfs;")
+        con.execute(f"""
+            CREATE OR REPLACE VIEW users_view AS 
+            SELECT * FROM read_parquet('{PARQUET_URL}')
+        """)
+        _local.con = con
+    return _local.con
 
-# 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/CodeXDevloper/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.")
+    print("API start ho rahi hai...")
+    # Main thread ka connection startup pe bana lo
+    try:
+        con = get_con()
+        # Ek test query — parquet file check karo
+        con.execute("SELECT COUNT(*) FROM users_view").fetchone()
+        print("✅ Parquet file load ho gayi! API ready hai.")
+    except Exception as e:
+        print(f"⚠️ Startup error: {e}")
     yield
-    con.close()
+    # Cleanup
+    if hasattr(_local, 'con') and _local.con:
+        _local.con.close()
 
 app = FastAPI(title="Super Fast Search API", lifespan=lifespan)
 
+# CORS — kisi bhi frontend se API call ho sake
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["GET"],
+    allow_headers=["*"],
+)
+
 @app.get("/")
 def home():
-    return {"message": "Fast API online hai! Use /search?mobile=YOUR_NUMBER"}
+    return {
+        "message": "Super Fast Search API online hai!",
+        "usage": "/search?mobile=YOUR_NUMBER",
+        "status": "running"
+    }
+
+@app.get("/health")
+def health():
+    """Keep-alive ping ke liye yeh endpoint use karo"""
+    return {"status": "ok"}
 
 @app.get("/search")
 async def search_mobile(mobile: str = Query(..., description="Enter mobile number to search")):
+    if not mobile.strip().isdigit():
+        return {"status": "error", "message": "Sirf numbers enter karein."}
+
     try:
-        # Ab hum direct cache kiye hue 'users_view' se data nikalenge
+        con = get_con()  # Thread-safe connection
+        
         query = """
             SELECT mobile, name, fname, address, alt, circle, id, email 
             FROM users_view 
-            WHERE mobile = ? 
+            WHERE CAST(mobile AS VARCHAR) = ?
             LIMIT 1
         """
-        
-        result = con.execute(query, [str(mobile)]).fetchall()
-        
+        result = con.execute(query, [mobile.strip()]).fetchall()
+
         if not result:
             return {
-                "status": "error", 
+                "status": "error",
                 "message": f"Mobile number {mobile} nahi mila."
             }
-        
-        # Data mapping
+
         columns = [desc[0] for desc in con.description]
         row_dict = dict(zip(columns, result[0]))
-        
+
         return {
             "status": "success",
             "data": row_dict
         }
-        
+
     except Exception as e:
-        # Internal error ko secure rakhne ke liye mask kiya hai
-        raise HTTPException(status_code=500, detail="Query timeout ya process nahi ho payi.")
+        print(f"Query error: {e}")
+        # Connection reset karo agar error aaye
+        _local.con = None
+        raise HTTPException(status_code=500, detail="Query process nahi ho payi. Dobara try karein.")