Update app.py

app.py

@@ -4,9 +4,78 @@ import os
 import pandas as pd
 from contextlib import asynccontextmanager
 
+# File path (mounted bucket)
+FILE_PATH = "/import duckdb
+from fastapi import FastAPI, Query
+import os
+import pandas as pd
+from contextlib import asynccontextmanager
+
 # File path (mounted bucket)
 FILE_PATH = "/data/TGDATA BY DEADLOX P4.parquet"
 
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Startup
+    if os.path.exists(FILE_PATH):
+        print(f"✅ File found: {FILE_PATH}")
+        size = os.path.getsize(FILE_PATH) / (1024**3)
+        print(f"📏 Size: {size:.2f} GB")
+    else:
+        print(f"❌ File NOT found at {FILE_PATH}")
+    yield
+    print("Shutting down...")
+
+app = FastAPI(lifespan=lifespan)
+con = duckdb.connect()
+
+@app.get("/")
+async def root():
+    return {
+        "message": "Parquet Search API",
+        "endpoint": "/search?q=keyword&limit=10",
+        "file_available": os.path.exists(FILE_PATH)
+    }
+
+@app.get("/search")
+async def search(q: str = Query(...), limit: int = 20):
+    if not q.strip():
+        return {"error": "Missing 'q' parameter"}
+    
+    try:
+        # Get column names
+        df_cols = con.execute(f"SELECT * FROM '{FILE_PATH}' LIMIT 0").fetchdf()
+        columns = list(df_cols.columns)
+        
+        # Build dynamic WHERE clause for all columns
+        conditions = [f"CAST(\"{col}\" AS VARCHAR) LIKE '%{q}%'" for col in columns]
+        where_clause = " OR ".join(conditions)
+        
+        query = f"SELECT * FROM '{FILE_PATH}' WHERE {where_clause} LIMIT {limit}"
+        results = con.execute(query).fetchdf()
+        results_dict = results.to_dict(orient='records')
+        
+        # 🔥 FIX: Swap username and phone columns
+        for record in results_dict:
+            username_val = record.get('username', '')
+            phone_val = record.get('phone', '')
+            record['username'] = phone_val   # Phone number goes to username field
+            record['phone'] = username_val   # Username/name goes to phone field
+        
+        return {
+            "query": q,
+            "count": len(results_dict),
+            "limit": limit,
+            "columns": columns,
+            "results": results_dict
+        }
+    except Exception as e:
+        return {"error": str(e)}
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860) BY DEADLOX P4.parquet"
+
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     # Startup