Update app.py

app.py

@@ -1,12 +1,17 @@
 from fastapi import FastAPI, Query, HTTPException
 import duckdb
 import time
-import os
+import gradio as gr
 
-app = FastAPI(title="PK DB API")
+# ============================================================
+# FASTAPI APP
+# ============================================================
+fastapi_app = FastAPI(title="PK DB API")
 
+# ---------- CONFIG ----------
 PARQUET_URL = "https://huggingface.co/datasets/parthislive/pk-db/resolve/main/parts/*.parquet"
 
+# ---------- DUCKDB ----------
 _con = None
 
 def get_con():
@@ -21,72 +26,200 @@ def get_con():
     return _con
 
 
-@app.get("/")
+# ============================================================
+# FASTAPI ROUTES
+# ============================================================
+@fastapi_app.get("/")
 def root():
-    return {"app": "PK DB API", "endpoints": ["/api/number", "/api/cnic", "/api/name", "/api/address"]}
+    return {
+        "app": "PK DB API",
+        "dataset": "parthislive/pk-db",
+        "columns": ["NUMBER", "CNIC", "NAME", "ADDRESS"],
+        "endpoints": {
+            "/api/number?number=<num>": "Search by NUMBER",
+            "/api/cnic?cnic=<cnic>": "Search by CNIC",
+            "/api/name?name=<name>": "Search by NAME",
+            "/api/address?address=<addr>": "Search by ADDRESS",
+            "/ui": "Gradio UI"
+        }
+    }
 
-@app.get("/health")
+
+@fastapi_app.get("/health")
 def health():
     return {"status": "ok"}
 
-@app.get("/api/number")
+
+@fastapi_app.get("/api/number")
 def api_number(number: str = Query(...)):
     try:
         start = time.time()
         con = get_con()
-        sql = f"SELECT NUMBER, CNIC, NAME, ADDRESS FROM read_parquet('{PARQUET_URL}') WHERE NUMBER = ? LIMIT 10"
+        sql = f"""
+            SELECT NUMBER, CNIC, NAME, ADDRESS
+            FROM read_parquet('{PARQUET_URL}')
+            WHERE NUMBER = ?
+            LIMIT 10
+        """
         result = con.execute(sql, [int(number)]).fetchdf()
         elapsed = round(time.time() - start, 2)
         if result.empty:
             raise HTTPException(404, f"No data for NUMBER={number}")
-        return {"status": "success", "number": number, "count": len(result), "time": f"{elapsed}s", "data": result.to_dict(orient="records")}
+        return {
+            "status": "success",
+            "number": number,
+            "count": len(result),
+            "time": f"{elapsed}s",
+            "data": result.to_dict(orient="records")
+        }
     except HTTPException:
         raise
     except Exception as e:
         raise HTTPException(500, str(e))
 
-@app.get("/api/cnic")
+
+@fastapi_app.get("/api/cnic")
 def api_cnic(cnic: str = Query(...)):
     try:
         start = time.time()
         con = get_con()
-        sql = f"SELECT NUMBER, CNIC, NAME, ADDRESS FROM read_parquet('{PARQUET_URL}') WHERE CNIC = ? LIMIT 10"
+        sql = f"""
+            SELECT NUMBER, CNIC, NAME, ADDRESS
+            FROM read_parquet('{PARQUET_URL}')
+            WHERE CNIC = ?
+            LIMIT 10
+        """
         result = con.execute(sql, [str(cnic).strip()]).fetchdf()
         elapsed = round(time.time() - start, 2)
         if result.empty:
             raise HTTPException(404, f"No data for CNIC={cnic}")
-        return {"status": "success", "cnic": cnic, "count": len(result), "time": f"{elapsed}s", "data": result.to_dict(orient="records")}
+        return {
+            "status": "success",
+            "cnic": cnic,
+            "count": len(result),
+            "time": f"{elapsed}s",
+            "data": result.to_dict(orient="records")
+        }
     except HTTPException:
         raise
     except Exception as e:
         raise HTTPException(500, str(e))
 
-@app.get("/api/name")
+
+@fastapi_app.get("/api/name")
 def api_name(name: str = Query(...)):
     try:
         start = time.time()
         con = get_con()
-        sql = f"SELECT NUMBER, CNIC, NAME, ADDRESS FROM read_parquet('{PARQUET_URL}') WHERE NAME ILIKE ? LIMIT 50"
+        sql = f"""
+            SELECT NUMBER, CNIC, NAME, ADDRESS
+            FROM read_parquet('{PARQUET_URL}')
+            WHERE NAME ILIKE ?
+            LIMIT 50
+        """
         result = con.execute(sql, [f"%{name}%"]).fetchdf()
         elapsed = round(time.time() - start, 2)
-        return {"status": "success" if not result.empty else "not_found", "name": name, "count": len(result), "time": f"{elapsed}s", "data": result.to_dict(orient="records")}
+        return {
+            "status": "success" if not result.empty else "not_found",
+            "name": name,
+            "count": len(result),
+            "time": f"{elapsed}s",
+            "data": result.to_dict(orient="records")
+        }
     except Exception as e:
         raise HTTPException(500, str(e))
 
-@app.get("/api/address")
+
+@fastapi_app.get("/api/address")
 def api_address(address: str = Query(...)):
     try:
         start = time.time()
         con = get_con()
-        sql = f"SELECT NUMBER, CNIC, NAME, ADDRESS FROM read_parquet('{PARQUET_URL}') WHERE ADDRESS ILIKE ? LIMIT 50"
+        sql = f"""
+            SELECT NUMBER, CNIC, NAME, ADDRESS
+            FROM read_parquet('{PARQUET_URL}')
+            WHERE ADDRESS ILIKE ?
+            LIMIT 50
+        """
         result = con.execute(sql, [f"%{address}%"]).fetchdf()
         elapsed = round(time.time() - start, 2)
-        return {"status": "success" if not result.empty else "not_found", "address": address, "count": len(result), "time": f"{elapsed}s", "data": result.to_dict(orient="records")}
+        return {
+            "status": "success" if not result.empty else "not_found",
+            "address": address,
+            "count": len(result),
+            "time": f"{elapsed}s",
+            "data": result.to_dict(orient="records")
+        }
     except Exception as e:
         raise HTTPException(500, str(e))
 
 
+# ============================================================
+# GRADIO UI
+# ============================================================
+def search_ui(query, field):
+    if not query or not query.strip():
+        return "⚠️ Kuch toh search karo bhai."
+    
+    query = query.strip()
+    con = get_con()
+    
+    try:
+        if field == "NUMBER":
+            sql = f"SELECT NUMBER, CNIC, NAME, ADDRESS FROM read_parquet('{PARQUET_URL}') WHERE NUMBER = ? LIMIT 10"
+            result = con.execute(sql, [int(query)]).fetchdf()
+        elif field == "CNIC":
+            sql = f"SELECT NUMBER, CNIC, NAME, ADDRESS FROM read_parquet('{PARQUET_URL}') WHERE CNIC = ? LIMIT 10"
+            result = con.execute(sql, [query]).fetchdf()
+        elif field == "NAME":
+            sql = f"SELECT NUMBER, CNIC, NAME, ADDRESS FROM read_parquet('{PARQUET_URL}') WHERE NAME ILIKE ? LIMIT 20"
+            result = con.execute(sql, [f"%{query}%"]).fetchdf()
+        else:  # ADDRESS
+            sql = f"SELECT NUMBER, CNIC, NAME, ADDRESS FROM read_parquet('{PARQUET_URL}') WHERE ADDRESS ILIKE ? LIMIT 20"
+            result = con.execute(sql, [f"%{query}%"]).fetchdf()
+        
+        if result.empty:
+            return f"❌ Kuch nahi mila for {field}={query}"
+        
+        return result.to_string()
+    except Exception as e:
+        return f"❌ Error: {e}"
+
+
+with gr.Blocks(title="PK DB Search") as demo:
+    gr.Markdown("# 🔍 PK DB Search")
+    gr.Markdown("Search **218M Pakistan records** — NUMBER, CNIC, NAME, ADDRESS")
+    
+    with gr.Row():
+        query_input = gr.Textbox(
+            label="Search Query",
+            placeholder="NUMBER, CNIC, NAME ya ADDRESS daalo...",
+            lines=1,
+        )
+        field_select = gr.Radio(
+            choices=["NUMBER", "CNIC", "NAME", "ADDRESS"],
+            label="Search Field",
+            value="NUMBER",
+        )
+    
+    search_btn = gr.Button("🔍 Search", variant="primary", size="lg")
+    output = gr.Textbox(label="Results", lines=20)
+    
+    search_btn.click(fn=search_ui, inputs=[query_input, field_select], outputs=output)
+    query_input.submit(fn=search_ui, inputs=[query_input, field_select], outputs=output)
+    
+    gr.Markdown("---")
+    gr.Markdown("**API Endpoints:** `/api/number`, `/api/cnic`, `/api/name`, `/api/address`")
+
+
+# ============================================================
+# MOUNT GRADIO ON FASTAPI
+# ============================================================
+app = gr.mount_gradio_app(fastapi_app, demo, path="/ui")
+
+
+# ============================================================
+# LAUNCH (HF Space Ke Liye)
+# ============================================================
 if __name__ == "__main__":
-    import uvicorn
-    port = int(os.environ.get("PORT", 7860))
-    uvicorn.run(app, host="0.0.0.0", port=port)
+    demo.launch(server_name="0.0.0.0", server_port=7860)