CodeXDevloper commited on
Commit
2874220
·
verified ·
1 Parent(s): 36b55d5

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +70 -0
app.py ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, request, jsonify
2
+ import pyarrow.parquet as pq
3
+
4
+ app = Flask(__name__)
5
+
6
+ PARQUET_FILE = "mera_data.parquet"
7
+
8
+ def search_data(query):
9
+ query = str(query).strip().lower()
10
+ pf = pq.ParquetFile(PARQUET_FILE)
11
+
12
+ for batch in pf.iter_batches(batch_size=10000):
13
+ for i in range(batch.num_rows):
14
+ tg_id = str(batch.column("TG_ID")[i].as_py() or "").lower()
15
+ tg_username = str(batch.column("TG_USERNAME")[i].as_py() or "").lower()
16
+
17
+ if query == tg_id or query == tg_username:
18
+ return {
19
+ "TG_ID": batch.column("TG_ID")[i].as_py(),
20
+ "FIRST_NAME": batch.column("FIRST_NAME")[i].as_py(),
21
+ "TG_USERNAME": batch.column("TG_USERNAME")[i].as_py(),
22
+ "PHONE": batch.column("PHONE")[i].as_py(),
23
+ }
24
+ return None
25
+
26
+ @app.route("/search", methods=["GET"])
27
+ def search():
28
+ query = request.args.get("search", "").strip()
29
+
30
+ if not query:
31
+ return jsonify({
32
+ "status": "error",
33
+ "message": "Search value required!",
34
+ "example": "/search?search=Kudesner77",
35
+ "owner": "Code X",
36
+ "username": "@C4DX1"
37
+ }), 400
38
+
39
+ result = search_data(query)
40
+
41
+ if result:
42
+ return jsonify({
43
+ "status": "found",
44
+ "query": query,
45
+ "data": result,
46
+ "owner": "Code X",
47
+ "username": "@C4DX1"
48
+ })
49
+ else:
50
+ return jsonify({
51
+ "status": "not found",
52
+ "query": query,
53
+ "message": "No data found!",
54
+ "owner": "Code X",
55
+ "username": "@C4DX1"
56
+ }), 404
57
+
58
+ @app.route("/", methods=["GET"])
59
+ def home():
60
+ return jsonify({
61
+ "status": "online",
62
+ "name": "Code X API",
63
+ "owner": "Code X",
64
+ "username": "@C4DX1",
65
+ "usage": "/search?search=TG_USERNAME or TG_ID",
66
+ "example": "/search?search=Kudesner77"
67
+ })
68
+
69
+ if __name__ == "__main__":
70
+ app.run(host="0.0.0.0", port=7860)