CodeXDevloper commited on
Commit
af8c0c9
·
verified ·
1 Parent(s): 0368873

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -0
app.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI, Query
2
+ from fastapi.middleware.cors import CORSMiddleware
3
+ from datasets import load_dataset
4
+
5
+ app = FastAPI()
6
+
7
+ app.add_middleware(
8
+ CORSMiddleware,
9
+ allow_origins=["*"],
10
+ allow_methods=["*"],
11
+ allow_headers=["*"],
12
+ )
13
+
14
+ DATASET = "CodeXDevloper/MergedTgDataset"
15
+
16
+ @app.get("/")
17
+ def root():
18
+ return {"status": "running", "usage": "/search?uid=385167100"}
19
+
20
+ @app.get("/search")
21
+ def search_uid(uid: str = Query(...)):
22
+ ds = load_dataset(DATASET, split="train", streaming=True)
23
+ for row in ds:
24
+ if uid == str(row.get("user_id", "")):
25
+ return {"found": True, "uid": uid, "data": row}
26
+ return {"found": False, "uid": uid}