Update app.py

app.py

@@ -1,125 +1,86 @@
+
+BOT_TOKEN = "6740023413:AAHZ5iEiXjkY7IGwljg37llFbekKLbIVkkw"
+
+# 🔴 HARD CODED VALUES
+GROUP_ID = -1003799294466
+
 import json
-import os
 import time
 from datetime import datetime
-from telegram import (
-    Update,
-    InputFile,
-    InlineKeyboardButton,
-    InlineKeyboardMarkup
-)
+from telegram import Update, InputFile
 from telegram.ext import (
     ApplicationBuilder,
     MessageHandler,
     CommandHandler,
-    CallbackQueryHandler,
     ContextTypes,
     filters
 )
 
-BOT_TOKEN = "6740023413:AAHZ5iEiXjkY7IGwljg37llFbekKLbIVkkw"
-
-# 🔴 HARD CODED VALUES
-GROUP_ID = -1003799294466
-
 
 waiting_for_file = set()
 
 
-# 📁 JSON file name = userID.json
-def get_json_file(user_id):
-    return f"{user_id}.json"
+# 🔍 FIND USER JSON FILE FROM GROUP
+async def find_user_json(context, user_id):
+    async for msg in context.bot.get_chat_history(GROUP_ID, limit=100):
+        if msg.document:
+            if msg.document.file_name == f"{user_id}.json":
+                return msg.document.file_id
+    return None
 
 
-# 📥 LOAD JSON (safe)
-def load_data(user_id):
-    file = get_json_file(user_id)
+# 📥 LOAD JSON FROM TELEGRAM (NO LOCAL FILE)
+async def load_data(context, user_id):
+    file_id = await find_user_json(context, user_id)
 
-    if os.path.exists(file):
-        try:
-            with open(file, "r") as f:
-                data = json.load(f)
+    if not file_id:
+        return {"files": []}
 
-                if "files" not in data or not isinstance(data["files"], list):
-                    return {"files": []}
+    file = await context.bot.get_file(file_id)
+    content = await file.download_as_bytearray()
 
-                return data
-        except:
-            return {"files": []}
-
-    return {"files": []}
+    try:
+        data = json.loads(content.decode("utf-8"))
 
+        if "files" not in data:
+            return {"files": []}
 
-# 💾 SAVE JSON
-def save_data(user_id, data):
-    with open(get_json_file(user_id), "w") as f:
-        json.dump(data, f, indent=4)
+        return data
+    except:
+        return {"files": []}
 
 
-# 🚀 START UI
-async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    keyboard = [
-        [InlineKeyboardButton("📤 Upload", callback_data="upload")],
-        [InlineKeyboardButton("📂 My Files", callback_data="files")],
-        [InlineKeyboardButton("📊 Stats", callback_data="stats")]
-    ]
+# 📤 SAVE JSON BACK TO GROUP
+async def save_data(context, user_id, data):
+    json_bytes = json.dumps(data, indent=4).encode("utf-8")
 
-    await update.message.reply_text(
-        "🚀 Cloud Storage Bot",
-        reply_markup=InlineKeyboardMarkup(keyboard)
+    await context.bot.send_document(
+        chat_id=GROUP_ID,
+        document=json_bytes,
+        filename=f"{user_id}.json",
+        caption=f"📁 Updated Index {user_id}"
     )
 
 
-# 🔘 BUTTON HANDLER
-async def buttons(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    query = update.callback_query
-    await query.answer()
-
-    user_id = query.from_user.id
-
-    if query.data == "upload":
-        waiting_for_file.add(user_id)
-        await query.message.reply_text("📁 Send file(s)...")
-
-    elif query.data == "files":
-        data = load_data(user_id)
-
-        if not data["files"]:
-            await query.message.reply_text("❌ No files found")
-            return
-
-        msg = "📂 Your Files:\n\n"
-        for f in data["files"][-10:]:
-            msg += f"• {f['file_name']}\n"
-
-        await query.message.reply_text(msg)
-
-    elif query.data == "stats":
-        data = load_data(user_id)
-        await query.message.reply_text(f"📊 Total files: {len(data['files'])}")
-
-
 # 🟢 /save
 async def save_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
     waiting_for_file.add(update.effective_user.id)
     await update.message.reply_text("📁 Send file to save")
 
 
-# 📤 FILE HANDLER (MAIN LOGIC)
+# 📤 FILE HANDLER
 async def handle_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    user = update.effective_user
-    user_id = user.id
+    user_id = update.effective_user.id
 
     if user_id not in waiting_for_file:
         return
 
-    document = update.message.document
-    if not document:
+    doc = update.message.document
+    if not doc:
         return
 
     try:
-        # file info
-        original_name = document.file_name
+        original_name = doc.file_name
         ext = original_name.split('.')[-1] if '.' in original_name else "dat"
 
         timestamp = int(time.time())
@@ -130,56 +91,48 @@ async def handle_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
         # ✅ STEP 1: send file FIRST
         sent = await context.bot.send_document(
             chat_id=GROUP_ID,
-            document=document.file_id,
+            document=doc.file_id,
             filename=new_name
         )
 
-        # ✅ STEP 2: JSON update
-        data = load_data(user_id)
+        # ✅ STEP 2: load JSON from group
+        data = await load_data(context, user_id)
 
         entry = {
             "file_name": new_name,
             "original_name": original_name,
             "file_id": sent.document.file_id,
-            "file_size": document.file_size,
+            "file_size": doc.file_size,
             "upload_date": upload_date,
             "timestamp": timestamp
         }
 
         data["files"].append(entry)
-        save_data(user_id, data)
 
-        # ✅ STEP 3: send JSON with correct filename
-        await context.bot.send_document(
-            chat_id=GROUP_ID,
-            document=InputFile(
-                get_json_file(user_id),
-                filename=f"{user_id}.json"
-            ),
-            caption=f"📁 Index of {user_id}"
-        )
+        # ✅ STEP 3: save JSON BACK to group
+        await save_data(context, user_id, data)
 
-        await update.message.reply_text("✅ File saved successfully!")
+        await update.message.reply_text("✅ File saved (cloud mode ON ☁️)")
 
     except Exception as e:
-        print("ERROR:", e)
+        print(e)
         await update.message.reply_text(f"❌ Error: {e}")
 
     waiting_for_file.discard(user_id)
 
 
-# 📋 /myfiles
+# 📂 /myfiles
 async def myfiles(update: Update, context: ContextTypes.DEFAULT_TYPE):
     user_id = update.effective_user.id
-    data = load_data(user_id)
+    data = await load_data(context, user_id)
 
     if not data["files"]:
-        await update.message.reply_text("❌ No files found")
+        await update.message.reply_text("❌ No files")
         return
 
-    msg = "📂 Your Files:\n\n"
+    msg = "📂 Files:\n\n"
     for f in data["files"]:
-        msg += f"• {f['file_name']} ({f['upload_date']})\n"
+        msg += f"• {f['file_name']}\n"
 
     await update.message.reply_text(msg)
 
@@ -187,7 +140,7 @@ async def myfiles(update: Update, context: ContextTypes.DEFAULT_TYPE):
 # 📥 /get
 async def get_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
     user_id = update.effective_user.id
-    data = load_data(user_id)
+    data = await load_data(context, user_id)
 
     if not context.args:
         await update.message.reply_text("Usage: /get filename")
@@ -203,60 +156,13 @@ async def get_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
             )
             return
 
-    await update.message.reply_text("❌ File not found")
-
-
-# 🗑️ /delete
-async def delete_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    user_id = update.effective_user.id
-    data = load_data(user_id)
-
-    if not context.args:
-        await update.message.reply_text("Usage: /delete filename")
-        return
-
-    filename = context.args[0]
-
-    new_files = [f for f in data["files"] if f["file_name"] != filename]
-
-    if len(new_files) == len(data["files"]):
-        await update.message.reply_text("❌ File not found")
-        return
-
-    data["files"] = new_files
-    save_data(user_id, data)
-
-    await update.message.reply_text("🗑️ File deleted")
-
-
-# 🔎 /search
-async def search(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    user_id = update.effective_user.id
-    data = load_data(user_id)
-
-    if not context.args:
-        await update.message.reply_text("Usage: /search keyword")
-        return
-
-    keyword = context.args[0].lower()
-
-    results = [f for f in data["files"] if keyword in f["file_name"].lower()]
-
-    if not results:
-        await update.message.reply_text("❌ No results")
-        return
-
-    msg = "🔎 Results:\n\n"
-    for f in results:
-        msg += f"• {f['file_name']}\n"
-
-    await update.message.reply_text(msg)
+    await update.message.reply_text("❌ Not found")
 
 
 # 📊 /stats
 async def stats(update: Update, context: ContextTypes.DEFAULT_TYPE):
     user_id = update.effective_user.id
-    data = load_data(user_id)
+    data = await load_data(context, user_id)
 
     total = len(data["files"])
     total_size = sum(f["file_size"] for f in data["files"])
@@ -266,19 +172,15 @@ async def stats(update: Update, context: ContextTypes.DEFAULT_TYPE):
     )
 
 
-# 🚀 RUN BOT
+# 🚀 RUN
 app = ApplicationBuilder().token(BOT_TOKEN).build()
 
-app.add_handler(CommandHandler("start", start))
 app.add_handler(CommandHandler("save", save_command))
 app.add_handler(CommandHandler("myfiles", myfiles))
 app.add_handler(CommandHandler("get", get_file))
-app.add_handler(CommandHandler("delete", delete_file))
-app.add_handler(CommandHandler("search", search))
 app.add_handler(CommandHandler("stats", stats))
 
-app.add_handler(CallbackQueryHandler(buttons))
 app.add_handler(MessageHandler(filters.Document.ALL, handle_file))
 
-print("🔥 FINAL BOT RUNNING...")
+print("🔥 CLOUD MODE BOT RUNNING...")
 app.run_polling()