Update app.py

app.py

@@ -1,20 +1,32 @@
 import os
+import sys
 import asyncio
-import tempfile
+import logging
 from pathlib import Path
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI
 from fastapi.responses import RedirectResponse
 from huggingface_hub import HfApi
 from telethon import TelegramClient, events
-from telethon.tl.types import DocumentAttributeVideo, DocumentAttributeAudio
-
-# --- Configuration from Environment Variables ---
-HF_TOKEN = os.environ["HF_TOKEN"]
-HF_REPO = os.environ["HF_REPO"]  # Example: username/dataset-name
-API_ID = int(os.environ["TG_API_ID"])
-API_HASH = os.environ["TG_API_HASH"]
-BOT_TOKEN = os.environ["TG_BOT_TOKEN"]
-SESSION_NAME = "my_session"
+from telethon.sessions import MemorySession
+
+# --- Logging Setup for HF Space ---
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    handlers=[logging.StreamHandler(sys.stdout)]
+)
+logger = logging.getLogger("MyCloudBot")
+
+# --- Configuration ---
+HF_TOKEN = os.environ.get("HF_TOKEN")
+HF_REPO = os.environ.get("HF_REPO")
+API_ID = int(os.environ.get("TG_API_ID", 0))
+API_HASH = os.environ.get("TG_API_HASH", "")
+BOT_TOKEN = os.environ.get("TG_BOT_TOKEN", "")
+
+if not all([HF_TOKEN, HF_REPO, API_ID, API_HASH, BOT_TOKEN]):
+    logger.error(" Missing environment variables! Check HF Space Secrets.")
+    raise ValueError("Missing required secrets")
 
 # Initialize Hugging Face API
 hf_api = HfApi(token=HF_TOKEN)
@@ -22,71 +34,104 @@ hf_api = HfApi(token=HF_TOKEN)
 # Initialize FastAPI
 app = FastAPI(title="My Cloud Bot")
 
-# Initialize Telethon Client (MTProto)
-client = TelegramClient(SESSION_NAME, API_ID, API_HASH)
+# Initialize Telethon Client with MemorySession (No file persistence needed)
+client = TelegramClient(MemorySession(), API_ID, API_HASH)
 
-# Helper: Get public CDN URL
 def get_cdn_url(filename):
     return f"https://huggingface.co/datasets/{HF_REPO}/resolve/main/{filename}"
 
-# --- Telegram Bot Logic ---
+# --- Telegram Bot Handlers ---
 
 @client.on(events.NewMessage(pattern='/start'))
 async def start_handler(event):
-    await event.reply("👋 Welcome to My Cloud Bot!\n\nSend me any file (up to 4GB) and I will upload it to the cloud and give you a permanent link.")
+    await event.reply(
+        "👋 *Welcome to My Cloud Bot!*\n\n"
+        "Send me any file and I will upload it to Hugging Face.\n"
+        "You'll get a permanent CDN link instantly.",
+        parse_mode='markdown'
+    )
 
 @client.on(events.NewMessage)
 async def handle_file(event):
+    # Ignore commands
+    if event.message.text and event.message.text.startswith('/'):
+        return
+        
     if not event.message.file:
         return
 
-    message = await event.reply("📥 Downloading file from Telegram...")
+    message = await event.reply("📥 *Downloading from Telegram...*", parse_mode='markdown')
+    file_name = event.message.file.name or "unknown_file"
     
     try:
-        # 1. Download File using Telethon (Supports large files)
-        file_path = await client.download_media(event.message.file)
+        logger.info(f"📥 Starting download: {file_name}")
+        
+        # Download with progress
+        def progress(current, total):
+            percent = (current / total) * 100
+            if int(percent) % 20 == 0:  # Log every 20%
+                logger.info(f" Download progress: {percent:.1f}%")
+                
+        file_path = await client.download_media(
+            event.message.file,
+            progress_callback=progress
+        )
         
         if not file_path:
-            await message.edit("❌ Failed to download file.")
-            return
-
-        filename = Path(file_path).name
+            raise Exception("download_media returned None - File might be too large or restricted")
+            
+        logger.info(f"✅ Download complete: {file_path}")
         
-        # Update progress
-        await message.edit(f"📤 Uploading {filename} to Hugging Face...")
-
-        # 2. Upload to Hugging Face Dataset
+        # Update status
+        await message.edit(f"📤 *Uploading {file_name} to Cloud...*", parse_mode='markdown')
+        
+        # Upload to HF Dataset
         hf_api.upload_file(
             path_or_fileobj=file_path,
-            path_in_repo=filename,
+            path_in_repo=file_name,
             repo_id=HF_REPO,
             repo_type="dataset",
-            commit_message=f"Upload via Telegram: {filename}"
+            commit_message=f"Upload via Telegram: {file_name}"
         )
-
-        # 3. Generate CDN URL
-        cdn_link = get_cdn_url(filename)
-
-        # 4. Reply with Link
+        
+        cdn_link = get_cdn_url(file_name)
+        file_size_mb = os.path.getsize(file_path) / (1024*1024)
+        
         response_text = (
-            f"✅ **Upload Complete!**\n\n"
-            f"📁 **File:** `{filename}`\n"
-            f"📦 **Size:** {os.path.getsize(file_path) / (1024*1024):.2f} MB\n\n"
-            f"🔗 **CDN Link:**\n{cdn_link}\n\n"
-            f"_You can use this link anywhere._"
+            f"✅ *Upload Complete!*\n\n"
+            f"📁 *File:* `{file_name}`\n"
+            f"📦 *Size:* {file_size_mb:.2f} MB\n\n"
+            f"🔗 *CDN Link:*\n{cdn_link}\n\n"
+            f"_Link is public and permanent._"
         )
         
         await message.edit(response_text, parse_mode='markdown')
-
-        # Cleanup local file
+        logger.info(f"🔗 Link generated: {cdn_link}")
+        
+        # Cleanup
         os.remove(file_path)
+        logger.info(f"🧹 Cleaned up local file: {file_name}")
 
     except Exception as e:
-        await message.edit(f"❌ Error: {str(e)}")
-        if 'file_path' in locals() and os.path.exists(file_path):
+        error_type = type(e).__name__
+        error_msg = str(e)[:400]
+        
+        logger.error(f"❌ FAILED [{error_type}]: {error_msg}")
+        
+        safe_error = error_msg.replace("`", "'")
+        await message.edit(
+            f"❌ *Operation Failed*\n\n"
+            f"*Type:* `{error_type}`\n"
+            f"*Details:* `{safe_error}`\n\n"
+            f"_Check HF Space Logs for more info._",
+            parse_mode='markdown'
+        )
+        
+        # Cleanup on error
+        if 'file_path' in locals() and file_path and os.path.exists(file_path):
             os.remove(file_path)
 
-# --- FastAPI Endpoints (Optional for Web Interface) ---
+# --- FastAPI Endpoints ---
 
 @app.get("/")
 def home():
@@ -98,17 +143,21 @@ def health():
 
 @app.get("/cdn/{filename:path}")
 def cdn_redirect(filename: str):
-    url = get_cdn_url(filename)
-    return RedirectResponse(url)
+    return RedirectResponse(get_cdn_url(filename))
 
-# --- Startup Event to Run Bot ---
+# --- Startup & Shutdown ---
 
 @app.on_event("startup")
 async def startup_event():
-    print("Starting Telegram Bot...")
-    await client.start(bot_token=BOT_TOKEN)
-    print("Bot is running!")
+    logger.info(" Starting My Cloud Bot...")
+    try:
+        await client.start(bot_token=BOT_TOKEN)
+        logger.info("✅ Telegram Bot connected successfully!")
+    except Exception as e:
+        logger.error(f"❌ Failed to start Telegram client: {e}")
+        raise
 
 @app.on_event("shutdown")
 async def shutdown_event():
+    logger.info("🛑 Shutting down bot...")
     await client.disconnect()