Update app.py

app.py

@@ -1,7 +1,6 @@
 from fastapi import FastAPI, HTTPException, Query
 import duckdb
 from contextlib import asynccontextmanager
-import sys
 
 # Global connection object
 con = duckdb.connect()
@@ -9,142 +8,53 @@ con = duckdb.connect()
 # 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...")
-    try:
-        con.execute("INSTALL httpfs; LOAD httpfs;")
-        
-        # Parquet file ka correct URL
-        PARQUET_URL = "https://huggingface.co/datasets/nonamearyan/Inddatainonefile/resolve/main/users_data.parquet"
-        
-        # 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.")
-        
-    except Exception as e:
-        print(f"❌ Startup error: {e}")
-        print("⚠️ Server continue karega, lekin data nahi milega")
-        # Fail silent, but server continue
+    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"
+    
+    # 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", 
-    description="3 endpoints - Mobile, ID, Email se search karein",
-    version="2.0.0",
-    lifespan=lifespan
-)
+app = FastAPI(title="Super Fast Search API", lifespan=lifespan)
 
 @app.get("/")
 def home():
-    return {
-        "status": "online",
-        "message": "🚀 API ready hai!",
-        "endpoints": {
-            "/search/mobile": {"method": "GET", "params": "mobile=YOUR_NUMBER"},
-            "/search/id": {"method": "GET", "params": "id=YOUR_ID"},
-            "/search/email": {"method": "GET", "params": "email=YOUR_EMAIL"}
-        }
-    }
+    return {"message": "Fast API online hai! Use /search?mobile=YOUR_NUMBER"}
 
-# ============================================
-# Helper Function - Response Build
-# ============================================
-def build_response(result, query_field, query_value):
-    """Response build karne ka helper function"""
-    if not result:
-        return {
-            "status": "error",
-            "message": f"{query_field} '{query_value}' nahi mila.",
-            "data": []
-        }
-    
-    columns = [desc[0] for desc in con.description] if con.description else []
-    results = []
-    for row in result:
-        results.append(dict(zip(columns, row)))
-    
-    return {
-        "status": "success",
-        "message": f"{len(results)} result(s) found",
-        "data": results
-    }
-
-# ============================================
-# ENDPOINT 1: Mobile Se Search
-# ============================================
-@app.get("/search/mobile")
-async def search_mobile(mobile: str = Query(..., min_length=10, max_length=15, description="Enter mobile number to search")):
+@app.get("/search")
+async def search_mobile(mobile: str = Query(..., description="Enter mobile number to search")):
     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.strip())]).fetchall()
-        return build_response(result, "Mobile number", mobile)
         
-    except Exception as e:
-        print(f"❌ Mobile search error: {e}")
-        return {
-            "status": "error",
-            "message": "Search failed. Please try again.",
-            "data": []
-        }
-
-
-# ============================================
-# ENDPOINT 2: ID Se Search (Integer)
-# ============================================
-@app.get("/search/id")
-async def search_id(id: int = Query(..., description="Enter ID to search (numeric)")):
-    try:
-        query = """
-            SELECT mobile, name, fname, address, alt, circle, id, email 
-            FROM users_view 
-            WHERE id = ? 
-        """
-        result = con.execute(query, [id]).fetchall()
-        return build_response(result, "ID", str(id))
+        result = con.execute(query, [str(mobile)]).fetchall()
+        
+        if not result:
+            return {
+                "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]))
         
-    except Exception as e:
-        print(f"❌ ID search error: {e}")
         return {
-            "status": "error",
-            "message": "Search failed. Please try again.",
-            "data": []
+            "status": "success",
+            "data": row_dict
         }
-
-
-# ============================================
-# ENDPOINT 3: Email Se Search
-# ============================================
-@app.get("/search/email")
-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 = ? 
-        """
-        result = con.execute(query, [str(email.strip())]).fetchall()
-        return build_response(result, "Email", email)
         
     except Exception as e:
-        print(f"❌ Email search error: {e}")
-        return {
-            "status": "error",
-            "message": "Search failed. Please try again.",
-            "data": []
-        }
-
-
-# ============================================
-# Health Check Endpoint
-# ============================================
-@app.get("/health")
-async def health_check():
-    try:
-        con.execute("SELECT 1").fetchone()
-        return {"status": "healthy", "message": "Database connected"}
-    except:
-        return {"status": "unhealthy", "message": "Database not connected"}
+        # Internal error ko secure rakhne ke liye mask kiya hai
+        raise HTTPException(status_code=500, detail="Query timeout ya process nahi ho payi.")