Update main.py

main.py

@@ -2,9 +2,8 @@ import os
 import asyncio
 from telethon.sync import TelegramClient
 from telethon.sessions import StringSession
-from telethon.tl.functions.upload import GetFileRequest
 from fastapi import FastAPI, UploadFile, File, HTTPException
-from fastapi.responses import RedirectResponse
+from fastapi.responses import StreamingResponse
 import uvicorn
 import tempfile
 
@@ -39,13 +38,13 @@ async def upload_file(file: UploadFile = File(...)):
             doc = message.document
             original_filename = next((attr.file_name for attr in doc.attributes if hasattr(attr, 'file_name')), file.filename)
 
+            # Simplified response with only the essentials
             return {
                 "message": "File uploaded successfully. Use the 'message_id' to download.",
                 "telegram_details": {
                     "message_id": message.id,
                     "filename": original_filename,
                     "size_bytes": doc.size,
-                    "mime_type": doc.mime_type,
                     "date": message.date.isoformat()
                 }
             }
@@ -58,42 +57,40 @@ async def upload_file(file: UploadFile = File(...)):
 @app.get("/v1/download/{message_id}")
 async def download_file(message_id: int):
     """
-    Generates a direct, temporary download link from Telegram's CDN and redirects the client to it.
-    This is extremely fast and efficient.
+    Streams a file directly from Telegram to the client, ensuring a fast download start.
     """
     if not all([SESSION_STRING, API_ID, API_HASH]):
         raise HTTPException(status_code=500, detail="Server is not configured for downloads.")
 
+    client = TelegramClient(StringSession(SESSION_STRING), API_ID, API_HASH)
+    await client.connect()
+    
     try:
-        async with TelegramClient(StringSession(SESSION_STRING), API_ID, API_HASH) as client:
-            # Get the message object
-            message = await client.get_messages(TARGET_CHANNEL, ids=message_id)
-            
-            if not message or not message.document:
-                raise HTTPException(status_code=404, detail=f"A message with ID '{message_id}' was not found or is not a file.")
+        message = await client.get_messages(TARGET_CHANNEL, ids=message_id)
+        if not message or not message.media:
+            raise HTTPException(status_code=404, detail=f"Message not found or has no media.")
+    except Exception as e:
+        await client.disconnect()
+        raise HTTPException(status_code=500, detail=f"Could not retrieve file details: {e}")
 
-            # Get the data center information where the file is stored
-            dc_id, location = message.document.dc_id, message.document
-            
-            # Use a low-level request to generate a temporary download URL
-            # This is the key to creating a CDN link
-            req = GetFileRequest(
-                location=location,
-                cdn_supported=True,
-                offset=0,
-                limit=0
-            )
-            result = await client(req)
-            
-            # The result contains the temporary URL.
-            # Format: https://{cdn_redirect_ip}/.../{file_token}
-            cdn_url = f"https://{result.cdn_redirect}/file/{result.bytes.hex()}"
-            
-            # Immediately redirect the user's client to this URL
-            return RedirectResponse(url=cdn_url, status_code=307)
+    original_filename = message.file.name if message.file and message.file.name else f"download_{message_id}"
+    mime_type = message.file.mime_type if message.file else 'application/octet-stream'
+    
+    headers = {
+        'Content-Disposition': f'attachment; filename="{original_filename}"'
+    }
 
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Failed to generate download link: {e}")
+    async def file_iterator():
+        """This async generator streams the file and ensures the client disconnects."""
+        try:
+            # Stream the file chunk by chunk
+            async for chunk in client.iter_download(message):
+                yield chunk
+        finally:
+            # Crucially, disconnect after the download is complete or fails
+            await client.disconnect()
+
+    return StreamingResponse(file_iterator(), media_type=mime_type, headers=headers)
 
 if __name__ == "__main__":
     if not all([SESSION_STRING, API_ID, API_HASH]):