Update app.py

app.py

@@ -3,28 +3,38 @@ import asyncio
 from telethon import TelegramClient, events
 
 # ===============================
-# Telegram credentials
+# Telegram credentials (ENV)
 # ===============================
-API_ID = int(os.environ.get("API_ID"))
-API_HASH = os.environ.get("API_HASH")
-BOT_TOKEN = os.environ.get("BOT_TOKEN")
+API_ID = int(os.environ["API_ID"])
+API_HASH = os.environ["API_HASH"]
+BOT_TOKEN = os.environ["BOT_TOKEN"]
 
 # ===============================
-# Persistent storage path
+# Persistent storage
 # ===============================
 UPLOAD_DIR = "/data/uploads"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 # ===============================
-# Initialize bot client
+# Telegram Client (IPv4 forced)
 # ===============================
-client = TelegramClient("bot", API_ID, API_HASH)
+client = TelegramClient(
+    "bot",
+    API_ID,
+    API_HASH,
+    connection_retries=10,
+    timeout=30,
+    use_ipv6=False
+)
 
+# ===============================
+# Handlers
+# ===============================
 @client.on(events.NewMessage(pattern="/start"))
 async def start_handler(event):
     await event.reply(
-        "👋 سلام!\n"
-        "هر فایلی بفرستی، مستقیم روی HuggingFace Spaces ذخیره می‌کنم."
+        "🤖 Bot is online.\n"
+        "هر فایلی بفرستی، روی HuggingFace ذخیره می‌کنم."
     )
 
 @client.on(events.NewMessage)
@@ -35,18 +45,22 @@ async def file_handler(event):
     try:
         file_path = await event.download_media(file=UPLOAD_DIR)
         await event.reply(
-            "✅ فایل با موفقیت ذخیره شد\n\n"
+            "✅ فایل ذخیره شد\n\n"
             f"📂 مسیر:\n{file_path}"
         )
-        print(f"Saved file: {file_path}")
+        print(f"Saved: {file_path}")
 
     except Exception as e:
-        await event.reply(f"❌ خطا در ذخیره فایل:\n{e}")
-        print(e)
+        await event.reply(f"❌ خطا:\n{e}")
+        print("Download error:", e)
 
+# ===============================
+# Main loop
+# ===============================
 async def main():
+    print("Starting Telegram bot...")
     await client.start(bot_token=BOT_TOKEN)
-    print("🤖 Telegram bot is running...")
+    print("Connected to Telegram")
     await client.run_until_disconnected()
 
 if __name__ == "__main__":