Update app.py

app.py

@@ -3,15 +3,19 @@ BOT_TOKEN = "6740023413:AAHZ5iEiXjkY7IGwljg37llFbekKLbIVkkw"
 
 # 🔴 HARD CODED VALUES
 GROUP_ID = -1003799294466
-
 import json
 import time
 from datetime import datetime
-from telegram import Update, InputFile
+from telegram import (
+    Update,
+    InlineKeyboardButton,
+    InlineKeyboardMarkup
+)
 from telegram.ext import (
     ApplicationBuilder,
     MessageHandler,
     CommandHandler,
+    CallbackQueryHandler,
     ContextTypes,
     filters
 )
@@ -20,16 +24,15 @@ from telegram.ext import (
 waiting_for_file = set()
 
 
-# 🔍 FIND USER JSON FILE FROM GROUP
+# 🔍 FIND USER JSON IN 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
+    async for msg in context.bot.get_chat_history(GROUP_ID, limit=200):
+        if msg.document and msg.document.file_name == f"{user_id}.json":
+            return msg.document.file_id
     return None
 
 
-# 📥 LOAD JSON FROM TELEGRAM (NO LOCAL FILE)
+# 📥 LOAD JSON FROM GROUP
 async def load_data(context, user_id):
     file_id = await find_user_json(context, user_id)
 
@@ -41,16 +44,14 @@ async def load_data(context, user_id):
 
     try:
         data = json.loads(content.decode("utf-8"))
-
         if "files" not in data:
             return {"files": []}
-
         return data
     except:
         return {"files": []}
 
 
-# 📤 SAVE JSON BACK TO GROUP
+# 📤 SAVE JSON TO GROUP
 async def save_data(context, user_id, data):
     json_bytes = json.dumps(data, indent=4).encode("utf-8")
 
@@ -62,6 +63,49 @@ async def save_data(context, user_id, data):
     )
 
 
+# 🚀 START
+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")]
+    ]
+
+    await update.message.reply_text(
+        "🚀 Cloud Storage Bot (No Server Storage ☁️)",
+        reply_markup=InlineKeyboardMarkup(keyboard)
+    )
+
+
+# 🔘 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 to upload")
+
+    elif query.data == "files":
+        data = await load_data(context, user_id)
+
+        if not data["files"]:
+            await query.message.reply_text("❌ No files")
+            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 = await load_data(context, 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)
@@ -109,10 +153,10 @@ async def handle_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
 
         data["files"].append(entry)
 
-        # ✅ STEP 3: save JSON BACK to group
+        # ✅ STEP 3: save JSON back to group
         await save_data(context, user_id, data)
 
-        await update.message.reply_text("✅ File saved (cloud mode ON ☁️)")
+        await update.message.reply_text("✅ File saved successfully ☁️")
 
     except Exception as e:
         print(e)
@@ -127,12 +171,12 @@ async def myfiles(update: Update, context: ContextTypes.DEFAULT_TYPE):
     data = await load_data(context, user_id)
 
     if not data["files"]:
-        await update.message.reply_text("❌ No files")
+        await update.message.reply_text("❌ No files found")
         return
 
-    msg = "📂 Files:\n\n"
+    msg = "📂 Your Files:\n\n"
     for f in data["files"]:
-        msg += f"• {f['file_name']}\n"
+        msg += f"• {f['file_name']} ({f['upload_date']})\n"
 
     await update.message.reply_text(msg)
 
@@ -156,7 +200,7 @@ async def get_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
             )
             return
 
-    await update.message.reply_text("❌ Not found")
+    await update.message.reply_text("❌ File not found")
 
 
 # 📊 /stats
@@ -172,15 +216,17 @@ async def stats(update: Update, context: ContextTypes.DEFAULT_TYPE):
     )
 
 
-# 🚀 RUN
+# 🚀 RUN BOT
 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("stats", stats))
 
+app.add_handler(CallbackQueryHandler(buttons))
 app.add_handler(MessageHandler(filters.Document.ALL, handle_file))
 
-print("🔥 CLOUD MODE BOT RUNNING...")
+print("🔥 CLOUD STORAGE BOT RUNNING...")
 app.run_polling()