update main.py

bot/server/main.py

@@ -250,3 +250,66 @@ async def transmit_file_without(file_id):
             yield last_chunk
 
     return Response(file_stream(), headers=headers, status=status_code)
+    
+    
+@bp.route('/nocode/<int:file_id>', methods=['GET', 'POST', 'HEAD'])
+async def transmit_file_stream(file_id):
+    file = await get_message(file_id) or abort(404)
+    range_header = request.headers.get('Range')
+
+    file_name, file_size, mime_type = get_file_properties(file)
+
+    start = 0
+    end = file_size - 1
+    chunk_size = 1 * 1024 * 1024  # 1 MB
+
+    if range_header:
+        range_match = re_match(r'bytes=(\d+)-(\d*)', range_header)
+        if range_match:
+            start = int(range_match.group(1))
+            end = int(range_match.group(2)) if range_match.group(2) else file_size - 1
+            if start > end or start >= file_size:
+                abort(416, 'Requested range not satisfiable')
+        else:
+            abort(400, 'Invalid Range header')
+
+    offset_chunks = start // chunk_size
+    total_bytes_to_stream = end - start + 1
+    chunks_to_stream = ceil(total_bytes_to_stream / chunk_size)
+
+    content_length = total_bytes_to_stream
+    headers = {
+        'Content-Type': mime_type,
+        'Content-Disposition': f'attachment; filename={file_name}',
+        'Content-Range': f'bytes {start}-{end}/{file_size}',
+        'Accept-Ranges': 'bytes',
+        'Content-Length': str(content_length),
+    }
+    status_code = 206 if range_header else 200
+
+    async def file_stream():
+        bytes_streamed = 0
+        chunk_index = 0
+        async for chunk in TelegramBot.stream_media(
+            file,
+            offset=offset_chunks,
+            limit=chunks_to_stream,
+        ):
+            if chunk_index == 0:  # Trim the first chunk if necessary
+                trim_start = start % chunk_size
+                if trim_start > 0:
+                    chunk = chunk[trim_start:]
+
+            remaining_bytes = content_length - bytes_streamed
+            if remaining_bytes <= 0:
+                break
+
+            if len(chunk) > remaining_bytes:  # Trim the last chunk if necessary
+                chunk = chunk[:remaining_bytes]
+
+            yield chunk
+            bytes_streamed += len(chunk)
+            chunk_index += 1
+
+    return Response(file_stream(), headers=headers, status=status_code)
+