Update app.py

app.py

@@ -4,81 +4,49 @@ BOT_TOKEN = "6740023413:AAHZ5iEiXjkY7IGwljg37llFbekKLbIVkkw"
 # 🔴 HARD CODED VALUES
 GROUP_ID = -1003799294466
 
-
 import json
 import os
 import time
-from telegram import Update, InputFile, InlineKeyboardButton, InlineKeyboardMarkup
+from datetime import datetime
+from telegram import Update, InputFile
 from telegram.ext import (
     ApplicationBuilder,
     MessageHandler,
     CommandHandler,
-    CallbackQueryHandler,
     ContextTypes,
     filters
 )
 
+BOT_TOKEN = "YOUR_BOT_TOKEN"
+GROUP_ID = -5111627225
 
 waiting_for_file = set()
 
 
+# 📁 JSON file path
 def get_json_file(user_id):
-    return f"{user_id}_savedFiles.json"
+    return f"{user_id}.json"
 
 
+# 📥 load JSON
 def load_data(user_id):
     file = get_json_file(user_id)
     if os.path.exists(file):
         with open(file, "r") as f:
             return json.load(f)
-    return []
+    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)
 
 
-# 🚀 START MENU
-async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    keyboard = [
-        [InlineKeyboardButton("📤 Upload File", callback_data="upload")],
-        [InlineKeyboardButton("📂 My Files", callback_data="myfiles")],
-        [InlineKeyboardButton("📊 Stats", callback_data="stats")]
-    ]
-    await update.message.reply_text(
-        "🚀 Welcome to Cloud Bot",
-        reply_markup=InlineKeyboardMarkup(keyboard)
-    )
-
-
-# 🔘 BUTTON HANDLER
-async def button_handler(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) now...")
-
-    elif query.data == "myfiles":
-        data = load_data(user_id)
-
-        if not data:
-            await query.message.reply_text("❌ No files found")
-            return
-
-        msg = "📂 Your Files:\n\n"
-        for f in data[-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)}")
+# 🟢 /save command
+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
@@ -94,107 +62,100 @@ async def handle_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
         return
 
     try:
-        ext = document.file_name.split('.')[-1] if '.' in document.file_name else "dat"
+        # file info
+        original_name = document.file_name
+        ext = original_name.split('.')[-1] if '.' in original_name else "dat"
+
         timestamp = int(time.time())
+        upload_date = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
         new_name = f"{user_id}_{timestamp}.{ext}"
 
+        # 1️⃣ send file FIRST
         sent = await context.bot.send_document(
             chat_id=GROUP_ID,
             document=document.file_id,
-            filename=new_name,
-            caption=f"📂 {user.username or user_id}"
+            filename=new_name
         )
 
+        # 2️⃣ JSON load/update
         data = load_data(user_id)
 
-        data.append({
+        file_entry = {
             "file_name": new_name,
+            "original_name": original_name,
             "file_id": sent.document.file_id,
+            "file_size": document.file_size,
+            "upload_date": upload_date,
             "timestamp": timestamp
-        })
-
-        save_data(user_id, data)
+        }
 
-        await update.message.reply_text("✅ Saved!")
+        data["files"].append(file_entry)
 
-    except Exception as e:
-        await update.message.reply_text(f"❌ {e}")
-
-
-# 📥 GET FILE
-async def get_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    user_id = update.effective_user.id
+        save_data(user_id, data)
 
-    if not context.args:
-        await update.message.reply_text("Usage: /get filename")
-        return
+        # 3️⃣ send JSON AFTER file
+        await context.bot.send_document(
+            chat_id=GROUP_ID,
+            document=InputFile(get_json_file(user_id)),
+            filename=f"{user_id}.json",
+            caption=f"📁 File index for user {user_id}"
+        )
 
-    filename = context.args[0]
-    data = load_data(user_id)
+        await update.message.reply_text("✅ File saved & indexed!")
 
-    for f in data:
-        if f["file_name"] == filename:
-            await context.bot.send_document(update.effective_chat.id, f["file_id"])
-            return
+    except Exception as e:
+        print("ERROR:", e)
+        await update.message.reply_text(f"❌ Error: {e}")
 
-    await update.message.reply_text("❌ Not found")
+    waiting_for_file.discard(user_id)
 
 
-# 🗑️ DELETE FILE
-async def delete_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
+# 📋 /myfiles
+async def myfiles(update: Update, context: ContextTypes.DEFAULT_TYPE):
     user_id = update.effective_user.id
-
-    if not context.args:
-        await update.message.reply_text("Usage: /delete filename")
-        return
-
-    filename = context.args[0]
     data = load_data(user_id)
 
-    new_data = [f for f in data if f["file_name"] != filename]
-
-    if len(new_data) == len(data):
-        await update.message.reply_text("❌ File not found")
+    if not data["files"]:
+        await update.message.reply_text("❌ No files found")
         return
 
-    save_data(user_id, new_data)
-    await update.message.reply_text("🗑️ Deleted")
+    msg = "📂 Your Files:\n\n"
+    for f in data["files"][-10:]:
+        msg += f"• {f['file_name']} ({f['upload_date']})\n"
+
+    await update.message.reply_text(msg)
 
 
-# 🔎 SEARCH
-async def search(update: Update, context: ContextTypes.DEFAULT_TYPE):
+# 📥 /get filename
+async def get_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: /search keyword")
+        await update.message.reply_text("Usage: /get filename")
         return
 
-    keyword = context.args[0].lower()
-    data = load_data(user_id)
-
-    results = [f for f in data if keyword in f["file_name"].lower()]
-
-    if not results:
-        await update.message.reply_text("❌ No results")
-        return
+    filename = context.args[0]
 
-    msg = "🔎 Results:\n\n"
-    for f in results:
-        msg += f"• {f['file_name']}\n"
+    for f in data["files"]:
+        if f["file_name"] == filename:
+            await context.bot.send_document(
+                chat_id=update.effective_chat.id,
+                document=f["file_id"]
+            )
+            return
 
-    await update.message.reply_text(msg)
+    await update.message.reply_text("❌ File not found")
 
 
 # 🚀 BOT START
 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(CallbackQueryHandler(button_handler))
 app.add_handler(MessageHandler(filters.Document.ALL, handle_file))
 
-print("🔥 OP BOT RUNNING...")
+print("🔥 BOT RUNNING...")
 app.run_polling()