Create app.py

app.py

@@ -0,0 +1,265 @@
+import os
+import sys
+import logging
+import duckdb
+from fastapi import FastAPI, HTTPException, Query
+from fastapi.responses import JSONResponse
+from datetime import datetime, timedelta
+from dotenv import load_dotenv
+from typing import Optional
+import gradio as gr
+import spaces  # 🔥 ZeroGPU ke liye
+
+# ========== LOGGING ==========
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+logger.info(f"🐍 Python version: {sys.version}")
+
+# ========== LOAD ENV ==========
+load_dotenv()
+
+# ========== CONFIG ==========
+API_KEY = os.getenv("API_KEY", "@Aerivue")
+DATASET_URL = os.getenv("DATASET_URL", "https://huggingface.co/datasets/learnonix/icmr/resolve/main/icmr.parquet")
+DEPLOYMENT_DATE = datetime.strptime(os.getenv("DEPLOYMENT_DATE", "2026-06-27"), "%Y-%m-%d")
+EXPIRY_DAYS = int(os.getenv("EXPIRY_DAYS", 21))
+MEMORY_LIMIT = os.getenv("MEMORY_LIMIT", "400MB")
+
+def is_api_valid():
+    return datetime.now() < DEPLOYMENT_DATE + timedelta(days=EXPIRY_DAYS)
+
+# ========== FASTAPI APP ==========
+app = FastAPI(title="ICMR Data Search API - ZeroGPU", version="3.0.0")
+
+# ========== GLOBALS ==========
+con = None
+TABLE_EXISTS = False
+TABLE_LOADING = False
+
+# ========== STARTUP (CPU PAR) ==========
+@app.on_event("startup")
+async def startup_event():
+    global con, TABLE_EXISTS, TABLE_LOADING
+    logger.info("🟢 Initializing DuckDB on CPU...")
+    
+    try:
+        con = duckdb.connect()
+        con.execute(f"PRAGMA memory_limit='{MEMORY_LIMIT}'")
+        con.execute("SET enable_object_cache=true")
+        con.execute("INSTALL httpfs; LOAD httpfs;")
+        
+        TABLE_LOADING = True
+        logger.info("📥 Loading dataset (CPU)...")
+        
+        con.execute(f"""
+            CREATE OR REPLACE TABLE icmr_data AS 
+            SELECT * FROM read_parquet('{DATASET_URL}')
+        """)
+        con.execute("CREATE INDEX IF NOT EXISTS idx_phone ON icmr_data(phoneNumber)")
+        con.execute("CREATE INDEX IF NOT EXISTS idx_aadhar ON icmr_data(aadharNumber)")
+        
+        count = con.execute("SELECT COUNT(*) FROM icmr_data").fetchone()[0]
+        TABLE_EXISTS = True
+        logger.info(f"✅ Dataset loaded! Rows: {count:,}")
+        
+    except Exception as e:
+        logger.error(f"❌ Dataset load failed: {e}")
+        TABLE_EXISTS = False
+    finally:
+        TABLE_LOADING = False
+
+# ========== FASTAPI ROUTES (CPU) ==========
+@app.get("/")
+async def home(key: str = Query(...)):
+    if key != API_KEY:
+        raise HTTPException(401, "Invalid API key")
+    return {
+        "status": "success",
+        "message": "API Live with ZeroGPU!",
+        "data_loaded": TABLE_EXISTS,
+        "hardware": "CPU + ZeroGPU (on-demand)",
+        "developer": {"name": "@Aerivue", "team": "TEAM DARK KNIGHT APIs"}
+    }
+
+@app.get("/search")
+async def search_data(
+    key: str = Query(...),
+    number: Optional[str] = Query(None),
+    aadhar: Optional[str] = Query(None),
+    q: Optional[str] = Query(None)
+):
+    """Search endpoint - redirects to GPU via Gradio function"""
+    if key != API_KEY:
+        raise HTTPException(401, "Invalid API key")
+    if not is_api_valid():
+        raise HTTPException(403, "API expired")
+    if not number and not aadhar and not q:
+        raise HTTPException(400, "Provide number, aadhar, or q")
+    
+    if TABLE_LOADING:
+        return {"status": "error", "message": "Dataset loading, please wait..."}
+    if not TABLE_EXISTS:
+        return {"status": "error", "message": "Dataset not available"}
+    
+    search_term = q or number or aadhar
+    logger.info(f"🔍 Searching for: {search_term} (via GPU)")
+    
+    # 🔥 Call GPU function
+    result = search_dataset_gpu(search_term)
+    
+    if not result:
+        return {"status": "error", "message": f"'{search_term}' not found"}
+    
+    return {
+        "status": "success", 
+        "data": result,
+        "hardware": "GPU (ZeroGPU)"
+    }
+
+@app.get("/stats")
+async def stats(key: str = Query(...)):
+    if key != API_KEY:
+        raise HTTPException(401, "Invalid API key")
+    row_count = 0
+    if TABLE_EXISTS:
+        row_count = con.execute("SELECT COUNT(*) FROM icmr_data").fetchone()[0]
+    return {
+        "status": "success",
+        "stats": {
+            "data_loaded": TABLE_EXISTS,
+            "total_rows": row_count,
+            "expires_on": (DEPLOYMENT_DATE + timedelta(days=EXPIRY_DAYS)).strftime("%Y-%m-%d"),
+            "gpu_enabled": True,
+            "memory_limit": MEMORY_LIMIT
+        },
+        "developer": {"name": "@Aerivue", "team": "TEAM DARK KNIGHT APIs"}
+    }
+
+# ========== 🔥 ZEROGPU SEARCH FUNCTION ==========
+@spaces.GPU(duration=20)  # 20 seconds GPU time per call
+def search_dataset_gpu(search_term: str):
+    """GPU-accelerated search function"""
+    global con
+    
+    logger.info(f"🟣 GPU Activated for search: {search_term}")
+    
+    query = """
+        SELECT name, fathersName, phoneNumber, aadharNumber, 
+               address, district, pincode, state, town 
+        FROM icmr_data
+        WHERE phoneNumber = ? 
+           OR aadharNumber = ? 
+           OR LOWER(name) LIKE LOWER(?)
+           OR LOWER(district) LIKE LOWER(?)
+        LIMIT 5
+    """
+    
+    like_term = f"%{search_term}%"
+    result = con.execute(query, [
+        search_term,
+        search_term,
+        like_term,
+        like_term
+    ]).fetchall()
+    
+    if not result:
+        return None
+    
+    columns = [desc[0] for desc in con.description]
+    return [dict(zip(columns, row)) for row in result]
+
+# ========== GRADIO INTERFACE ==========
+def gradio_search(key: str, search_term: str):
+    """Gradio interface with GPU support"""
+    if not key or not search_term:
+        return "❌ Please provide both API key and search term"
+    
+    if key != API_KEY:
+        return "❌ Invalid API key"
+    
+    if TABLE_LOADING:
+        return "⏳ Dataset is loading, please wait 30-60 seconds..."
+    
+    if not TABLE_EXISTS:
+        return "❌ Dataset not available. Check logs."
+    
+    try:
+        # 🔥 This call goes to GPU
+        results = search_dataset_gpu(search_term)
+        
+        if not results:
+            return f"❌ '{search_term}' not found in database"
+        
+        output = "✅ **Search Results (GPU Accelerated):**\n\n"
+        for i, data in enumerate(results, 1):
+            output += f"**Result {i}**\n"
+            output += f"👤 **Name:** {data.get('name', 'N/A')}\n"
+            output += f"📞 **Phone:** {data.get('phoneNumber', 'N/A')}\n"
+            output += f"🆔 **Aadhar:** {data.get('aadharNumber', 'N/A')}\n"
+            output += f"🏠 **Address:** {data.get('address', 'N/A')}\n"
+            output += f"🏛️ **District:** {data.get('district', 'N/A')}\n"
+            output += f"📍 **State:** {data.get('state', 'N/A')}\n"
+            output += "---\n"
+        
+        output += f"\n⚡ Powered by ZeroGPU (H200)"
+        return output
+        
+    except Exception as e:
+        logger.error(f"GPU Search error: {e}")
+        return f"❌ Error: {str(e)}"
+
+# Create Gradio interface
+demo = gr.Interface(
+    fn=gradio_search,
+    inputs=[
+        gr.Textbox(
+            label="🔑 API Key", 
+            placeholder="Enter your API key",
+            type="password",
+            value=""
+        ),
+        gr.Textbox(
+            label="🔍 Search Term", 
+            placeholder="Phone, Aadhar, Name, or District",
+            value=""
+        )
+    ],
+    outputs=gr.Textbox(label="📊 Results", lines=20),
+    title="🔍 ICMR Data Search - ZeroGPU Edition",
+    description="Search ICMR dataset using GPU acceleration",
+    examples=[
+        ["@Aerivue", "9876543210"],
+        ["@Aerivue", "Rahul"],
+        ["@Aerivue", "Mumbai"],
+        ["@Aerivue", "Delhi"]
+    ],
+    cache_examples=False  # Keep False for ZeroGPU
+)
+
+# ========== MOUNT GRADIO ON FASTAPI ==========
+app = gr.mount_gradio_app(app, demo, path="/")
+
+# ========== HEALTH CHECK ==========
+@app.get("/health")
+async def health():
+    return {
+        "status": "healthy" if TABLE_EXISTS else "loading",
+        "data_loaded": TABLE_EXISTS,
+        "data_loading": TABLE_LOADING,
+        "database_connected": con is not None,
+        "gpu_available": True
+    }
+
+# ========== RUN ==========
+if __name__ == "__main__":
+    import uvicorn
+    port = int(os.getenv("PORT", 7860))
+    uvicorn.run(
+        app,
+        host="0.0.0.0",
+        port=port,
+        log_level="info"
+    )