Update app.py

app.py

@@ -1,255 +1,43 @@
-import os
-import json
-import sqlite3
-import asyncio
-from datetime import datetime
-from fastapi import FastAPI, Query
-from fastapi.responses import JSONResponse
-import uvicorn
-from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
-from telegram.ext import Application, CommandHandler, CallbackQueryHandler, MessageHandler, filters, ContextTypes
 import aiohttp
-import threading
-
-# ========== FASTAPI APP ==========
-app = FastAPI(title="Telegram Search API + Bot", developer="Gopal Parmar")
-
-# ========== BOT CONFIG ==========
-BOT_TOKEN = "8902956864:AAH0eFvFemPm9inl5JiMLmu3aZ7ZT6hBFLs"
-OWNER_ID = 8501984482
-FILE_PATH = "/data/Telegram_70M.json"  # Mounted bucket path
-
-# ========== DATABASE ==========
-def init_db():
-    conn = sqlite3.connect('bot_data.db')
-    c = conn.cursor()
-    c.execute('''CREATE TABLE IF NOT EXISTS users (
-        user_id INTEGER PRIMARY KEY,
-        username TEXT,
-        first_name TEXT,
-        registered_at TIMESTAMP,
-        last_active TIMESTAMP,
-        total_searches INTEGER DEFAULT 0,
-        coins INTEGER DEFAULT 0
-    )''')
-    conn.commit()
-    conn.close()
-
-def save_user(user_id, username, first_name):
-    conn = sqlite3.connect('bot_data.db')
-    c = conn.cursor()
-    c.execute('INSERT OR IGNORE INTO users (user_id, username, first_name, registered_at, last_active) VALUES (?, ?, ?, ?, ?)',
-              (user_id, username, first_name, datetime.now(), datetime.now()))
-    c.execute('UPDATE users SET last_active = ? WHERE user_id = ?', (datetime.now(), user_id))
-    conn.commit()
-    conn.close()
-
-def add_coins(user_id, amount):
-    conn = sqlite3.connect('bot_data.db')
-    c = conn.cursor()
-    c.execute('UPDATE users SET coins = coins + ? WHERE user_id = ?', (amount, user_id))
-    conn.commit()
-    conn.close()
-
-def get_leaderboard(limit=10):
-    conn = sqlite3.connect('bot_data.db')
-    c = conn.cursor()
-    c.execute('SELECT user_id, first_name, coins FROM users ORDER BY coins DESC LIMIT ?', (limit,))
-    rows = c.fetchall()
-    conn.close()
-    return rows
-
-# ========== EMOJIS ==========
-E = {
-    "premium": "👑💎",
-    "search": "🔍✨",
-    "coin": "🪙✨",
-    "leader": "🏆🏅",
-    "refer": "🔗🎁",
-    "loading": "⏳🌀",
-    "error": "❌💔",
-    "success": "✅🎉",
-    "dev": "👨‍💻❤️",
-    "menu": "📋📌"
-}
-
-# ========== KEYBOARDS ==========
-def main_menu():
-    keyboard = [
-        [InlineKeyboardButton(f"{E['search']} Search", callback_data="menu_search")],
-        [InlineKeyboardButton(f"{E['coin']} Coins", callback_data="menu_coins")],
-        [InlineKeyboardButton(f"{E['leader']} Leaderboard", callback_data="menu_leader")],
-        [InlineKeyboardButton(f"{E['refer']} Refer", callback_data="menu_refer")],
-        [InlineKeyboardButton(f"{E['dev']} About", callback_data="menu_about")]
-    ]
-    return InlineKeyboardMarkup(keyboard)
-
-def search_menu():
-    keyboard = [
-        [InlineKeyboardButton("🔍 By Phone", callback_data="search_phone")],
-        [InlineKeyboardButton("👤 By Username", callback_data="search_username")],
-        [InlineKeyboardButton("🔙 Back", callback_data="menu_back")]
-    ]
-    return InlineKeyboardMarkup(keyboard)
-
-# ========== API SEARCH ==========
-@app.get("/")
-async def root():
-    return {
-        "message": "🔍 Telegram 70M Search API + Bot",
-        "developer": "Gopal Parmar",
-        "file_exists": os.path.exists(FILE_PATH),
-        "endpoints": {
-            "/search?q=79254677980": "Search by phone/username",
-            "/stats": "Database statistics"
-        }
-    }
-
-@app.get("/stats")
-async def stats():
-    if not os.path.exists(FILE_PATH):
-        return {"error": "File not mounted", "developer": "Gopal Parmar"}
-    size_gb = os.path.getsize(FILE_PATH) / (1024**3)
-    return {"developer": "Gopal Parmar", "file": "Telegram_70M.json", "size_gb": round(size_gb, 2), "status": "ready"}
+import asyncio
 
-@app.get("/search")
-async def search(q: str = Query(..., min_length=2), limit: int = 10):
-    if not os.path.exists(FILE_PATH):
-        return {"error": "File not mounted", "developer": "Gopal Parmar"}
-    
-    results = []
-    try:
-        with open(FILE_PATH, 'r', encoding='utf-8', errors='ignore') as f:
-            for line in f:
-                if q in line:
-                    try:
-                        results.append(json.loads(line))
-                    except:
-                        results.append({"raw": line.strip()})
-                    if len(results) >= limit:
-                        break
-    except Exception as e:
-        return {"error": str(e), "developer": "Gopal Parmar"}
-    
-    return {"developer": "Gopal Parmar", "query": q, "count": len(results), "results": results}
+API_URL = "https://gopallikehack-Tgjson123.hf.space/search?q="
 
-# ========== BOT HANDLERS ==========
-async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    user = update.effective_user
-    save_user(user.id, user.username, user.first_name)
-    await update.message.reply_text(
-        f"{E['premium']} *ULTRA PREMIUM BOT* {E['premium']}\n\n"
-        f"Welcome {user.first_name}!\n\n🔍 Search 70M+ users\n💰 Earn coins\n🎁 Refer & earn\n🏆 Leaderboard\n\n👇 *Select:*",
-        parse_mode='Markdown',
-        reply_markup=main_menu()
-    )
+async def search_user(query):
+    async with aiohttp.ClientSession() as session:
+        async with session.get(f"{API_URL}{query}&limit=5") as resp:
+            data = await resp.json()
+            return data
 
-async def button_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    query = update.callback_query
-    await query.answer()
-    data = query.data
-    user_id = query.from_user.id
-    save_user(user_id, query.from_user.username, query.from_user.first_name)
-    
-    if data == "menu_back":
-        await query.message.edit_text("📋 *Main Menu*", parse_mode='Markdown', reply_markup=main_menu())
-    
-    elif data == "menu_search":
-        await query.message.edit_text("🔍 *Search Options*", parse_mode='Markdown', reply_markup=search_menu())
-    
-    elif data == "menu_coins":
-        conn = sqlite3.connect('bot_data.db')
-        c = conn.cursor()
-        c.execute('SELECT coins FROM users WHERE user_id = ?', (user_id,))
-        row = c.fetchone()
-        conn.close()
-        coins = row[0] if row else 0
-        await query.message.edit_text(
-            f"{E['coin']} *Your Coins*\n\n🪙 Total: `{coins}`\n\nEarn by searching!",
-            parse_mode='Markdown',
-            reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("🔙 Back", callback_data="menu_back")]])
-        )
-    
-    elif data == "menu_leader":
-        leaders = get_leaderboard(10)
-        text = f"{E['leader']} *LEADERBOARD*\n\n"
-        for idx, (uid, name, coins) in enumerate(leaders, 1):
-            medal = "🥇" if idx == 1 else "🥈" if idx == 2 else "🥉" if idx == 3 else f"{idx}."
-            text += f"{medal} `{name[:15]}` — 🪙 {coins}\n"
-        await query.message.edit_text(
-            text,
-            parse_mode='Markdown',
-            reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("🔙 Back", callback_data="menu_back")]])
-        )
+def format_result(data):
+    if data.get("count", 0) == 0:
+        return "❌ No results found"
     
-    elif data == "menu_refer":
-        bot_username = (await context.bot.get_me()).username
-        ref_link = f"https://t.me/{bot_username}?start={user_id}"
-        await query.message.edit_text(
-            f"{E['refer']} *Refer & Earn*\n\nShare:\n`{ref_link}`\n\nYou get 10 coins per referral!",
-            parse_mode='Markdown',
-            reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("🔙 Back", callback_data="menu_back")]])
-        )
+    text = f"🔍 *Results for* `{data['query']}`\n✅ Found: `{data['count']}`\n\n"
     
-    elif data == "menu_about":
-        await query.message.edit_text(
-            f"{E['dev']} *About Bot*\n\n👨‍💻 Developer: Gopal Parmar\n📊 Database: 70M+\n🆘 Support: @Kingx18rcb",
-            parse_mode='Markdown',
-            reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("🔙 Back", callback_data="menu_back")]])
-        )
+    for idx, rec in enumerate(data.get("results", [])[:5], 1):
+        text += f"*📌 Result {idx}*\n"
+        text += f"🆔 ID: `{rec.get('account_id') or rec.get('user_id') or 'N/A'}`\n"
+        text += f"📞 Phone: `{rec.get('phone') or 'N/A'}`\n"
+        text += f"👤 Username: @{rec.get('username') or 'N/A'}\n"
+        text += f"📝 Name: {rec.get('first_name') or ''} {rec.get('last_name') or ''}\n\n"
     
-    elif data.startswith("search_"):
-        context.user_data['search_type'] = data.replace("search_", "")
-        await query.message.edit_text(
-            f"Send *{data.replace('search_', '').upper()}* to search.\nExample: `79254677980`",
-            parse_mode='Markdown'
-        )
+    return text
 
-async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    user_id = update.effective_user.id
-    save_user(user_id, update.effective_user.username, update.effective_user.first_name)
+async def main():
+    print("=" * 40)
+    print("🔍 SEARCH BY USER ID OR USERNAME")
+    print("=" * 40)
     
-    query_text = update.message.text.strip()
-    await update.message.reply_text(f"{E['loading']} Searching...")
+    query = input("Enter User ID or Username: ").strip()
     
-    async with aiohttp.ClientSession() as session:
-        async with session.get(f"https://gopallikehack-Tgjson123.hf.space/search?q={query_text}&limit=10") as resp:
-            data = await resp.json()
-    
-    add_coins(user_id, 1)
-    
-    if data.get("count", 0) == 0:
-        await update.message.reply_text(f"{E['error']} No results")
+    if not query:
+        print("❌ Please enter something")
         return
     
-    text = f"{E['premium']} *RESULTS*\n🔍 `{query_text}`\n✅ Found: `{data['count']}`\n🪙 +1 coin\n\n"
-    for idx, rec in enumerate(data.get("results", [])[:5], 1):
-        text += f"*📌 Result {idx}*\n"
-        text += f"🆔 `{rec.get('account_id') or rec.get('user_id') or 'N/A'}`\n"
-        text += f"📞 `{rec.get('phone') or 'N/A'}`\n"
-        text += f"👤 @{rec.get('username') or 'N/A'}\n"
-        text += f"📝 {rec.get('first_name') or ''} {rec.get('last_name') or ''}\n\n"
-    
-    text += f"\n👨‍💻 Gopal Parmar"
-    await update.message.reply_text(text, parse_mode='Markdown')
+    print(f"\n⏳ Searching for `{query}`...\n")
+    result = await search_user(query)
+    print(format_result(result))
 
-# ========== RUN BOT IN BACKGROUND ==========
-def run_bot():
-    init_db()
-    bot_app = Application.builder().token(BOT_TOKEN).build()
-    bot_app.add_handler(CommandHandler("start", start))
-    bot_app.add_handler(CallbackQueryHandler(button_handler))
-    bot_app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
-    print("🤖 Bot running...")
-    bot_app.run_polling()
-
-# ========== MAIN ==========
 if __name__ == "__main__":
-    # Start bot in background thread
-    bot_thread = threading.Thread(target=run_bot, daemon=True)
-    bot_thread.start()
-    
-    # Run FastAPI
-    print("🚀 API running on http://0.0.0.0:7860")
-    uvicorn.run(app, host="0.0.0.0", port=7860)
-    
+    asyncio.run(main())