Files changed (1) hide show
  1. app.py +32 -0
app.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+ import duckdb
3
+
4
+ app = FastAPI()
5
+
6
+ HF_PATH = "hf://datasets/tfqdeadlo/Inddata"
7
+
8
+ @app.get("/")
9
+ def home():
10
+ return {"status": "API running"}
11
+
12
+ @app.get("/search")
13
+ def search(mobile: str):
14
+ try:
15
+ prefix = mobile[:3]
16
+
17
+ query = f"""
18
+ SELECT *
19
+ FROM read_parquet('{HF_PATH}/{prefix}/*.parquet')
20
+ WHERE mobile = '{mobile}'
21
+ LIMIT 10
22
+ """
23
+
24
+ df = duckdb.query(query).to_df()
25
+
26
+ return {
27
+ "found": len(df),
28
+ "data": df.to_dict(orient="records")
29
+ }
30
+
31
+ except Exception as e:
32
+ return {"error": str(e)}