Update app.py

app.py

@@ -1,180 +1,144 @@
 from fastapi import FastAPI, HTTPException, Query
 import duckdb
 from contextlib import asynccontextmanager
-from typing import Optional, List
 
-# Global connection
-con: Optional[duckdb.DuckDBPyConnection] = None
+# Global connection object
+con = duckdb.connect()
 
+# Lifespan use karke app start hote hi heavy initialization ek baar karenge
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    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
+    print("DuckDB extensions load ho rahi hain aur View ban raha hai...")
+    con.execute("INSTALL httpfs; LOAD httpfs;")
     
-    yield
+    # Parquet file ka correct URL
+    PARQUET_URL = "https://huggingface.co/datasets/nonamearyan/Inddatainonefile/resolve/main/users_data.parquet"
     
-    if con:
-        con.close()
-        print("🔒 Connection closed.")
+    # Startup par hi Ek baar View bana kar metadata cache kar lete hain
+    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="Search API",
-    description="Mobile, ID aur Email se search - Saare results dikhayega",
-    version="4.0.0",
-    lifespan=lifespan
-)
+app = FastAPI(title="Super Fast Search API", lifespan=lifespan)
 
 @app.get("/")
 def home():
     return {
+        "message": "Fast API online hai!",
         "endpoints": {
-            "/search/mobile": "?mobile=9876543210",
-            "/search/id": "?id=123456789012",
-            "/search/email": "?email=user@example.com"
+            "/search/mobile": "?mobile=YOUR_NUMBER",
+            "/search/id": "?id=YOUR_ID",
+            "/search/email": "?email=YOUR_EMAIL"
         }
     }
 
-
 # ============================================
-# ENDPOINT 1: Mobile Se Search (All Results)
+# ENDPOINT 1: Mobile Se Search (Saare 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")
-    
+async def search_mobile(mobile: str = Query(..., description="Enter mobile number to search")):
     try:
         query = """
             SELECT mobile, name, fname, address, alt, circle, id, email 
             FROM users_view 
             WHERE mobile = ? 
         """
+        # LIMIT 1 HATAYA - Saare results
         
-        result = con.execute(query, [mobile.strip()]).fetchall()
+        result = con.execute(query, [str(mobile)]).fetchall()
         
         if not result:
             return {
-                "success": False,
-                "data": []
+                "status": "error", 
+                "message": f"Mobile number {mobile} nahi mila."
             }
         
-        description = con.description
-        columns = [desc[0] for desc in description] if description else []
+        columns = [desc[0] for desc in con.description]
         
+        # Saare results ko list mein convert karo
         results = []
         for row in result:
             results.append(dict(zip(columns, row)))
         
         return {
-            "success": True,
+            "status": "success",
             "data": results
         }
         
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail="Query timeout ya process nahi ho payi.")
 
 
 # ============================================
-# ENDPOINT 2: ID Se Search (All Results)
+# ENDPOINT 2: ID Se Search (Saare 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")
-    
+async def search_id(id: str = Query(..., description="Enter ID to search")):
     try:
         query = """
             SELECT mobile, name, fname, address, alt, circle, id, email 
             FROM users_view 
             WHERE id = ? 
         """
+        # LIMIT 1 HATAYA - Saare results
         
-        result = con.execute(query, [id.strip()]).fetchall()
+        result = con.execute(query, [str(id)]).fetchall()
         
         if not result:
             return {
-                "success": False,
-                "data": []
+                "status": "error", 
+                "message": f"ID {id} nahi mila."
             }
         
-        description = con.description
-        columns = [desc[0] for desc in description] if description else []
+        columns = [desc[0] for desc in con.description]
         
+        # Saare results ko list mein convert karo
         results = []
         for row in result:
             results.append(dict(zip(columns, row)))
         
         return {
-            "success": True,
+            "status": "success",
             "data": results
         }
         
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail="Query timeout ya process nahi ho payi.")
 
 
 # ============================================
-# ENDPOINT 3: Email Se Search (All Results)
+# ENDPOINT 3: Email Se Search (Saare 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")
-    
+async def search_email(email: str = Query(..., description="Enter email to search")):
     try:
         query = """
             SELECT mobile, name, fname, address, alt, circle, id, email 
             FROM users_view 
             WHERE email = ? 
         """
+        # LIMIT 1 HATAYA - Saare results
         
-        result = con.execute(query, [email.strip()]).fetchall()
+        result = con.execute(query, [str(email)]).fetchall()
         
         if not result:
             return {
-                "success": False,
-                "data": []
+                "status": "error", 
+                "message": f"Email {email} nahi mila."
             }
         
-        description = con.description
-        columns = [desc[0] for desc in description] if description else []
+        columns = [desc[0] for desc in con.description]
         
+        # Saare results ko list mein convert karo
         results = []
         for row in result:
             results.append(dict(zip(columns, row)))
         
         return {
-            "success": True,
+            "status": "success",
             "data": results
         }
         
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail="Query timeout ya process nahi ho payi.")