Update app.py

app.py

@@ -1,60 +1,180 @@
 from fastapi import FastAPI, HTTPException, Query
 import duckdb
 from contextlib import asynccontextmanager
+from typing import Optional, List
 
-# Global connection object
-con = duckdb.connect()
+# Global connection
+con: Optional[duckdb.DuckDBPyConnection] = None
 
-# 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/nonamearyan/Inddatainonefile/resolve/main/users_data.parquet"
+    global con
+    try:
+        print("🔄 DuckDB initialize ho raha hai...")
+        con = duckdb.connect()
+        con.execute("INSTALL httpfs; LOAD httpfs;")
+        
+        PARQUET_URL = "https://huggingface.co/datasets/nonamearyan/Inddatainonefile/resolve/main/users_data.parquet"
+        
+        con.execute(f"""
+            CREATE OR REPLACE VIEW users_view AS 
+            SELECT * FROM read_parquet('{PARQUET_URL}')
+        """)
+        
+        print("✅ Server ready!")
+        
+    except Exception as e:
+        print(f"❌ Startup error: {e}")
+        raise
     
-    # 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()
+    
+    if con:
+        con.close()
+        print("🔒 Connection closed.")
 
-app = FastAPI(title="Super Fast Search API", lifespan=lifespan)
+app = FastAPI(
+    title="Search API",
+    description="Mobile, ID aur Email se search - Saare results dikhayega",
+    version="4.0.0",
+    lifespan=lifespan
+)
 
 @app.get("/")
 def home():
-    return {"message": "Fast API online hai! Use /search?mobile=YOUR_NUMBER"}
+    return {
+        "endpoints": {
+            "/search/mobile": "?mobile=9876543210",
+            "/search/id": "?id=123456789012",
+            "/search/email": "?email=user@example.com"
+        }
+    }
 
-@app.get("/search")
-async def search_mobile(mobile: str = Query(..., description="Enter mobile number to search")):
+
+# ============================================
+# ENDPOINT 1: Mobile Se Search (All Results)
+# ============================================
+@app.get("/search/mobile")
+async def search_by_mobile(
+    mobile: str = Query(..., min_length=10, max_length=15)
+):
+    if con is None:
+        raise HTTPException(status_code=503, detail="DB not connected")
+    
     try:
-        # Ab hum direct cache kiye hue 'users_view' se data nikalenge
         query = """
             SELECT mobile, name, fname, address, alt, circle, id, email 
             FROM users_view 
             WHERE mobile = ? 
-            LIMIT 1
         """
         
-        result = con.execute(query, [str(mobile)]).fetchall()
+        result = con.execute(query, [mobile.strip()]).fetchall()
         
         if not result:
             return {
-                "status": "error", 
-                "message": f"Mobile number {mobile} nahi mila."
+                "success": False,
+                "data": []
             }
         
-        # Data mapping
-        columns = [desc[0] for desc in con.description]
-        row_dict = dict(zip(columns, result[0]))
+        description = con.description
+        columns = [desc[0] for desc in description] if description else []
+        
+        results = []
+        for row in result:
+            results.append(dict(zip(columns, row)))
+        
+        return {
+            "success": True,
+            "data": results
+        }
+        
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+# ============================================
+# ENDPOINT 2: ID Se Search (All Results)
+# ============================================
+@app.get("/search/id")
+async def search_by_id(
+    id: str = Query(..., 
+                   min_length=12, 
+                   max_length=12,
+                   description="12-digit ID number")
+):
+    if con is None:
+        raise HTTPException(status_code=503, detail="DB not connected")
+    
+    if not id.isdigit():
+        raise HTTPException(status_code=400, detail="ID sirf numbers mein honi chahiye")
+    
+    try:
+        query = """
+            SELECT mobile, name, fname, address, alt, circle, id, email 
+            FROM users_view 
+            WHERE id = ? 
+        """
+        
+        result = con.execute(query, [id.strip()]).fetchall()
+        
+        if not result:
+            return {
+                "success": False,
+                "data": []
+            }
+        
+        description = con.description
+        columns = [desc[0] for desc in description] if description else []
+        
+        results = []
+        for row in result:
+            results.append(dict(zip(columns, row)))
+        
+        return {
+            "success": True,
+            "data": results
+        }
+        
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+# ============================================
+# ENDPOINT 3: Email Se Search (All Results)
+# ============================================
+@app.get("/search/email")
+async def search_by_email(
+    email: str = Query(..., description="Email ID")
+):
+    if con is None:
+        raise HTTPException(status_code=503, detail="DB not connected")
+    
+    try:
+        query = """
+            SELECT mobile, name, fname, address, alt, circle, id, email 
+            FROM users_view 
+            WHERE email = ? 
+        """
+        
+        result = con.execute(query, [email.strip()]).fetchall()
+        
+        if not result:
+            return {
+                "success": False,
+                "data": []
+            }
+        
+        description = con.description
+        columns = [desc[0] for desc in description] if description else []
+        
+        results = []
+        for row in result:
+            results.append(dict(zip(columns, row)))
         
         return {
-            "status": "success",
-            "data": row_dict
+            "success": True,
+            "data": results
         }
         
     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.")
+        raise HTTPException(status_code=500, detail=str(e))