Upload app.py with huggingface_hub

app.py

@@ -2,18 +2,15 @@ from flask import Flask, request
 import pyarrow.parquet as pq
 
 app = Flask(__name__)
-
 PARQUET_FILE = "mera_data.parquet"
 
 def search_data(query):
     query = str(query).strip().lower()
     pf = pq.ParquetFile(PARQUET_FILE)
-
     for batch in pf.iter_batches(batch_size=10000):
         for i in range(batch.num_rows):
             tg_id = str(batch.column("TG_ID")[i].as_py() or "").lower()
             tg_username = str(batch.column("TG_USERNAME")[i].as_py() or "").lower()
-
             if query == tg_id or query == tg_username:
                 return {
                     "TG_ID": batch.column("TG_ID")[i].as_py(),
@@ -26,112 +23,54 @@ def search_data(query):
 @app.route("/search", methods=["GET"])
 def search():
     query = request.args.get("search", "").strip()
-
     if not query:
-        response_text = (
-            '{
-'
-            '  "status": "error",
-'
-            '  "message": "Search value required!",
-'
-            '  "example": "/search?search=Kudesner77",
-'
-            '
-
-
-
-'
-            '  "owner": "Code X",
-'
-            '  "username": "@C4DX1"
-'
-            '}'
-        )
-        return app.response_class(response=response_text, status=400, mimetype='application/json')
-
+        r = '{"status":"error","message":"Search value required!","example":"/search?search=Kudesner77"}'
+        return app.response_class(response=r, status=400, mimetype="application/json")
     result = search_data(query)
-
     if result:
-        response_text = (
-            '{
-'
-            f'  "status": "found",
-'
-            f'  "query": "{query}",
-'
-            f'  "data": {{
-'
-            f'    "TG_ID": "{result["TG_ID"]}",
-'
-            f'    "TG_USERNAME": "{result["TG_USERNAME"]}",
-'
-            f'    "FIRST_NAME": "{result["FIRST_NAME"]}",
-'
-            f'    "PHONE": "{result["PHONE"]}"
-'
-            f'  }},
-'
-            '
-
-
-
-'
-            '  "owner": "Code X",
-'
-            '  "username": "@C4DX1"
-'
-            '}'
+        r = (
+            "{\n"
+            '  "status": "found",\n'
+            f'  "query": "{query}",\n'
+            '  "data": {\n'
+            f'    "TG_ID": "{result["TG_ID"]}",\n'
+            f'    "TG_USERNAME": "{result["TG_USERNAME"]}",\n'
+            f'    "FIRST_NAME": "{result["FIRST_NAME"]}",\n'
+            f'    "PHONE": "{result["PHONE"]}"\n'
+            "  },\n"
+            "\n\n\n\n"
+            '  "owner": "Code X",\n'
+            '  "username": "@C4DX1"\n'
+            "}"
         )
-        return app.response_class(response=response_text, status=200, mimetype='application/json')
+        return app.response_class(response=r, status=200, mimetype="application/json")
     else:
-        response_text = (
-            '{
-'
-            f'  "status": "not found",
-'
-            f'  "query": "{query}",
-'
-            '  "message": "No data found!",
-'
-            '
-
-
-
-'
-            '  "owner": "Code X",
-'
-            '  "username": "@C4DX1"
-'
-            '}'
+        r = (
+            "{\n"
+            '  "status": "not found",\n'
+            f'  "query": "{query}",\n'
+            '  "message": "No data found!",\n'
+            "\n\n\n\n"
+            '  "owner": "Code X",\n'
+            '  "username": "@C4DX1"\n'
+            "}"
         )
-        return app.response_class(response=response_text, status=404, mimetype='application/json')
+        return app.response_class(response=r, status=404, mimetype="application/json")
 
 @app.route("/", methods=["GET"])
 def home():
-    response_text = (
-        '{
-'
-        '  "status": "online",
-'
-        '  "name": "Code X API",
-'
-        '  "usage": "/search?search=TG_USERNAME or TG_ID",
-'
-        '  "example": "/search?search=Kudesner77",
-'
-        '
-
-
-
-'
-        '  "owner": "Code X",
-'
-        '  "username": "@C4DX1"
-'
-        '}'
+    r = (
+        "{\n"
+        '  "status": "online",\n'
+        '  "name": "Code X API",\n'
+        '  "usage": "/search?search=TG_USERNAME or TG_ID",\n'
+        '  "example": "/search?search=Kudesner77",\n'
+        "\n\n\n\n"
+        '  "owner": "Code X",\n'
+        '  "username": "@C4DX1"\n'
+        "}"
     )
-    return app.response_class(response=response_text, status=200, mimetype='application/json')
+    return app.response_class(response=r, status=200, mimetype="application/json")
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)