Update main.py

main.py

@@ -1,7 +1,7 @@
 import os
 import math
 import logging
-from contextlib import asynccontextmanager  # <--- NEW IMPORT
+from contextlib import asynccontextmanager
 from pyrogram import Client
 from pyrogram.session import Session, Auth
 from pyrogram import raw
@@ -16,18 +16,18 @@ API_HASH = os.environ.get("API_HASH")
 BOT_TOKEN = os.environ.get("BOT_TOKEN")
 STORAGE_CHANNEL = int(os.environ.get("STORAGE_CHANNEL"))
 
-# --- BOT CLIENT ---
+# --- BOT CLIENT (FIXED) ---
+# Added 'ipv6=False' to force IPv4 connection (Fixes Timeout)
 client = Client(
     "worker_bot",
     api_id=API_ID,
     api_hash=API_HASH,
     bot_token=BOT_TOKEN,
     in_memory=True,
-    no_updates=True 
+    no_updates=True,
+    ipv6=False  # <--- THIS IS THE FIX
 )
 
-# --- 🚀 NEW LIFESPAN LOGIC ---
-# This replaces @app.on_event("startup")
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     print("⏳ Connecting to Telegram...")
@@ -37,7 +37,7 @@ async def lifespan(app: FastAPI):
     except Exception as e:
         print(f"❌ Failed to connect: {e}")
     
-    yield  # The application runs here
+    yield
     
     print("🛑 Stopping Worker...")
     try:
@@ -45,10 +45,9 @@ async def lifespan(app: FastAPI):
     except:
         pass
 
-# Initialize FastAPI with the lifespan handler
 app = FastAPI(lifespan=lifespan)
 
-# --- STREAMING LOGIC (Unchanged) ---
+# --- STREAMING LOGIC ---
 class ByteStreamer:
     def __init__(self, client: Client):
         self.client = client
@@ -146,3 +145,4 @@ async def stream_handler(req: Request, message_id: int, filename: str):
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)
+