Update app.py

app.py

@@ -1,48 +1,41 @@
-from flask import Flask, request, jsonify
-import pandas as pd
-from sqlalchemy import create_engine
-import os
+import gradio as gr
+from datasets import load_dataset
 
-app = Flask(__name__)
+DATASET = "CodeXDevloper/MergedTgDataset"
 
-# Database connection
-DATABASE_URL = os.environ.get('DATABASE_URL', 'postgresql://user:password@localhost:5432/database')
-engine = create_engine(DATABASE_URL)
-
-@app.route('/user/<int:user_id>', methods=['GET'])
-def get_user_data(user_id):
-    """Endpoint to fetch user data by ID"""
-    try:
-        # Query database for specific user
-        query = f"SELECT * FROM your_table WHERE user_id = {user_id}"
-        df = pd.read_sql(query, engine)
-        
-        if df.empty:
-            return jsonify({'error': 'User not found'}), 404
-        
-        return jsonify(df.to_dict('records')), 200
+def search_uid(uid):
+    if not uid:
+        return {"error": "UID daalo"}
     
-    except Exception as e:
-        return jsonify({'error': str(e)}), 500
-
-@app.route('/user/<int:user_id>/csv', methods=['GET'])
-def get_user_csv(user_id):
-    """Endpoint to download user data as CSV"""
     try:
-        query = f"SELECT * FROM your_table WHERE user_id = {user_id}"
-        df = pd.read_sql(query, engine)
+        ds = load_dataset(DATASET, split="train", streaming=True)
         
-        if df.empty:
-            return jsonify({'error': 'User not found'}), 404
+        for row in ds:
+            if uid == str(row.get("user_id", "")).strip():
+                return {
+                    "found": True,
+                    "user_id": row.get("user_id"),
+                    "username": row.get("username"),
+                    "first_name": row.get("first_name"),
+                    "last_name": row.get("last_name"),
+                    "phone": row.get("phone"),
+                    "email": row.get("email"),
+                    "status": row.get("status"),
+                    "linked_id": row.get("linked_id"),
+                    "linked_name": row.get("linked_name"),
+                    "linked_handle": row.get("linked_handle"),
+                }
         
-        # Return as CSV
-        return df.to_csv(index=False), 200, {
-            'Content-Type': 'text/csv',
-            'Content-Disposition': f'attachment; filename=user_{user_id}_data.csv'
-        }
-    
+        return {"found": False, "message": f"UID '{uid}' nahi mila"}
     except Exception as e:
-        return jsonify({'error': str(e)}), 500
+        return {"error": str(e)}
+
+demo = gr.Interface(
+    fn=search_uid,
+    inputs=gr.Textbox(label="UID daalo", placeholder="385167100"),
+    outputs=gr.JSON(label="Result"),
+    title="Telegram UID Search",
+    description="UID daal ke poore dataset mein search karo"
+)
 
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=5000)
+demo.launch()