Update app.py

app.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI, HTTPException, Query
 import duckdb
 from contextlib import asynccontextmanager
+import sys
 
 # Global connection object
 con = duckdb.connect()
@@ -8,107 +9,114 @@ 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...")
-    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.")
+    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
     yield
     con.close()
 
-app = FastAPI(title="Super Fast Search API", lifespan=lifespan)
+app = FastAPI(
+    title="Super Fast Search API", 
+    description="3 endpoints - Mobile, ID, Email se search karein",
+    version="2.0.0",
+    lifespan=lifespan
+)
 
 @app.get("/")
 def home():
     return {
-        "message": "Fast API online hai!",
+        "status": "online",
+        "message": "🚀 API ready hai!",
         "endpoints": {
-            "/search/mobile": "?mobile=YOUR_NUMBER",
-            "/search/id": "?id=YOUR_ID",
-            "/search/email": "?email=YOUR_EMAIL"
+            "/search/mobile": {"method": "GET", "params": "mobile=YOUR_NUMBER"},
+            "/search/id": {"method": "GET", "params": "id=YOUR_ID"},
+            "/search/email": {"method": "GET", "params": "email=YOUR_EMAIL"}
         }
     }
 
 # ============================================
-# ENDPOINT 1: Mobile Se Search (Saare Results)
+# 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(..., description="Enter mobile number to search")):
+async def search_mobile(mobile: str = Query(..., min_length=10, max_length=15, 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, [str(mobile)]).fetchall()
-        
-        if not result:
-            return {
-                "status": "error", 
-                "message": f"Mobile number {mobile} nahi mila."
-            }
-        
-        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)))
+        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": "success",
-            "data": results
+            "status": "error",
+            "message": "Search failed. Please try again.",
+            "data": []
         }
-        
-    except Exception as e:
-        raise HTTPException(status_code=500, detail="Query timeout ya process nahi ho payi.")
 
 
 # ============================================
-# ENDPOINT 2: ID Se Search (Saare Results)
+# ENDPOINT 2: ID Se Search (Integer)
 # ============================================
 @app.get("/search/id")
-async def search_id(id: str = Query(..., description="Enter ID to search")):
+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 = ? 
         """
-        # LIMIT 1 HATAYA - Saare results
-        
-        result = con.execute(query, [str(id)]).fetchall()
-        
-        if not result:
-            return {
-                "status": "error", 
-                "message": f"ID {id} nahi mila."
-            }
-        
-        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)))
+        result = con.execute(query, [id]).fetchall()
+        return build_response(result, "ID", str(id))
         
+    except Exception as e:
+        print(f"❌ ID search error: {e}")
         return {
-            "status": "success",
-            "data": results
+            "status": "error",
+            "message": "Search failed. Please try again.",
+            "data": []
         }
-        
-    except Exception as e:
-        raise HTTPException(status_code=500, detail="Query timeout ya process nahi ho payi.")
 
 
 # ============================================
-# ENDPOINT 3: Email Se Search (Saare Results)
+# ENDPOINT 3: Email Se Search
 # ============================================
 @app.get("/search/email")
 async def search_email(email: str = Query(..., description="Enter email to search")):
@@ -118,27 +126,25 @@ async def search_email(email: str = Query(..., description="Enter email to searc
             FROM users_view 
             WHERE email = ? 
         """
-        # LIMIT 1 HATAYA - Saare results
-        
-        result = con.execute(query, [str(email)]).fetchall()
-        
-        if not result:
-            return {
-                "status": "error", 
-                "message": f"Email {email} nahi mila."
-            }
-        
-        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)))
+        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": "success",
-            "data": results
+            "status": "error",
+            "message": "Search failed. Please try again.",
+            "data": []
         }
-        
-    except Exception as e:
-        raise HTTPException(status_code=500, detail="Query timeout ya process nahi ho payi.")
+
+
+# ============================================
+# 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"}