Upload app.py with huggingface_hub

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, request, jsonify
+from flask import Flask, request
 import pyarrow.parquet as pq
 
 app = Flask(__name__)
@@ -17,8 +17,8 @@ def search_data(query):
             if query == tg_id or query == tg_username:
                 return {
                     "TG_ID": batch.column("TG_ID")[i].as_py(),
-                    "FIRST_NAME": batch.column("FIRST_NAME")[i].as_py(),
                     "TG_USERNAME": batch.column("TG_USERNAME")[i].as_py(),
+                    "FIRST_NAME": batch.column("FIRST_NAME")[i].as_py(),
                     "PHONE": batch.column("PHONE")[i].as_py(),
                 }
     return None
@@ -28,43 +28,110 @@ def search():
     query = request.args.get("search", "").strip()
 
     if not query:
-        return jsonify({
-            "status": "error",
-            "message": "Search value required!",
-            "example": "/search?search=Kudesner77",
-            "owner": "Code X",
-            "username": "@C4DX1"
-        }), 400
+        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')
 
     result = search_data(query)
 
     if result:
-        return jsonify({
-            "status": "found",
-            "query": query,
-            "data": result,
-            "owner": "Code X",
-            "username": "@C4DX1"
-        })
+        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"
+'
+            '}'
+        )
+        return app.response_class(response=response_text, status=200, mimetype='application/json')
     else:
-        return jsonify({
-            "status": "not found",
-            "query": query,
-            "message": "No data found!",
-            "owner": "Code X",
-            "username": "@C4DX1"
-        }), 404
+        response_text = (
+            '{
+'
+            f'  "status": "not found",
+'
+            f'  "query": "{query}",
+'
+            '  "message": "No data found!",
+'
+            '
+
+
+
+'
+            '  "owner": "Code X",
+'
+            '  "username": "@C4DX1"
+'
+            '}'
+        )
+        return app.response_class(response=response_text, status=404, mimetype='application/json')
 
 @app.route("/", methods=["GET"])
 def home():
-    return jsonify({
-        "status": "online",
-        "name": "Code X API",
-        "owner": "Code X",
-        "username": "@C4DX1",
-        "usage": "/search?search=TG_USERNAME or TG_ID",
-        "example": "/search?search=Kudesner77"
-    })
+    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"
+'
+        '}'
+    )
+    return app.response_class(response=response_text, status=200, mimetype='application/json')
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)